Chunji Lv, Yangguang Wei, Junlin Liu, Yang Gao, Ming Liu, Xinming Wang, Jinyang Wu, Guoren Wang, Changsheng Li
5 min
In agentic reinforcement learning, agents often face sparse, delayed rewards that make credit assignment difficult. While on-policy self-distillation (OPSD) can provide dense token-level supervision using a privileged teacher, existing methods struggle to determine which teacher signals are reliable. The authors investigate how to effectively filter noisy teacher guidance to improve agent performance without relying on inference-time skills.
The authors propose Persistent Consistency Self-Distillation (PCSD), a framework that estimates teacher credibility by analyzing the local persistence of teacher-favoring signals. Instead of treating each token in isolation or applying uniform weights to entire steps, PCSD uses three mechanisms:
PCSD consistently outperforms baseline methods, including standard GRPO and existing self-distillation weighting schemes, across ALFWorld and WebShop benchmarks. Specifically, on ALFWorld, PCSD achieved significant gains over GRPO (15.6 and 13.3 points on different backbones) and outperformed the SDAR baseline. The results indicate that by focusing on persistent local evidence, the agent can effectively filter out noise and leverage high-quality teacher guidance to improve fine-grained credit assignment.
This work addresses the fundamental trade-off between token-level precision and robustness in self-distillation. By providing a principled way to weigh teacher supervision, PCSD enables more effective learning in long-horizon agentic tasks where environmental feedback is sparse. It demonstrates that incorporating local context into the distillation process is a powerful strategy for improving agent training without increasing inference-time computational costs.
Large language model agents have shown strong potential in complex interactive tasks, yet their reinforcement learning (RL) is often hindered by sparse rewards, as a long multi-turn trajectory may receive only a single outcome-level signal. On-policy self-distillation (OPSD) provides dense token-level supervision from a privileged teacher, but the teacher may not be reliable at every position. Existing methods commonly rely on isolated token-level discrepancies, which can be sensitive to noise, or assign a shared step-level weight that may overlook positional variation. We propose Persistent Consistency Self-Distillation (PCSD), which derives token-level distillation weights from the local persistence of teacher-favoring signals. PCSD combines adaptive windows with exponentially decayed aggregation to capture persistent relative teacher support, applies trend-aware modulation to attenuate locally declining support, and produces continuous weights through sigmoid gating. The resulting objective is jointly optimized with GRPO, combining dense teacher guidance with sparse environmental feedback. Without inference-time skills, PCSD achieves the best ALFWorld Overall results among all baselines on both backbones, exceeding GRPO by 15.6 and 13.3 points and SDAR by 6.2 and 5.5 points, while remaining competitive on WebShop and gaining 15.8 points over GRPO on unseen ALFWorld split.
Alex: So it's not just "is the teacher confident right now?" but "is the teacher's confidence holding up over time?"
Sam: Correct. And this is what makes the approach meaningful rather than just a minor tweak. Most systems either copy the teacher directly or ignore the teacher entirely. PCSD does something more nuanced—it continuously evaluates whether the teacher's guidance is worth following at each moment, and adjusts accordingly.
Alex: How does this connect to the broader training process? Because I'd imagine the student still needs some way to know if it actually succeeded at the task.
Sam: Good question. The system combines two sources of feedback. There's the final reward—a simple signal at the end of a task saying "you did it" or "you didn't." That tells the student whether the overall strategy worked. But on its own, that's a very sparse signal. You might take a hundred steps to complete a task, and only find out at the very end whether you were on the right track. The persistent consistency filter fills in that gap. It provides moment-to-moment guidance at each step, but only from the parts of the teacher's behavior that have proven stable and trustworthy. The student gets the best of both: a reliable final verdict, and a filtered, high-quality signal to learn from along the way.
Alex: So the student isn't just blindly copying the teacher, and it isn't flying blind either. It's using the teacher's guidance selectively, based on how consistent and reliable that guidance has been.
Sam: That's the idea. And the paper suggests this makes the student more robust on complex tasks—the kind where a single misstep early on can cascade into failure later. By learning from persistent, stable signals rather than noisy ones, the student internalizes the underlying skill rather than just mimicking surface-level behavior.
Alex: That's a meaningful distinction. It's the difference between a student who understands why something works versus one who just memorizes the answer.
Sam: Exactly. And it points to a broader principle in AI training: the quality of the learning signal matters as much as the quantity. More data from a noisy teacher isn't always better. A smaller, well-filtered signal can produce a more capable student. That's what PCSD is designed to deliver.
Alex: That's a clear and useful idea—and one that probably has implications well beyond this particular paper. Thanks for walking us through it.
Sam: Thanks for listening to ResearchPod.