arXiv:2604.19350cs.CV2026-04

用视觉大模型提升乳腺癌筛查准确率,聚焦关键病灶区域。

Attend what matters: Leveraging vision foundational models for breast cancer classification using mammograms

论文配图:Attend what matters: Leveraging vision foundational models for breast cancer classification using mammograms
图 1 · 摘自论文原文
  • 通过目标检测引导,减少高分辨率影像的冗余特征点
  • 结合对比学习增强微小病灶的区分能力,准确率显著提升
  • 适合医学影像分析、临床辅助诊断研究者参考

视觉变压器(ViT)在计算机视觉中表现优异,但在辅助诊断任务中仍受限。针对乳腺钼靶图像中的癌症检测,我们发现两大原因:一是医学图像分辨率高、病灶小,导致特征点过多,softmax注意力难以定位关键区域;二是分类任务细粒度强,类间差异小、类内差异大,标准交叉熵训练效果不足。为此,我们提出三部分框架:(1) 使用目标检测模型引导,基于感兴趣区域(RoI)进行特征点压缩;(2) 采用基于难负样本的对比学习,强化不同病灶间的判别力;(3) 采用预训练的DINOv2 ViT模型,获取具有定位感知和细粒度特征的表示,而非全局的CLIP特征。在公开钼靶数据集上的实验表明,本方法显著优于现有基线,验证了其有效性和临床应用潜力。代码已开源。

原文摘要 · Abstract (English)

Vision Transformers $(\texttt{ViT})$ have become the architecture of choice for many computer vision tasks, yet their performance in computer-aided diagnostics remains limited. Focusing on breast cancer detection from mammograms, we identify two main causes for this shortfall. First, medical images are high-resolution with small abnormalities, leading to an excessive number of tokens and making it difficult for the softmax-based attention to localize and attend to relevant regions. Second, medical image classification is inherently fine-grained, with low inter-class and high intra-class variability, where standard cross-entropy training is insufficient. To overcome these challenges, we propose a framework with three key components: (1) Region of interest $(\texttt{RoI})$ based token reduction using an object detection model to guide attention; (2) contrastive learning between selected $\texttt{RoI}$ to enhance fine-grained discrimination through hard-negative based training; and (3) a $\texttt{DINOv2}$ pretrained $\texttt{ViT}$ that captures localization-aware, fine-grained features instead of global $\texttt{CLIP}$ representations. Experiments on public mammography datasets demonstrate that our method achieves superior performance over existing baselines, establishing its effectiveness and potential clinical utility for large-scale breast cancer screening. Our code is available for reproducibility here: https://aih-iitd.github.io/publications/attend-what-matters

医学影像视觉大模型乳腺癌筛查对比学习

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