Bridging the gap between Imitation Learning and Reinforcement Learning through reward-free, self-generated environment interactions.
Autonomous language agents need to learn from experience, but existing paradigms face fundamental bottlenecks.
Using the agent's proposed non-expert actions and resulting future states as direct supervision—without requiring external rewards.
Trained on static expert demonstrations. Fails on unobserved states.
Agent proposes actions, collects future states, and converts environment feedback into direct self-supervision.
Optimizes cumulative rewards through trial & error. Constrained by simulator availability.
Transforming exploratory rollouts into effective learning objectives.
Concept: Internalizes environment dynamics by learning to predict future textual states resulting from non-expert actions.
Concept: Compares expert actions against proposed alternatives to synthesize fine-grained rationale monologues.
Select a method tab to inspect how early experience constructs learning data.
Environment: WebShop (E-commerce task: "Blue wireless headphones under $130")
*The policy internalizes that clicking "Prev" loses the current product selection state without needing explicit rewards.
Self-Reflection Monologue Generation Prompt & Target:
Consistent improvements over Supervised Fine-Tuning across model families (Llama-3.2-3B, Qwen-2.5-7B, Llama-3.1-8B).
| Environment Domain | Benchmark | SFT Baseline | Ours (IWM) | Ours (Self-Reflection) | Absolute Gain |
|---|---|---|---|---|---|
| Embodied Sim | ALFWorld | 80.5% | 85.9% | 85.2% | +5.4% |
| Scientific Sim | Science World | 54.7% | 57.0% | 68.0% | +13.3% |
| Planning | TravelPlanner | 17.2% | 25.0% | 32.2% | +15.0% |
| Tool-Use API | BFCLv3 | 6.7% | 20.0% | 20.0% | +13.3% |
| Customer Support | Tau-Bench | 35.9% | 40.8% | 41.7% | +5.8% |
| Web Navigation | WebShop | 47.3% | 58.6% | 58.2% | +11.3% |
Initializing post-hoc RL (GRPO) with Early Experience checkpoints yields substantially higher post-RL performance ceilings.
SFT → GRPO: 80.5%
IWM → GRPO: 91.4%
Higher convergence ceiling when environment rewards become available.
SFT → GRPO: 93.8%
SR → GRPO: 98.5%
Pre-grounding prevents policy collapse during early RL exploration steps.
SFT → GRPO: 47.1%
SR → GRPO: 51.0%
Reduces hallucinated search query generation during online RL steps.
Early Experience transforms unrewarded agent interaction into high-value self-supervision.