Think Twice to See More: Iterative Visual Reasoning in Medical VLMs

Kaitao Chen, Shaohao Rui, Yankai Jiang, Jiamin Wu, Qihao Zheng +4 more
10/11/2025
cs.CVcs.AI

No official code found. Best implementation: JLINEkai/ViTAR (16 stars, Apache-2.0, last commit 2025-09-30).

Abstract

Medical vision-language models (VLMs) excel at image-text understanding but typically rely on a single-pass reasoning that neglects localized visual cues. In clinical practice, however, human experts iteratively scan, focus, and refine the regions of interest before reaching a final diagnosis. To narrow this machine-human perception gap, we introduce ViTAR, a novel VLM framework that emulates the iterative reasoning process of human experts through a cognitive chain of "think-act-rethink-answer". ViTAR treats medical images as interactive objects, enabling models to engage multi-step visual reasoning. To support this approach, we curate a high-quality instruction dataset comprising 1K interactive examples that encode expert-like diagnostic behaviors. In addition, a 16K visual question answering training data has been curated towards fine-grained visual diagnosis. We introduce a two-stage training strategy that begins with supervised fine-tuning to guide cognitive trajectories, followed by the reinforcement learning to optimize decision-making. Extensive evaluations demonstrate that ViTAR outperforms strong state-of-the-art models. Visual attention analysis reveals that from the "think" to "rethink" rounds, ViTAR increasingly anchors visual grounding to clinically critical regions and maintains high attention allocation to visual tokens during reasoning, providing mechanistic insight into its improved performance. These findings demonstrate that embedding expert-style iterative thinking chains into VLMs enhances both performance and trustworthiness of medical AI.

View on arXivView PDF

Code Implementations(1)

JLINEkai/ViTARCommunity75%
Apache-2.0

Think Twice to See More: Iterative Visual Reasoning in Medical VLMs

162Shell, PythonSep 24, 20251 years agoApache-2.0

Similar Papers

Cite this paper

@article{chen2025think,
  title  = {Think Twice to See More: Iterative Visual Reasoning in Medical VLMs},
  author = {Kaitao Chen and Shaohao Rui and Yankai Jiang and Jiamin Wu and Qihao Zheng and Chunfeng Song and Xiaosong Wang and Mu Zhou and Mianxin Liu},
  year   = {2025},
  eprint = {2510.10052},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2510.10052v1}
}

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/2510.10052.svg)](https://codeofpaper.com/paper/2510.10052)

Discussion