Search papers, labs, and topics across Lattice.
This paper introduces LaMem-VLA, a novel framework that integrates dual latent memory directly into Vision-Language-Action (VLA) models to enhance their performance on long-horizon tasks. By reconstructing historical experiences into latent memory tokens and interweaving them with multimodal reasoning, LaMem-VLA allows for a more fluid integration of past context into action formation. Experimental results on SimplerEnv and LIBERO show that LaMem-VLA significantly outperforms existing memory-augmented VLA models, addressing the limitations of traditional Markovian approaches in robotic manipulation tasks.
LaMem-VLA seamlessly integrates historical experience into VLA reasoning, enabling robots to perform complex tasks with improved contextual awareness.
Mainstream Vision-Language-Action (VLA) models predict actions primarily from the current observation under a Markovian assumption, thus struggling with long-horizon, temporally dependent tasks. Existing memory-augmented VLAs either expand the observation window or retrieve history from the memory bank as auxiliary policy-side context. However, they leave memory outside the native latent embedding space of VLA reasoning, preventing historical experience from being fluidly interleaved with multimodal reasoning and action formation. To this end, we introduce LaMem-VLA, a latent-memory-native framework that reconstructs historical experience into latent memory tokens and directly interweaves them with VLA reasoning. At its core, LaMem-VLA introduces four coordinated components: (i) a curator that organizes historical experience into two complementary short-term and long-term memory vaults; (ii) a seeker that queries both vaults using the multimodal cognition to retrieve context-relevant evidence; (iii) a condenser that reconstructs the retrieved evidence into compact short-term and long-term latent memory tokens; and (iv) a weaver that injects these memory tokens with the current observation and instruction into one continuous embedding sequence. By representing, retrieving, and consuming historical experience entirely in the same continuous latent space, LaMem-VLA enables memory to directly participate in VLA reasoning and guide action generation under a bounded context. Extensive experiments on SimplerEnv and LIBERO demonstrate the superiority of our LaMem-VLA.
