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

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Rashid, Muhammad Shihab, Meem, Jannat Ara, Dong, Yue, Hristidis, Vagelis
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916262540476416
author Rashid, Muhammad Shihab
Meem, Jannat Ara
Dong, Yue
Hristidis, Vagelis
author_facet Rashid, Muhammad Shihab
Meem, Jannat Ara
Dong, Yue
Hristidis, Vagelis
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2402_10866
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle EcoRank: Budget-Constrained Text Re-ranking Using Large Language Models
Rashid, Muhammad Shihab
Meem, Jannat Ara
Dong, Yue
Hristidis, Vagelis
Computation and Language
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.
title EcoRank: Budget-Constrained Text Re-ranking Using Large Language Models
topic Computation and Language
url https://arxiv.org/abs/2402.10866