Self-Retrospection Distillation: Turning Post-hoc Experiences into Prior Foresight

Haoxiang Zhang, Pan Lu, Silvio Savarese and colleagues (Salesforce AI Research, Stanford) propose Self-Retrospection Distillation (SRD), which trains a policy to predict, before acting, the knowledge and failure modes that its completed trajectories revealed.
Ask this paper
Problem. RLVR learns only from outcome contrast. When every rollout in a group gets the same reward, which is common on long-horizon tasks, the group-relative advantage is zero and the samples teach nothing.
Method. SRD turns hindsight from finished trajectories into trajectory-blind foresight targets for the same policy. Foresight is a training target only and is not generated at inference.
All-failure regime. At 2B, where 98% of rollout groups are all-failure, RLVR ends at 0.0% success, while adding SRD reaches 60.6% under the same rollout budget.
Gains across objectives. For 4B GRPO, SRD adds 10.03, 8.33 and 8.05 pp on Math, Code and Search averages. At 9B, OPSD+SRD raises Math from 39.70% to 56.92%, and RLSD+SRD raises ALFWorld from 66.25% to 77.25%. The only regression is WebShop under 9B OPSD (-3.61 pp).
Transfer. Gains hold on unseen ALFWorld environments (1.12 to 13.74 pp) and on BrowseComp-Plus, which needs 10x more tool calls than training rollouts (up to 9.52 pp).
Abstract
Reinforcement learning with verifiable rewards (RLVR) turns agent experience into learning signals primarily through scalar outcome rewards after interaction. For group-relative objectives, however, this signal vanishes when all rollouts receive the same reward, even though their trajectories may reveal useful information about what the task requires and how the agent fails. We ask a complementary question: can hindsight teach an agent what it could have anticipated before acting? We introduce prospective learning, which uses post-hoc experience to supervise foresight predictions from the pre-interaction view, and instantiate it with Self-Retrospection Distillation (SRD). Intuitively, a completed trajectory reveals knowledge that would have been useful and pitfalls that should be avoided; SRD distills this privileged hindsight into trajectory-blind foresight of the same policy. Foresight serves only as a training target and need not be explicitly generated at inference time. Across 10 tool-integrated reasoning and long-horizon agentic tasks, SRD complements RLVR and self-distillation baselines with gains of up to 24.2 pp. Its advantage is especially pronounced when reward contrast is scarce: when 37--98% of rollout groups are reward-uniform across model scales, yet SRD can still exploit learning signal from sampled trajectories. In the 2B setting, where 98% of groups are all-failure, the RLVR training ends up at 0.0% success, while adding SRD reaches 60.6% under the same rollout budget. Our results suggest that post-hoc agent experience is useful not only for evaluating or improving behavior, but also for shaping predictive representations before available interaction.