Sliding Windows Are Not the End: Exploring Full Ranking with Long-Context Large Language Models

Wenhan Liu, Xinyu Ma, Yutao Zhu, Ziliang Zhao, Shuaiqiang Wang +2 more
12/19/2024
cs.IRcs.CL

No official code found. Best implementation: RUC-NLPIR/fullrank (39 stars, MIT, last commit 2026-04-06).

Abstract

Large Language Models (LLMs) have shown exciting performance in listwise passage ranking. Due to the limited input length, existing methods often adopt the sliding window strategy. Such a strategy, though effective, is inefficient as it involves repetitive and serialized processing, which usually re-evaluates relevant passages multiple times. As a result, it incurs redundant API costs, which are proportional to the number of inference tokens. The development of long-context LLMs enables the full ranking of all passages within a single inference, avoiding redundant API costs. In this paper, we conduct a comprehensive study of long-context LLMs for ranking tasks in terms of efficiency and effectiveness. Surprisingly, our experiments reveal that full ranking with long-context LLMs can deliver superior performance in the supervised fine-tuning setting with a huge efficiency improvement. Furthermore, we identify two limitations of fine-tuning the full ranking model based on existing methods: (1) sliding window strategy fails to produce a full ranking list as a training label, and (2) the language modeling loss cannot emphasize top-ranked passage IDs in the label. To alleviate these issues, we propose a new complete listwise label construction approach and a novel importance-aware learning objective for full ranking. Experiments show the superior performance of our method over baselines. Our codes are available at \url{https://github.com/8421BCD/fullrank}.

View on arXivView PDF

Code Implementations(1)

RUC-NLPIR/fullrankCommunity100%
MIT
393Shell, PythonDec 18, 20246 months agoMIT

Similar Papers

Cite this paper

@article{liu2024sliding,
  title  = {Sliding Windows Are Not the End: Exploring Full Ranking with Long-Context Large Language Models},
  author = {Wenhan Liu and Xinyu Ma and Yutao Zhu and Ziliang Zhao and Shuaiqiang Wang and Dawei Yin and Zhicheng Dou},
  year   = {2024},
  eprint = {2412.14574},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2412.14574v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2412.14574.svg)](https://codeofpaper.com/paper/2412.14574)

Discussion