Search papers, labs, and topics across Lattice.
This paper introduces RoboTTT, a novel robot model that scales visuomotor context to 8K timesteps, significantly enhancing robot capabilities without increasing inference latency. By integrating Test-Time Training into robot foundation models, RoboTTT achieves remarkable improvements in one-shot imitation, on-the-fly policy adaptation, and robustness across complex tasks. The results show an 87% performance increase over traditional single-step contexts and a 62% improvement compared to models trained with shorter contexts, indicating that context length is a crucial factor in robot policy development.
Scaling visuomotor context to 8K timesteps enables robots to master complex tasks and adapt in real-time, outperforming previous models by a staggering margin.
Recent robot foundation models operate with single-step or short-history visuomotor context. We introduce Test-Time-Training Robot Policies (RoboTTT), a robot model and training recipe that scale visuomotor context to 8K timesteps, three orders of magnitude beyond state-of-the-art policies, without growing inference latency. At this context length, we unlock new robot capabilities: one-shot in-context imitation from human video demonstrations, on-the-fly policy improvement, robustness to perturbations, and stronger performance on multi-stage, long-horizon tasks. We also observe, for the first time, steady gains in closed-loop performance as pretraining context length scales. At its core, RoboTTT integrates Test-Time Training into robot foundation models such as Vision-Language-Action policies, yielding a sequence model whose recurrent state consists of fast weights, parameters updated by gradient descent during both training and inference, compressing histories into weight space and retrieving contextual information for long-context conditioning. To scale training context length, the recipe combines sequence action forcing with truncated backpropagation through time. On challenging real-robot manipulation tasks, RoboTTT improves overall performance by 87% over the single-step context baseline and fully completes a five-minute, ten-stage assembly task, which no baseline ever does. RoboTTT trained with 8K-timestep context outperforms the same model pretrained with 1K timesteps by 62%, suggesting context length as a new scaling axis for robot foundation models. Videos are available at https://research.nvidia.com/labs/gear/robottt/