EcoRank: Budget-Constrained Text Re-ranking Using Large Language Models

Muhammad Rashid, Jannat Meem, Yue Dong, Vagelis Hristidis


Abstract
Large Language Models (LLMs) have achieved state-of-the-art performance in text re-ranking. This process includes queries and candidate passages in the prompts, utilizing pointwise, listwise, and pairwise prompting strategies. A limitation of these ranking strategies with LLMs is their cost: the process can become expensive due to API charges, which are based on the number of input and output tokens. We study how to maximize the re-ranking performance given a budget, by navigating the vast search spaces of prompt choices, LLM APIs, and budget splits. We propose a suite of budget-constrained methods to perform text re-ranking using a set of LLM APIs. Our most efficient method, called EcoRank, is a two-layered pipeline that jointly optimizes decisions regarding budget allocation across prompt strategies and LLM APIs. Our experimental results on four popular QA and passage reranking datasets show that EcoRank outperforms other budget-aware supervised and unsupervised baselines.
Anthology ID:
2024.findings-acl.773
Volume:
Findings of the Association for Computational Linguistics ACL 2024
Month:
August
Year:
2024
Address:
Bangkok, Thailand and virtual meeting
Editors:
Lun-Wei Ku, Andre Martins, Vivek Srikumar
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
13049–13063
Language:
URL:
https://aclanthology.org/2024.findings-acl.773
DOI:
Bibkey:
Cite (ACL):
Muhammad Rashid, Jannat Meem, Yue Dong, and Vagelis Hristidis. 2024. EcoRank: Budget-Constrained Text Re-ranking Using Large Language Models. In Findings of the Association for Computational Linguistics ACL 2024, pages 13049–13063, Bangkok, Thailand and virtual meeting. Association for Computational Linguistics.
Cite (Informal):
EcoRank: Budget-Constrained Text Re-ranking Using Large Language Models (Rashid et al., Findings 2024)
Copy Citation:
PDF:
https://aclanthology.org/2024.findings-acl.773.pdf