MEGA Hub

Group-Reflective Self-Distillation for Agentic Reinforcement Learning

Authors

Do you know Binbin Zheng?You can claim authorship or link another user.Do you know Zijun Xie?You can claim authorship or link another user.Do you know Guanqun Zhao?You can claim authorship or link another user.Do you know Enlei Gong?You can claim authorship or link another user.Do you know Xing Ma?You can claim authorship or link another user.Do you know Xiaoliang Fu?You can claim authorship or link another user.Do you know Zeyu Chen?You can claim authorship or link another user.

Abstract

Reinforcement learning with verifiable rewards (RLVR) is effective for training large language model agents. However, terminal rewards provide only coarse trajectory-level supervision, leaving successful behaviors, recurring mistakes, and incidental choices entangled in the same outcome signal. Existing agentic self-distillation methods enrich sparse supervision with natural-language skills, but skills retrieved externally or extracted from a single trajectory by stronger models may mismatch current experience, exceed the policy's capability, or remain path-specific. We propose Group-Reflective Self-Distillation (GRSD), which derives capability-aligned and outcome-discriminative guidance from the policy's own verified rollouts. For each prompt, the policy reflects on each verified trajectory in an on-policy group, and a stop-gradient snapshot contrasts the resulting reflections from successful and failed rollouts to construct group-level privileged guidance. Conditioned on this guidance, a self-teacher refines turn-level credit assignment by modulating outcome-based advantages while preserving the verifier-determined learning direction. Experiments across multiple agentic environments and model scales demonstrate that GRSD consistently outperforms competitive baselines and generalizes more effectively to unseen tasks.

Community

00