arXiv:2512.21414cs.CVcs.LG2025-12

用神经网络融合医学影像工具输出,提升模型可解释性与准确率

A Tool Bottleneck Framework for Clinically-Informed and Interpretable Medical Image Understanding

论文配图:A Tool Bottleneck Framework for Clinically-Informed and Interpretable Medical Image Understanding
图 1 · 摘自论文原文
  • 通过学习的瓶颈模型融合多工具特征,替代文本组合方式
  • 在病理与皮肤科任务中达到或超越现有模型性能,尤其数据少时优势明显
  • 适合需要临床可解释性的医疗影像分析场景

基于视觉语言模型(VLM)的工具使用框架通过调用专用工具来提升图像理解能力。现有方法通常依赖文本指令组合工具,但在医学影像中,关键信息为局部空间特征,难以仅通过文本融合。为此,我们提出工具瓶颈框架(TBF),利用预训练医学VLM从工具箱中选择提取临床相关特征的工具,并通过学习的工具瓶颈模型(TBM)以神经网络方式融合工具输出,生成最终预测。TBF支持任意VLM的工具选择策略,无需修改架构。在组织病理学和皮肤病学任务上验证表明,该框架在数据有限条件下表现优异,性能达到或超过深度学习分类器、VLM及先进工具使用框架,同时具备更强的临床可解释性。项目详情与代码见https://christinaliu2020.github.io/tbm/

原文摘要 · Abstract (English)

Recent tool-use frameworks powered by vision-language models (VLMs) improve image understanding by grounding model predictions with specialized tools. Broadly, these frameworks leverage VLMs and a pre-specified toolbox to decompose the prediction task into multiple tool calls (often deep learning models) which are composed to make a prediction. The dominant approach to composing tools is using text, via function calls embedded in VLM-generated code or natural language. However, these methods often perform poorly on medical image understanding, where salient information is encoded as spatially-localized features that are difficult to compose or fuse via text alone. To address this, we propose a tool-use framework for medical image understanding called the Tool Bottleneck Framework (TBF), which composes VLM-selected tools using a learned Tool Bottleneck Model (TBM). For a given image and task, TBF leverages an off-the-shelf medical VLM to select tools from a toolbox that each extract clinically-relevant features. Instead of text-based composition, these tools are composed by the TBM, which computes and fuses the tool outputs using a neural network before outputting the final prediction. We propose a simple and effective strategy for TBMs to make predictions with any arbitrary VLM tool selection. Overall, our framework not only improves tool-use in medical imaging contexts, but also yields more interpretable, clinically-grounded predictors. We evaluate TBF on tasks in histopathology and dermatology and find that these advantages enable our framework to perform on par with or better than deep learning-based classifiers, VLMs, and state-of-the-art tool-use frameworks, with particular gains in data-limited regimes. The project details and the code are available at https://christinaliu2020.github.io/tbm/.

医学影像工具使用可解释性视觉语言模型

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