|国家预印本平台
首页|AcuRank: Uncertainty-Aware Adaptive Computation for Listwise Reranking

AcuRank: Uncertainty-Aware Adaptive Computation for Listwise Reranking

AcuRank: Uncertainty-Aware Adaptive Computation for Listwise Reranking

来源:Arxiv_logoArxiv
英文摘要

Listwise reranking with large language models (LLMs) enhances top-ranked results in retrieval-based applications. Due to the limit in context size and high inference cost of long context, reranking is typically performed over a fixed size of small subsets, with the final ranking aggregated from these partial results. This fixed computation disregards query difficulty and document distribution, leading to inefficiencies. We propose AcuRank, an adaptive reranking framework that dynamically adjusts both the amount and target of computation based on uncertainty estimates over document relevance. Using a Bayesian TrueSkill model, we iteratively refine relevance estimates until reaching sufficient confidence levels, and our explicit modeling of ranking uncertainty enables principled control over reranking behavior and avoids unnecessary updates to confident predictions. Results on the TREC-DL and BEIR benchmarks show that our method consistently achieves a superior accuracy-efficiency trade-off and scales better with compute than fixed-computation baselines. These results highlight the effectiveness and generalizability of our method across diverse retrieval tasks and LLM-based reranking models.

Soyoung Yoon、Gyuwan Kim、Gyu-Hwung Cho、Seung-won Hwang

计算技术、计算机技术

Soyoung Yoon,Gyuwan Kim,Gyu-Hwung Cho,Seung-won Hwang.AcuRank: Uncertainty-Aware Adaptive Computation for Listwise Reranking[EB/OL].(2025-05-24)[2025-06-24].https://arxiv.org/abs/2505.18512.点此复制

评论