ReAttn: Improving Attention-based Re-ranking via Attention Re-weighting
Not on the map. Why?
Abstract (arXiv)
The strong capabilities of recent Large Language Models (LLMs) have made them highly effective for zero-shot re-ranking task. Attention-based re-ranking methods, which derive relevance scores directly from attention weights, offer an efficient and interpretable alternative to generation-based re-ranking methods. However, they still face two major limitations. First, attention signals are highly concentrated a small subset of tokens within a few documents, making others indistinguishable. Second, attention often overemphasizes phrases lexically similar to the query, yielding biased rankings that irrelevant documents with mere lexical resemblance are regarded as relevant. In this paper, we propose ReAttn, a post-hoc re-weighting strategy for attention-based re-ranking methods. It first compute the cross-document IDF weighting to down-weight attention on query-overlapping tokens that frequently appear across the candidate documents, reducing lexical bias and emphasizing distinctive terms. It then employs entropy-based regularization to mitigate over-concentrated attention, encouraging a more balanced distribution across informative tokens. Both adjustments operate directly on existing attention weights without additional training or supervision. Extensive experiments demonstrate the effectiveness of our method.
Key facts. Cited by 5 papers in the map (data through 7 Oct 2026). Builds on 36 of its 51 references that are in the map. First posted on arXiv 23 Feb 2026.
Builds on
36 of 51 references.
References · 36
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,160
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
- 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,779
- 2021Calibrate Before Use: Improving Few-Shot Performance of Language Modelscited by 1,371
- 2018FEVER: a large-scale dataset for Fact Extraction and VERificationcited by 1,362
- 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,225
- 2022In-context Learning and Induction Headscited by 930
- 2019Analyzing Multi-Head Self-Attention: Specialized Heads Do the Heavy Lifting, the Rest Can Be Prunedcited by 844
- 2019Are Sixteen Heads Really Better than One?cited by 803
- 2023Extending Context Window of Large Language Models via Positional Interpolationcited by 721
- 2024LongMemEval: Benchmarking Chat Assistants on Long-Term Interactive Memorycited by 656
- 2019Attention is not not Explanationcited by 629
24 more
- 2020Fact or Fiction: Verifying Scientific Claimscited by 546
- 2020SPECTER: Document-level Representation Learning using Citation-informed Transformerscited by 426
- 2019Is Attention Interpretable?cited by 387
- 2020Document Ranking with a Pretrained Sequence-to-Sequence Modelcited by 373
- 2020Distilling Knowledge from Reader to Retriever for Question Answeringcited by 226
- 2024Retrieval Head Mechanistically Explains Long-Context Factualitycited by 219
- 2019Multi-Stage Document Ranking with BERTcited by 200
- 2020CLIMATE-FEVER: A Dataset for Verification of Real-World Climate Claimscited by 195
- 2024Data Engineering for Scaling Language Models to 128K Contextcited by 183
- 2022Improving Passage Retrieval with Zero-Shot Question Generationcited by 164
- 2024LLM Maybe LongLM: Self-Extend LLM Context Window Without Tuningcited by 133
- 2023Attention Sorting Combats Recency Bias In Long Context Language Modelscited by 89
- 2024Cutting Off the Head Ends the Conflict: A Mechanism for Interpreting and Mitigating Knowledge Conflicts in Language Modelscited by 87
- 2025Which Attention Heads Matter for In-Context Learning?cited by 61
- 2023ConvGQR: Generative Query Reformulation for Conversational Searchcited by 59
- 2024Identifying Semantic Induction Heads to Understand In-Context Learningcited by 46
- 2024Attention in Large Language Models Yields Efficient Zero-Shot Re-Rankerscited by 42
- 2024RATT: A Thought Structure for Coherent and Correct LLM Reasoningcited by 39
- 2024IRCAN: Mitigating Knowledge Conflicts in LLM Generation via Identifying and Reweighting Context-Aware Neuronscited by 36
- 2025Query-Focused Retrieval Heads Improve Long-Context Reasoning and Re-rankingcited by 30
- 2025Entropy-based Exploration Conduction for Multi-step Reasoningcited by 28
- 2025UniConv: Unifying Retrieval and Response Generation for Large Language Models in Conversationscited by 11
- 2026OpenDecoder: Open Large Language Model Decoding to Incorporate Document Quality in RAGcited by 11
- 2025Focus Directions Make Your Language Models Pay More Attention to Relevant Contextscited by 9
Built on by
Citing papers per year: 2026: 5.
5 papers.
- 2026Preference Heads in Large Language Models: A Mechanistic Framework for Interpretable Personalizationcited by 2
- 2026Context-Fidelity Boosting: Enhancing Faithful Generation through Watermark-Inspired Decodingcited by 1
- 2026Learning to Route Queries to Heads for Attention-based Re-ranking with Large Language Modelscited by 0
- 2026Knowing Before Answering: Decoding Language Models for Reliable RAGcited by 0
- 2026How Calibration Content Shapes Attention-Based Rerankingcited by 0
- 2026How Calibration Content Shapes Attention-Based Reranking15 Sep 2026
- 2026Knowing Before Answering: Decoding Language Models for Reliable RAG27 Aug 2026
- 2026Learning to Route Queries to Heads for Attention-based Re-ranking with Large Language Models27 Apr 2026
- 2026Preference Heads in Large Language Models: A Mechanistic Framework for Interpretable Personalization24 Apr 2026
- 2026Context-Fidelity Boosting: Enhancing Faithful Generation through Watermark-Inspired Decoding24 Apr 2026