Token-Sparse Medical Multimodal Reasoning via Dual-Stream Reinforcement Learning

Kaitao Chen, Weiqian Zhao, Jiamin Wu, Qihao Zheng, Shangquan Sun +4 more
6/30/2026
cs.CVcs.AI

No official code found. Best implementation: JLINEkai/ViTos (1 star, last commit 2026-05-26).

Abstract

Vision-language models (VLMs) combining reinforcement learning (RL) ignite remarkable progress in multimodal reasoning, yet still struggle with medical images, which typically exhibit extremely sparse visual evidence to inform clinical decision-making. We recognize that pruning visual tokens outside the grounding region greatly enhances medical reasoning. However, a united RL framework for active visual token pruning (VTP) and medical multimodal reasoning remains unestablished. Here, we propose a dual-stream RL framework, ViToS, to fulfill token pruning and question answering. ViToS trains one policy model with two task branches, where one focuses on grounding while the other conducts token-sparse reasoning after VTP. Furthermore, we solve the coupled policy learning problem by introducing the cross-feedback sequential optimization, avoiding gradient conflict and facilitating convergence of the shared policy model. Evaluated on seven medical benchmarks, our method reduces visual tokens to 77% of the original sequence length while achieving a 108.27% relative performance on Lingshu-7B and 104.16% relative performance on HuatuoGPT-Vision-7B. Overall, ViToS delivers superior performance and inference speedup, establishing an efficient paradigm for medical multimodal reasoning.

View on arXivView PDF

Code Implementations(1)

JLINEkai/ViTosCommunity59%

Token-Sparse Medical Multimodal Reasoning via Dual-Stream Reinforcement Learning (ICML 26)

10May 26, 20264 months ago

Similar Papers

Cite this paper

@article{chen2026tokensparse,
  title  = {Token-Sparse Medical Multimodal Reasoning via Dual-Stream Reinforcement Learning},
  author = {Kaitao Chen and Weiqian Zhao and Jiamin Wu and Qihao Zheng and Shangquan Sun and Chunfeng Song and Xiaosong Wang and Mu Zhou and Mianxin Liu},
  year   = {2026},
  eprint = {2606.31599},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2606.31599v1}
}

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

Discussion