Hit-RAG: Learning to Reason with Long Contexts via Preference Alignment
On the map: LLM reasoning → Test-time compute and efficient thinking
also in 3 more
Abstract (arXiv)
Despite the promise of Retrieval-Augmented Generation in grounding Multimodal Large Language Models with external knowledge, the transition to extensive contexts often leads to significant attention dilution and reasoning hallucinations. The surge in information density causes critical evidence to be submerged by voluminous noise, which complicates the discernment of relevant fragments within a dense input. In this paper, we propose Hit-RAG, a multi-stage preference alignment framework designed to resolve these cognitive bottlenecks through a progressive optimization pipeline. Our approach systematically refines the utilization of external evidence via three distinct stages. First, Supervised Fine-tuning establishes baseline context awareness to minimize information neglect. Next, Discriminative Preference Alignment enhances robustness against misleading distractors. Finally, Group-Relative Policy Optimization stabilizes logical synthesis to prevent reasoning collapse. Extensive evaluations on eight benchmarks demonstrate that Hit-RAG consistently yields substantial performance gains, enabling models to bridge the gap between context acquisition and accurate reasoning while surpassing much larger counterparts in long-context scenarios.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 10 of its 65 references that are in the map. First posted on arXiv 7 Mar 2026.
Builds on
10 of 65 references.
References · 10
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2024GPT-4o System Cardcited by 4,467
- 2025Gemini 2.5: Pushing the Frontier with Advanced Reasoning, Multimodality, Long Context, and Next Generation Agentic Capabilitiescited by 3,655
- 2017TriviaQA: A Large Scale Distantly Supervised Challenge Dataset for Reading Comprehensioncited by 2,895
- 2025Qwen3-VL Technical Reportcited by 2,534
- 2025OpenAI GPT-5 System Cardcited by 910
- 2023AlpacaFarm: A Simulation Framework for Methods that Learn from Human Feedbackcited by 738
- 2023Enabling Large Language Models to Generate Text with Citationscited by 584
- 2023Interpretable Unified Language Checkingcited by 26
- 2023MM-BigBench: Evaluating Multimodal Models on Multimodal Content Comprehension Taskscited by 18
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025Retrieval is Not Enough: Enhancing RAG Reasoning through Test-Time Critique and Optimization
- 2025Aligning Deep Implicit Preferences by Learning to Reason Defensively
- 2024Understand What LLM Needs: Dual Preference Alignment for Retrieval-Augmented Generation
- 2024Large Language Models Can Self-Improve in Long-context Reasoning
- 2026A Decomposition Perspective to Long-context Reasoning for LLMs
- 2026LongR: Unleashing Long-Context Reasoning via Reinforcement Learning with Dense Utility Rewards
- 2026SPOT: Span-level Pause-of-Thought for Efficient and Interpretable Latent Reasoning in Large Language Models
- 2025PrismRAG: Boosting RAG Factuality with Distractor Resilience and Strategized Reasoning