IGSD: Environment-Verified Hindsight Self-Distillation for Search Agents
cs.AI, cs.CL
Submitted: 2026-09-26
Updated: 2026-09-26
Code: https://github.com/PhilipGAQ/igsd_code
License: http://creativecommons.org/licenses/by-nc-sa/4.0/
The gist: On-policy self-distillation densifies agent training without external teachers: a policy conditioned on privileged hindsight provides step-level guidance for its own unprivileged rollouts.
Terminology
Abstract
On-policy self-distillation densifies agent training without external teachers: a policy conditioned on privileged hindsight provides step-level guidance for its own unprivileged rollouts. For search agents, however, hindsight can make the teacher prefer a query that does not improve retrieval from the student's state. Existing methods either distill this preference directly or filter it with model-internal scores, but neither strategy verifies the query's executed retrieval consequence. We propose Information-Gain-Gated Self-Distillation (IGSD), which verifies on-policy token proposals with environment feedback before distilling them. Treating each query token as a micro-action, IGSD completes the teacher's token proposal and the student's sampled token into matched queries and executes both from the same failed state with the same retriever. Shared counterfactual controls account for query-conditioned shifts in answer likelihood, so their difference, the executed paired information gain, provides a relative utility contrast for the retrieved documents. IGSD uses this contrast as a positive-only soft weight for candidate-pair distillation, while leaving the GRPO objective unchanged and confining verification to training. Across seven single-hop and multi-hop QA benchmarks, IGSD reaches macro-average exact-match accuracies of 42.8% and 47.0% with 3B and 7B policies, respectively, without inference-time verification. These results support environment-verified hindsight as an effective approach to reliable action-level supervision for search agents.
Sources
- ReSearch: Learning to Reason with Search for LLMs via Reinforcement Learning
- Group-in-Group Policy Optimization for LLM Agent Training
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learning
- Search-o1: Agentic Search-Enhanced Large Reasoning Models
- IG-Search: Step-Level Information Gain Rewards for Search-Augmented Reasoning
- Search-E1: Self-Distillation Drives Self-Evolution in Search-Augmented Reasoning
- Self-Distilled Agentic Reinforcement Learning
- SD-Search: On-Policy Hindsight Self-Distillation for Search-Augmented Reasoning
- Qwen2.5 Technical Report
- Search and Refine During Think: Facilitating Knowledge Refinement for Improved Retrieval-Augmented Reasoning
- Information Gain-based Policy Optimization: A Simple and Effective Approach for Multi-Turn Search Agents
- Skill-SD: Skill-Conditioned Self-Distillation for Multi-turn LLM Agents
- Text Embeddings by Weakly-Supervised Contrastive Pre-training
- SmartSearch: Process Reward-Guided Query Refinement for Search Agents
- SIGHT: Reinforcement Learning with Self-Evidence and Information-Gain Diverse Branching for Search Agent
- LOTAPO: Leave-One-Turn Attribution for Self-Generated Process Rewards in Multi-Turn Search Reasoning
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems