Skip to content

Mandeep-Rathee/llmgar

Repository files navigation

Guiding Retrieval using LLM-based Listwise Rankers

This is official repository for our paper Guiding Retrieval using LLM-based Listwise Rankers accepted in ECIR, 2025.

Setup

Requirements

We have added all dependencies in requirements.txt file which can be downloaded as follows:

pip install "rerankers[all]"
pip install --upgrade git+https://github.com/terrierteam/pyterrier_adaptive.git
pip install pyterrier_pisa==0.0.6
pip install pyterrier_alpha

Corpus Graph

We use the coprus graphs from the GAR paper. For instance, the bm25 based corpus graph can be downloaded using:

import pyterrier_alpha as pta
graph = pta.Artifact.from_hf('macavaney/msmarco-passage.corpusgraph.bm25.128').to_limit_k(16)

Reproduction

Our results can be reproduced by using the evaluate.py file. Additionally, we have also provided the saved runs in the saved_pyterrier_runs/ folder.

python3 evaluate.py --model_name zephyr --model_type rankllm --budget 50 --verbose --dl_type 19

Citation

@article{rathee2025guiding,
  title={Guiding Retrieval using LLM-based Listwise Rankers},
  author={Rathee, Mandeep and MacAvaney, Sean and Anand, Avishek},
  journal={arXiv preprint arXiv:2501.09186},
  year={2025}
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages