Search papers, labs, and topics across Lattice.
SAIL reframes robot imitation learning as an iterative refinement problem, leveraging test-time compute to improve robustness under environmental variation. It uses Monte Carlo Tree Search to explore trajectory refinements, guided by a vision language model (VLM) for scoring and step-level feedback. Experiments across six manipulation tasks show that increasing test-time compute improves success rates, reaching up to 95% on complex tasks.
Scaling test-time compute can dramatically improve the success rate of robot imitation learning, achieving up to 95% on complex manipulation tasks.
In-context imitation learning allows robots to acquire skills from demonstrations, yet one-shot trajectory generation remains fragile under environmental variation. We propose SAIL, a framework that reframes robot imitation as an iterative refinement problem capable of scaling with test-time compute. SAIL utilizes Monte Carlo Tree Search, where each node is a complete trajectory and edges correspond to trajectory refinements. The process is guided by three core components: an automated archive of successful trajectories for contextually relevant retrieval, a vision language model-based scoring mechanism for trajectory evaluation, and a step-level feedback that provides trajectory-aligned scores for iterative refinement. Experiments across six diverse manipulation tasks in simulation and real-world validation clearly demonstrate that increasing test-time compute consistently improves success rates, achieving up to 95% on complex tasks. Our results suggest that trajectory-level test-time scaling is a robust path toward more generalizable robotic agents.