Imagination Helps Visual Reasoning, But Not Yet in Latent Space

You Li, Chi Chen, Yanghao Li, Fanhu Zeng, Kaiyu Huang +2 more
2/26/2026
cs.CL

No official code found. Best implementation: AI9Stars/CapImagine (29 stars, Apache-2.0, last commit 2026-02-27). 1 more below.

Abstract

Latent visual reasoning aims to mimic human's imagination process by meditating through hidden states of Multimodal Large Language Models. While recognized as a promising paradigm for visual reasoning, the underlying mechanisms driving its effectiveness remain unclear. Motivated to demystify the true source of its efficacy, we investigate the validity of latent reasoning using Causal Mediation Analysis. We model the process as a causal chain: the input as the treatment, the latent tokens as the mediator, and the final answer as the outcome. Our findings uncover two critical disconnections: (a) Input-Latent Disconnect: dramatic perturbations on the input result in negligible changes to the latent tokens, suggesting that latent tokens do not effectively attend to the input sequence. (b) Latent-Answer Disconnect: perturbations on the latent tokens yield minimal impact on the final answer, indicating the limited causal effect latent tokens imposing on the outcome. Furthermore, extensive probing analysis reveals that latent tokens encode limited visual information and exhibit high similarity. Consequently, we challenge the necessity of latent reasoning and propose a straightforward alternative named CapImagine, which teaches the model to explicitly imagine using text. Experiments on vision-centric benchmarks show that CapImagine significantly outperforms complex latent-space baselines, highlighting the superior potential of visual reasoning through explicit imagination.

View on arXivView PDF

Code Implementations(2)

Apache-2.0

Official Implementation of "Imagination Helps Visual Reasoning, But Not Yet in Latent Space"

290Feb 27, 20267 months agoApache-2.0
Apache-2.0

Official Implementation of "Imagination Helps Visual Reasoning, But Not Yet in Latent Space"

40Feb 25, 20267 months agoApache-2.0

Similar Papers

Cite this paper

@article{li2026imagination,
  title  = {Imagination Helps Visual Reasoning, But Not Yet in Latent Space},
  author = {You Li and Chi Chen and Yanghao Li and Fanhu Zeng and Kaiyu Huang and Jinan Xu and Maosong Sun},
  year   = {2026},
  eprint = {2602.22766},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.22766v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.22766.svg)](https://codeofpaper.com/paper/2602.22766)

Discussion