MemoryVAM: Integrating Memory into Video Action Model for Robot Manipulation
For robot manipulation tasks requiring long-horizon memory, this work provides a practical memory module that significantly boosts performance over memory-free baselines.
MemoryVAM integrates an episodic memory mechanism into video-world-model policies for robot manipulation, addressing the non-Markovian nature of long-horizon tasks. It improves success from 5% to 42.5% on LIBERO-Mem and achieves 78.3% on real robot counting tasks.
Video-world-model policies learn action-relevant representations by predicting future observations. However, they condition on only a short observation window, which renders long-horizon manipulation non-Markovian when the correct action depends on earlier events that are no longer visible. We present MemoryVAM, an episodic memory mechanism for video-world-model policies. We employ a Recap-Cue (RC) module, in which a Perceiver-based Recap Compressor maps per-frame CLIP embeddings into compact memory tokens, and a lightweight Cue Gate estimates task completion from memory and language. These tokens are injected into both the video backbone and the action decoder, aligning policy imagination with episode progress and conditioning actions on history. Our model trains the memory module with video prediction, a delta-reconstruction auxiliary loss, and episode-boundary supervision, requiring no per-frame progress labels. The same mechanism applies to UNet and Diffusion Transformer (DiT) backbones by changing only the cross-attention injection interface. On LIBERO-Mem, our model improves average success from 5% to 42.5%. On real robots, it achieves 78.3% success on counting tasks, 80.0% on spatial recall, and 75.0% on sequential tracking. Project page: https://MemoryVAM.github.io/