🤖 AI Summary
This work addresses the challenge in outcome-based reinforcement learning where trajectory-level sparse rewards provide insufficient guidance for token-level decision-making. To this end, the authors propose the SEED framework, which leverages a self-evolving retrospective skill mechanism to transform completed on-policy trajectories into dense distillation signals aligned with the current policy distribution. SEED jointly optimizes decision-making and skill extraction during policy updates and integrates large language models, natural language skill generation, and policy distillation to construct a skill-augmented contextual re-scoring mechanism. Experimental results demonstrate that SEED significantly improves performance, sample efficiency, and generalization to unseen scenarios across both text-based and vision-based agent tasks.
📝 Abstract
Large language models are increasingly trained as interactive agents for long-horizon tasks involving multi-turn interaction, tool use, and environment feedback. Outcome-based reinforcement learning (RL) provides a practical optimization paradigm, but its sparse trajectory-level rewards offer limited guidance on intermediate decisions, leaving a supervision gap between episode-level outcomes and token-level policy learning. We propose SEED (SElf-Evolving On-Policy Distillation), a self-evolving framework that converts completed on-policy trajectories into training-time hindsight skills and distills their behavioral effect back into the policy model. SEED first fine-tunes the policy to analyze completed trajectories and generate natural-language skills that capture reusable workflows, decisive observations, or failure-avoidance rules. During RL, the current policy both collects trajectories and serves as the analyzer that extracts hindsight skills from them. Policy updates therefore improve subsequent decision making and skill analysis together, allowing hindsight supervision to evolve with the policy. SEED then re-scores the sampled actions under ordinary and skill-augmented contexts, converting the skill-induced probability shift into a dense token-level on-policy distillation signal. This signal is jointly optimized with outcome-based RL, keeping the auxiliary supervision aligned with the current trajectory distribution. Extensive experiments on text-based and vision-based agentic tasks show that SEED consistently improves performance and sample efficiency, exhibiting robust generalization to unseen scenarios. Our code is available at https://github.com/jinyangwu/SEED.