Why Does Feedback-Augmented Self-Distillation Fail to Improve Retrieval-Interleaved Search Agents?

2026-07-20Artificial Intelligence

Artificial Intelligence
AI summary

The authors study a method called on-policy self-distillation (OPSD) for training language models without a separate teacher. They find a problem called decoding collapse, where the model repeats similar patterns that don't relate well to the input question, making learning signals less useful. To fix this, the authors identify inconsistencies in the feedback, especially from prompts, and propose using an exponential moving average (EMA) teacher to provide steadier guidance. Although this approach needs a warm-up period, it eventually improves the model's learning stability and performance.

on-policy self-distillationlarge language modelsagentic searchdecoding collapseKL divergenceself-teacher learningmodel inconsistencyprompt inconsistencyexponential moving averagesupervision signals
Authors
Fan Yang, Rui Meng, Yuxin Wen
Abstract
On-policy self-distillation (OPSD) offers a promising approach for training large language models without relying on a separate teacher model. However, its effectiveness on complex agentic tasks remains largely unexplored. In this work, we instantiate Feedback-Augmented Self-Distillation (FA-SD), a self-distillation algorithm for agentic search that leverages successful demonstrations as privileged information. We identify that models can rely on recurring reasoning-and-search output templates, producing trajectories that appear diverse but are largely agnostic to the input question, making the KL-based self-distillation signal uninformative. We term this phenomenon decoding collapse, a failure mode that can be missed by existing evaluation metrics. To understand its underlying cause, we show that although the self-teacher achieves stronger performance, learning remains inherently unstable due to inconsistent supervision signals. We further decompose this inconsistency into model inconsistency and prompt inconsistency, and show that the latter can significantly degrade the quality of the supervision signal, limiting the effectiveness of self-teacher learning. To mitigate this inconsistency, we introduce an exponential moving average (EMA) teacher to stabilize the self-teacher and provide more consistent supervision signals. Although the EMA teacher requires a warm-up phase during which performance may temporarily regress, it ultimately improves model performance by providing more stable supervision.