English

HeadRank: Decoding-Free Passage Reranking via Preference-Aligned Attention Heads

Information Retrieval 2026-05-20 v2 Artificial Intelligence

Abstract

Decoding-free reranking methods that read relevance signals directly from LLM attention weights offer significant latency advantages over autoregressive approaches, yet suffer from attention score homogenization: middle-context documents receive near-identical scores, destroying the fine-grained distinctions required for ranking. We propose HeadRank, a framework that lifts preference optimization from discrete token space into the continuous attention domain through entropy-regularized head selection, hard adjacent-level preference pairs, and a distribution regularizer that jointly sharpen discriminability in the homogenized middle zone. Depth truncation at the deepest selected layer further reduces inference to O(1)\mathcal{O}(1) forward passes. Across 14 benchmarks on three Qwen3 scales (0.6B--4B) using only 211 training queries, HeadRank achieves the highest average NDCG@10 at every scale, outperforming both generative and decoding-free baselines on the majority of benchmarks with 100\% formatting success. At 4B, 57.4\% of relevant middle-zone documents reach the top quartile versus 14.2\% for irrelevant ones -- a 43-percentage-point selectivity gap that demonstrates the effectiveness of attention-space preference alignment for listwise reranking.

Keywords

Cite

@article{arxiv.2604.17237,
  title  = {HeadRank: Decoding-Free Passage Reranking via Preference-Aligned Attention Heads},
  author = {Juyuan Wang and Chenxing Wang and Yuchen Fang and Huiyun Hu and Junwu Du and Aolin Li and Shunlin Rong and Haijun Wu and Jin Xu and Ligang Liu and Dongliang Liao},
  journal= {arXiv preprint arXiv:2604.17237},
  year   = {2026}
}
R2 v1 2026-07-01T12:16:30.810Z