Dual-Latent Memory Routing for Vision-Language Reasoning
Organizations: School of Artificial Intelligence, Nanjing University, China · National Key Laboratory for Novel Software Technology, Nanjing University, China · Institute of Automation, Chinese Academy of Sciences, Beijing, China
Abstract
Multimodal large language models (MLLMs) have recently made strong progress in vision-language reasoning, yet their performance often degrades as generations grow longer. A key factor is that they frequently lose track of earlier visual evidence and intermediate constraints under a monolithic growing context. Inspired by how humans separately recall what they see and what they infer when solving complex tasks, we propose DLMR, a parameter-efficient mechanism that equips MLLMs with Dual Latent Memories: a visual memory that compresses image evidence and a reasoning memory that tracks intermediate conclusions and constraints. A Router then dynamically decides which memory and how much to reuse during inference, preserving visual grounding while maintaining coherent long-horizon reasoning. DLMR is trained in three stages, from latent memory construction to selective router learning, while keeping the base MLLM frozen, yielding substantial gains on both general and reasoning benchmarks with only a small number of additional trainable parameters. Analyses further show interpretable, state-dependent routing with specialized memory roles and reduced decoding tokens over long generations. Code is available at https://github.com/Hunter-Wrynn/DLMR.
Figures & tables
| Method | General | Reasoning | |||||||
| MMVet | MMStar | RealWorldQA | Avg | MMMU | MathVerse | MathVision | MathVista | Avg | |
| Qwen2.5-VL-7B | |||||||||
| Vanilla | 63.30 | 62.60 | 68.10 | 64.67 | 53.11 | 42.64 | 22.37 | 68.30 | 46.61 |
| CoT | 64.55 | 63.73 | 68.88 | 65.72 | 53.78 | 43.24 | 23.02 | 68.80 | 47.21 |
| CCoT | 64.67 | 64.11 | 69.12 | 65.97 | 53.33 | 43.31 | 24.67 | 67.90 | 47.30 |
| SFT | 64.61 | 62.98 | 69.27 | 65.62 | 54.16 | 43.38 | 23.68 | 68.90 | 47.53 |
| Variant | MMMU | MVerse | MathV | MVista | Avg |
|---|---|---|---|---|---|
| Frozen injector | 58.14 | 44.32 | 27.74 | 69.96 | 50.44 |
| Trainable injector | 61.00 | 46.95 | 35.32 | 72.10 | 53.84 |
| Base Model | General | Reasoning | |||||||
|---|---|---|---|---|---|---|---|---|---|
| MMVet | MMStar | RealWorldQA | Avg | MMMU | MathVerse | MathVision | MathVista | Avg | |
| LLaVA-OV-1.5-4B | 66.0 | 62.6 | 68.1 | 64.6 | 56.0 | 42.6 | 22.3 | 68.3 | 46.6 |
| + DLMR | 75.1 9.1 | 68.9 6.3 | 71.3 3.2 | 71.4 6.8 | 63.9 7.9 | 46.9 4.3 | 35.3 13.0 | 72.1 3.8 | 53.8 7.2 |
| Qwen2.5-VL-3B | 61.2 | 55.1 | 65.3 | 60.6 | 51.2 | 31.2 | 21.9 | 61.2 | 41.4 |
| + DLMR | 69.8 8.6 | 60.6 5.5 | 70.1 4.8 | 66.8 6.2 | 58.9 7.7 | 33.2 2.0 | 36.5 4.6 | 68.6 7.4 | 49.3 7.9 |
| InternVL-3-2B | 65.1 | 61.3 | 64.5 | 63.6 | 48.6 | 24.1 | 21.7 | 54.7 | 37.3 |
| Variant | Avg Acc | Avg token length |
|---|---|---|
| No router, fixed | 51.55 | 664 |
| No router, fixed | 52.71 | 732 |
| No router, fixed | 52.04 | 765 |
| Full DLMR (adaptive ) | 53.84 | 677 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Model & Method | General | Reasoning | ||
|---|---|---|---|---|
| Time (s) | Acc (%) | Time (s) | Acc (%) | |
| Qwen2.5-VL-7B | ||||
| Vanilla | 10.1 | 64.7 | 28.5 | 46.6 |
| SFT | 5.6 | 65.6 | 14.0 | 47.5 |
| DLMR | 5.4 | 71.4 | 11.5 | 53.8 |
| InternVL3-8B | ||||
| Method | LogicVista | OCR-Bench | MMT-Bench |
|---|---|---|---|
| Vanilla | 43.1 | 86.1 | 51.5 |
| CoT | 47.5 | 89.1 | 55.0 |
| SFT | 49.0 | 90.1 | 56.5 |
| GRPO | 50.2 | 88.4 | 58.0 |
| DLMR | 55.0 | 92.0 | 62.5 |
| Configurations | Parameters | Qwen2.5-VL-7B | InternVL3-8B |
|---|---|---|---|
| Text backbone | hidden_size | 3584 | 3584 |
| num_layers | 28 | 28 | |
| num_heads | 28 | 28 | |
| num_kv_heads | 4 | 4 | |
| mlp_intermediate | 18944 | 18944 | |
| max_position_emb | 128000 | 32768 |
| Benchmark | Domain | # Eval | Answer fmt. | Metric |
| General (vision–language) | ||||
| MMVet ( Yu et al., 2023 ) | Composite VL skills | 218 | Open-ended | LLM-judge acc. |
| MMStar ( Chen et al., 2024 ) | Vision-indispensable VL | 1,500 | MCQ | Accuracy |
| RealWorldQA | Real-world VQA | 765 | MCQ / short ans. | Accuracy |
| Reasoning (vision–language) | ||||
| MMMU ( Yue et al., 2024 ) | College-level multidisciplinary | 900 (val) | MCQ / open | Accuracy |
| Configurations | Parameters | Stage 1 | Stage 2 | Stage 3 |
|---|---|---|---|---|
| Core | backbone | Qwen2.5-VL-7B / InternVL3-8B | Qwen2.5-VL-7B / InternVL3-8B | Qwen2.5-VL-7B / InternVL3-8B |
| visual_latents_len | 8 | 8 | – | |
| reasoning_latents_len | 16 | 16 | – | |
| max_visual_aug_num | 1 | 1 | 1 | |
| max_reasoning_aug_num | 5 | 5 | 5 | |
| random_reasoning_aug | True | – | – |
Memory-Augmented VLMs
Memory-augmented vision-language models use persistent or retrieved context to improve reasoning across long inputs and interactions.
See all 151 papers