HAMLET: Switch your Vision-Language-Action Model into a History-Aware Policy
About
Inherently, robotic manipulation tasks are history-dependent: leveraging past context could be beneficial. However, most existing Vision-Language-Action models (VLAs) have been designed without considering this aspect, i.e., they rely solely on the current observation, ignoring preceding context. In this paper, we propose HAMLET, a scalable framework to adapt VLAs to attend to the historical context during action prediction. Specifically, we introduce moment tokens that compactly encode perceptual information at each timestep. Their representations are initialized with time-contrastive learning, allowing them to better capture temporally distinctive aspects. Next, we employ a lightweight memory module that integrates the moment tokens across past timesteps into memory features, which are then leveraged for action prediction. Through empirical evaluation, we show that HAMLET successfully transforms a state-of-the-art VLA into a history-aware policy, especially demonstrating significant improvements on long-horizon tasks that require historical context. In particular, on top of GR00T N1.5, HAMLET achieves an average success rate of 76.4% on history-dependent real-world tasks, surpassing the baseline performance by 47.2%. Furthermore, HAMLET pushes prior art performance from 64.1% to 66.4% on RoboCasa Kitchen (100-demo setup) and from 95.6% to 97.7% on LIBERO, highlighting its effectiveness even under generic robot-manipulation benchmarks.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Robotic Manipulation | LIBERO | Spatial Success Rate85.6 | 314 | |
| Kitchen manipulation | RoboCasa 24 kitchen manipulation tasks | Average Success Rate66.4 | 12 | |
| Robot Manipulation | SimplerEnv Bridge | Spoon on Towel Grasp Rate91.7 | 8 | |
| Cover-and-Stack | Real-world | Cube Coverage95.8 | 6 | |
| Pick-and-Place Twice | Real-world | PnP Once91.7 | 6 | |
| Robotic Manipulation (Aggregate) | Real-world | Average Score76.4 | 6 | |
| Simulation Success Rate | RoboCasa Kitchen | Success Rate (30 steps)52.5 | 6 | |
| Simulation Success Rate | LIBERO | Spatial Success Rate99 | 6 | |
| Swap Cubes | Real-world | Stage Cube Rate95.8 | 6 | |
| Long-horizon robotic manipulation | Pick-and-Place Three Times | Success Rate37.5 | 2 |