mlsanityData to 14 Oct 2026

Answer First, Reason Later: Aligning Search Relevance via Mode-Balanced Reinforcement Learning

arXiv 2602.10006 ↗First posted 10 Feb 2026cs.LGcited by 0

On the map: LLM reasoning → Knowledge- and tool-augmented reasoning

Abstract (arXiv)

Building a search relevance model that achieves both low latency and high performance is a long-standing challenge in the search industry. To satisfy the millisecond-level response requirements of online systems while retaining the interpretable reasoning traces of Large Language Models (LLMs), we propose a novel Answer-First, Reason Later (AFRL) paradigm. This paradigm requires the model to output the definitive relevance score in the very first token, followed by a structured logical explanation. Inspired by the success of reasoning models, we adopt a "Supervised Fine-Tuning (SFT) + Reinforcement Learning (RL)" pipeline to achieve AFRL. However, directly applying existing RL training often leads to mode collapse in the search relevance task, where the model forgets complex long-tail rules in pursuit of high rewards. From an information theory perspective: RL inherently minimizes the Reverse KL divergence, which tends to seek probability peaks (mode-seeking) and is prone to "reward hacking." On the other hand, SFT minimizes the Forward KL divergence, forcing the model to cover the data distribution (mode-covering) and effectively anchoring expert rules. Based on this insight, we propose a Mode-Balanced Optimization strategy, incorporating an SFT auxiliary loss into Stepwise-GRPO training to balance these two properties. Furthermore, we construct an automated instruction evolution system and a multi-stage curriculum to ensure expert-level data quality. Extensive experiments demonstrate that our 32B teacher model achieves state-of-the-art performance. Moreover, the AFRL architecture enables efficient knowledge distillation, successfully transferring expert-level logic to a 0.6B model, thereby reconciling reasoning depth with deployment latency.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Knowledge- and tool-augmented reasoning, one of 17 directions in LLM reasoning. Builds on 25 of its 41 references that are in the map. First posted on arXiv 10 Feb 2026.

Builds on

25 of 41 references.

Closest ancestors · 3

Same direction, newest first

  1. 2025ReSearch: Learning to Reason with Search for LLMs via Reinforcement Learning25 Mar
  2. 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learning12 Mar
  3. 2025Rank-R1: Enhancing Reasoning in LLM-based Document Rerankers via Reinforcement Learning8 Mar

Other references · 22

  1. 2023GPT-4 Technical Reportcited by 18,396
  2. 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
  3. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  4. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  5. 2025Qwen3 Technical Reportcited by 8,820
  6. 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,330
  7. 2022Self-Consistency Improves Chain of Thought Reasoning in Language Modelscited by 5,708
  8. 2024OpenAI o1 System Cardcited by 1,244
  9. 2025Does Reinforcement Learning Really Incentivize Reasoning Capacity in LLMs Beyond the Base Model?cited by 982
  10. 2023Is ChatGPT Good at Search? Investigating Large Language Models as Re-Ranking Agentscited by 360
  11. 2025Reinforcement Learning for Reasoning in Large Language Models with One Training Examplecited by 295
  12. 2023Fine-Tuning LLaMA for Multi-Stage Text Retrievalcited by 272
10 more
  1. 2025The Landscape of Agentic Reinforcement Learning for LLMs: A Surveycited by 205
  2. 2025Retaining by Doing: The Role of On-Policy Data in Mitigating Forgettingcited by 94
  3. 2025AgentEvolver: Towards Efficient Self-Evolving Agent Systemcited by 85
  4. 2024Reinforcement Learning Enhanced LLMs: A Surveycited by 71
  5. 2025Reinforcement Learning Meets Large Language Models: A Survey of Advancements and Applications Across the LLM Lifecyclecited by 35
  6. 2024A Survey of Generative Search and Recommendation in the Era of Large Language Modelscited by 15
  7. 2025SEEA-R1: Tree-Structured Reinforcement Fine-Tuning for Self-Evolving Embodied Agentscited by 15
  8. 2025CLPO: Curriculum Learning meets Policy Optimization for LLM Reasoningcited by 13
  9. 2024Consolidating Ranking and Relevance Predictions of Large Language Models through Post-Processingcited by 9
  10. 2026ETR: Outcome-Guided Elastic Trust Regions for Policy Optimizationcited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Reinforcement Fine-Tuning for Reasoning towards Multi-Step Multi-Source Search in Large Language Models
  2. 2025SEM: Reinforcement Learning for Search-Efficient Large Language Models
  3. 2025Optimizing Generative Ranking Relevance via Reinforcement Learning in Xiaohongshu Search
  4. 2025An Empirical Study on Reinforcement Learning for Reasoning-Search Interleaved LLM Agents
  5. 2026Search-GRT: Guided Retrieval Training of Search Agents to Optimize for Complex Question Answering
  6. 2025TaoSR-AGRL: Adaptive Guided Reinforcement Learning Framework for E-commerce Search Relevance
  7. 2025Scent of Knowledge: Optimizing Search-Enhanced Reasoning with Information Foraging
  8. 2026IG-Search: Step-Level Information Gain Rewards for Search-Augmented Reasoning