Search papers, labs, and topics across Lattice.
This paper introduces Reflective Test-Time Planning for embodied LLMs, enabling robots to learn from mistakes during deployment. The approach integrates reflection-in-action (test-time scaling for action selection), reflection-on-action (test-time training based on external feedback), and retrospective reflection (hindsight-based model updates). Experiments on Long-Horizon Household and MuJoCo Cupboard Fitting benchmarks demonstrate significant performance improvements and behavioral correction through reflection, validated by ablations and real-robot trials.
Robots can now learn from their mistakes in real-time via a novel reflective planning framework, leading to significant performance gains in long-horizon tasks.
Embodied LLMs endow robots with high-level task reasoning, but they cannot reflect on what went wrong or why, turning deployment into a sequence of independent trials where mistakes repeat rather than accumulate into experience. Drawing upon human reflective practitioners, we introduce Reflective Test-Time Planning, which integrates two modes of reflection: \textit{reflection-in-action}, where the agent uses test-time scaling to generate and score multiple candidate actions using internal reflections before execution; and \textit{reflection-on-action}, which uses test-time training to update both its internal reflection model and its action policy based on external reflections after execution. We also include retrospective reflection, allowing the agent to re-evaluate earlier decisions and perform model updates with hindsight for proper long-horizon credit assignment. Experiments on our newly-designed Long-Horizon Household benchmark and MuJoCo Cupboard Fitting benchmark show significant gains over baseline models, with ablative studies validating the complementary roles of reflection-in-action and reflection-on-action. Qualitative analyses, including real-robot trials, highlight behavioral correction through reflection.
AI2
Stanford HAI
UW