针对乳腺影像中图像分辨率高、报告同质化问题,提出新模型提升零样本诊断性能。
Solving the Needle-in-a-Haystack Problem in Mammography Vision-Language Model with Differentiable Subset Sampling

- 通过可微分采样高分辨率病变区域,避免降采样损失信息
- 在多个临床任务上超越现有开源模型,零样本准确率显著提升
- 适合医学影像研究者和希望提升模型定位能力的开发者
将CLIP风格视觉语言模型(VLM)预训练应用于乳腺影像正受到越来越多关注。然而,直接使用标准CLIP架构与训练目标的模型,在癌症、发现类型及BI-RADS分类等关键临床任务上的零样本表现有限。我们认为这源于忽视了乳腺影像数据的两个特性:(1) 高分辨率图像,(2) 报告内容高度同质化,主要由检查中大量阴性/良性发现导致。为此,我们提出TopKSigLIP,一种新型视觉语言模型,通过创新架构与学习目标解决上述问题。不将高分辨率乳腺影像降采样以满足显存限制,而是引入TopK-Patch模块,学习从原始图像中采样可能包含病灶的稀疏高分辨率补丁,突破了分辨率与批量大小之间的权衡。所采补丁位置还具备内置定位功能。为应对报告同质化问题,将对比损失替换为基于结构化数据生成软标签的Sup-sigmoid损失,该损失扩展自SigLIP的sigmoid损失。在内部与外部基准测试中,TopKSigLIP在密度评估、BI-RADS分类、发现亚型分类及癌症预测等任务上均优于现有开源乳腺影像与通用医疗VLM。即使使用更小的视觉编码器与更小训练批次,其在线性探测下仍具竞争力。此外,TopK-Patch模块在病灶定位上优于事后Grad-CAM。代码与权重已公开:https://github.com/Youngseok0001/TopKSigLIP。
原文摘要 · Abstract (English)
There is growing interest in adopting CLIP-style vision--language model (VLM) pretraining for mammography. However, models that directly employ the standard CLIP architecture and training objective exhibit limited zero-shot performance in clinically important tasks such as cancer, finding-type, and BI-RADS predictions. We argue that this underwhelming performance is due to neglecting two characteristics of mammography data: (1) its high-res nature, and (2) homogeneity of radiology reports, largely driven by a predominance of negative/benign findings on examinations. We propose TopKSigLIP, a VLM designed to address these two limitations through a novel architecture and learning objectives. Instead of downscaling high-res mammography images to satisfy GPU memory constraints, TopKSigLIP introduces TopK-Patch module that learns to sample a sparse set of high-res patches likely to contain lesions, sidestepping the resolution--batch size tradeoff of VLM training. The sampled patch locations additionally serve as a built-in localization tool. To address report homogeneity, we replace the contrastive loss, which falsely repels semantically similar pairs, with a Sup-sigmoid loss. Sup-sigmoid loss extends the sigmoid loss from SigLIP with soft labels derived from structured data. TopKSigLIP outperforms existing open-source mammography and general medical VLMs on both internal and external benchmarks on density assessment, BI-RADS classification, finding subtyping, and cancer prediction under zero-shot evaluation. TopKSigLIP remains competitive under linear probing despite using a significantly smaller vision encoder and smaller training batches than baselines. The TopK-Patch module additionally achieves superior lesion localization over post-hoc Grad-CAM. Code and weights are made public:https://github.com/Youngseok0001/TopKSigLIP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。