ReMem-VLA: Empowering Vision-Language-Action Model with Memory via Dual-Level Recurrent Queries
Authors: Hang Li, Fengyi Shen, Dong Chen, Liudi Yang, Xudong Wang, Jinkui Shi, Zhenshan Bing, Ziyuan Liu, +1 more
Organizations: Technical University of Munich, Munich, Germany · Huawei Heisenberg Research Center, Munich, Germany · University of Freiburg, Freiburg, Germany · Huawei Technologies, China · Nanjing University, China
Vision-language-action (VLA) models for closed-loop robot control are typically cast under the Markov assumption, making them prone to errors on tasks requiring historical context. To incorporate memory, existing VLAs either retrieve from a memory bank, which can be misled by distractors, or extend the frame window, whose fixed horizon still limits long-term retention. In this paper, we introduce ReMem-VLA, a Recurrent Memory VLA model equipped with two sets of learnable queries: frame-level recurrent memory queries for propagating information across consecutive frames to support short-term memory, and chunk-level recurrent memory queries for carrying context across temporal chunks for long-term memory. These queries are trained end-to-end to aggregate and maintain relevant context over time, implicitly guiding the model's decisions without additional training or inference cost. Furthermore, to enhance visual memory, we introduce Past Observation Prediction as an auxiliary training objective. Through extensive memory-centric simulation and real-world robot experiments, we demonstrate that ReMem-VLA exhibits strong memory capabilities across multiple dimensions, including spatial, sequential, episodic, temporal, and visual memory. ReMem-VLA significantly outperforms memory-free VLA baselines π0.5 and OpenVLA-OFT and surpasses MemoryVLA on memory-dependent tasks by a large margin.
Figures & tables
Figure 1: Overview of ReMem-VLA. Compared to vanilla VLA models, ReMem-VLA is equipped with frame-level recurrent memory for short-term retention (e.g., maintaining a fixed pose over several seconds) and chunk-level recurrent memory for long-term memory (e.g., tracking overall task progress). Additionally, ReMem-VLA incorporates past observation prediction to recall historical visual information to strengthen visual memory. ReMem-VLA demonstrates superior performance across spatial, temporal, episodic, sequential, and visual memory benchmarks, substantially outperforming all baselines.
Figure 2: Model Architecture of ReMem-VLA. ReMem-VLA continuously processes visual observations and language instructions at each timestep, predicting an action chunk and a past frame. Simultaneously, two sets of learnable recurrent memory queries extract information from the VLM backbone and propagate it to future timesteps via exponential moving average (EMA) updates. Among them, Frame-level memory queries are updated at every frame for short-term memory, while chunk-level memory queries are updated only at chunk boundaries (every K frames) for long-term retention. Action queries and hindsight queries extract features from the VLM and interact with these recurrent memory queries through bidirectional attention in connector, fusing current observations with accumulated historical context to condition the predictions.
Figure 3: MemoryBench [ 17 ] with our extended long horizon task. 1)Put Block Back: Place the block to the center, presses a button, and then returns the block to its original location. 2)Rearrange Block: Move the block from the center to the unoccupied red patch, presses a button, and then relocates the block that was originally on the red patch to the center. 3)Reopen Drawer: Close the open drawer, presses a button, and then reopen the same drawer. 4) Long Horizon Task: We extend the memorybench with an additional long horizon task to evaluate the VLA’s memory over long horizon. This task is a combination of Put Block Back and Rearrange Block.
Methods
Scores (%)
Average
Put Block back
Rearrange Block
Reopen Drawer
Long Horizon Task
OpenVLA-OFT [ 27 ]
0
0
3
0
0.75
π0.5 [ 22 ]
6
4
20
3
8.25
MemoryVLA [ 45 ]
0
1
5
0
1.5
ReMem-VLA (ours)
93
99
100
86
94.5
Table 1: The extended MemoryBench [ 17 ] experiments. ReMem-VLA achieves the best performance across all tasks compared to baselines.
Figure 4: Real world experiments and quantitative results on memory-dependent tasks.
Methods
Scores (%)
Average
Put Block back
Rearrange Block
Reopen Drawer
Long Horizon Task
No Recurrent Query
7
32
27
5
17.75
Frame Level
90
95
96
70
87.75
Chunk Level
81
92
94
71
84.5
Dual Level
93
99
100
86
94.5
Table 2: Ablation on different recurrent-query levels.
Figure 5: Failure analysis and Ablation on Past Observation Prediction(POP).
Figure 6: Ablation study on (a) different recurrent path; (b) number of recurrent queries. (c) different EMA parameters
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.
Hongyu Qu, Jianzhe Gao, Xiaobin Hu +6
Nanjing University of Science and Technology · Zhejiang University · National University of Singapore
Vision-language-action (VLA) models have driven rapid progress in robotic manipulation, demonstrating strong fine-grained control and promising performance on long-horizon tasks. However, many existing VLAs lack explicit access to interaction history, making them vulnerable to perceptual aliasing: similar current observations and robot states at different task stages may induce action ambiguity and lower success rate. Existing methods incorporate temporal or progress cues through feature conditioning, action-prior modification, or sampling guidance. However, methods that jointly fine-tune memory modules and the base VLA incur additional policy-training costs, motivating the separation of trainable history-conditioned steering from frozen base-policy refinement. We propose ActMem-VLA, a dual-expert handover architecture that augments a frozen, fine-tuned VLA with a memory plugin comprising a Mamba-based memory module and a lightweight PreAction Expert (PAE). Specifically, Mamba encodes executed-action history into memory that conditions PAE alongside current context. With these inputs, PAE steers task progression during early, high-noise denoising, then passes the partially denoised action to the frozen Action Expert (AE) to refine action details during the remaining low-noise steps. The fine-tuned base VLA remains frozen throughout training, while only the Mamba module and PAE are jointly optimized. On LIBERO-Mem, ActMem-VLA achieves 80.8% average success across all ten tasks, compared with 65.2% for π0.5 and 49.5% for MemoryVLA, while introducing only 3.45% additional parameters. Across four real-world tasks, it improves the average success rate over π0.5 by 28.8%.
Yaxin Zhao, Dianye Huang, Chenwei Wang +2
Harbin Institute of Technology, Harbin, China. · Medical Intelligence and Robotic Cognition (MIRoC) Lab, Department of Mechanical Engineering, The University of Hong Kong (HKU), Hong Kong SAR, China. · Department of Computing, The Hong Kong Polytechnic University, Hong Kong SAR, China
Vision-language-action (VLA) models provide a unified paradigm for connecting visual perception, language understanding, and robotic control. However, existing VLA models still face major challenges in long-horizon tasks: sparse expert demonstrations constrain cross-task compositional generalization; the non-Markovian nature of long-horizon tasks makes it difficult for policies conditioned only on current observations to maintain temporal consistency; limited closed-loop error correction allows execution errors to accumulate; and end-to-end action fine-tuning may weaken the high-level semantic representations of vision-language model (VLM) backbones. To address these issues, we propose a hierarchical long-horizon VLA architecture with an explicit language-memory module. The central idea is to convert discrete temporal observations into a coherent textual memory sequence with temporal logic. The system is decoupled into a high-level VLM and a low-level VLA: the high-level VLM performs semantic reasoning through a visual question answering training paradigm, while the low-level VLA executes precise continuous control conditioned on subtask instructions and visual observations. The high-level VLM recursively updates both language memory and subtask instructions using the previous memory as a contextual anchor, enabling persistent temporal tracking and dynamic correction during long-horizon execution. We evaluate the proposed method in multiple simulation environments and conduct sim-to-real experiments on a real robotic platform. The results demonstrate that explicit language memory improves the success rate and robustness of VLA models on complex long-horizon tasks while providing an interpretable semantic account of the decision process.