Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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

About

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 $\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.

Juyuan Wang, Chenxing Wang, Yuchen Fang, Huiyun Hu, Junwu Du, Aolin Li, Shunlin Rong, Haijun Wu, Jin Xu, Ligang Liu, Dongliang Liao• 2026

Related benchmarks

TaskDatasetResultRank
Passage RerankingTREC Deep Learning 2019
NDCG@1064.33
19
Passage RerankingTREC Deep Learning 2020
NDCG@1059.89
19
Passage RerankingCOVID BEIR
NDCG@1073.94
19
Passage RerankingArguana BEIR
NDCG@1036.59
19
Passage RerankingScidocs BEIR
NDCG@1016.96
19
Passage RerankingNFC BEIR
NDCG@1036.44
19
Passage RerankingNQ BEIR
NDCG@1045.36
19
Passage RerankingFEVER BEIR
NDCG@1073.47
19
Passage RerankingNews BEIR
NDCG@1046.4
19
Passage RerankingFiQA BEIR
NDCG@1028.73
19
Showing 10 of 13 rows

Other info

Follow for update