arXiv:2605.26621cs.CVcs.AI2026-05

用强化学习让模型先找关键切片再画3D病灶,提升医学影像理解可解释性。

MedVol-R1: Reward-Driven Evidence Grounding for Volumetric Reasoning Segmentation

论文配图:MedVol-R1: Reward-Driven Evidence Grounding for Volumetric Reasoning Segmentation
图 1 · 摘自论文原文
  • 分两步走:先选2D证据切片和框,再转成3D分割掩码
  • 在三个数据集上达到当前最优,比纯监督训练提升明显
  • 无需复杂思维链标注,适合临床实用场景

体积分割推理(VRS)旨在从自由形式的临床提问中分割3D医学扫描中的目标区域,其中指代对象常隐含,需结合医学知识与体积分层推理。现有方法多依赖专用分割标记连接语言与掩码解码,导致决策过程被压缩为不可解释的潜在表征,限制了对多样叙述表达的泛化能力。本文提出MedVol-R1,一种基于强化学习的VRS框架,显式解耦证据定位与体积分割:视觉语言模型将临床推理锚定到可验证的2D证据(关键轴向切片及2D边界框),再由冻结的MedSAM2模块将其传播为一致的3D掩码。通过冷启动监督微调后接GRPO训练,采用多组件奖励机制,鼓励选择信息量高的证据、精准的2D空间定位及跨切片体积分歧一致性,无需昂贵的思维链标注。在M3D-Seg基准的CT-ORG、AbdomenCT-1K和KiTS23上实验表明,MedVol-R1持续优于强基线,达到当前最优性能,强化学习带来显著增益。

原文摘要 · Abstract (English)

Volumetric Reasoning Segmentation (VRS) aims to segment a target region in a 3D medical scan from a free-form clinical query, where the referent is often implicit and requires both medical knowledge and volume-grounded reasoning. Existing methods typically rely on specialized segmentation tokens to connect language with mask decoding, but this coupling collapses the decision process into opaque latent representations, limiting interpretability and generalization to diverse narrative expressions. In this paper, we present MedVol-R1, a reinforcement learning-based framework for VRS that explicitly decouples evidence grounding from volumetric delineation: the LVLM grounds clinical reasoning to a verifiable 2D evidence anchor (key axial slice and 2D bounding boxes), which is then propagated into a coherent 3D mask by a frozen MedSAM2 module. We train MedVol-R1 with cold-start supervised fine-tuning followed by GRPO, guided by a multi-component reward that encourages informative evidence selection, accurate 2D spatial grounding, and cross-slice volumetric coherence, without requiring costly chain-of-thought annotations. Experiments on CT-ORG, AbdomenCT-1K, and KiTS23 from the M3D-Seg benchmark demonstrate that MedVol-R1 consistently outperforms strong baselines and achieves state-of-the-art performance, with reinforcement learning providing clear gains over pure supervised fine-tuning.

医学影像强化学习3D分割可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。