arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2509.12512eess.IVcs.AIcs.CV

DinoAtten3D:用于3D脑部MRI异常分类的DinoV2切片级注意力聚合

DinoAtten3D: Slice-Level Attention Aggregation of DinoV2 for 3D Brain MRI Anomaly Classification

Arizona State University(亚利桑那州立大学);Mayo Clinic, Arizona(梅奥诊所(亚利桑那))

Fazle Rafsani, Jay Shah, Catherine D. Chong, Todd J. Schwedt, Teresa Wu

AI总结:

针对医学影像标注数据有限和类别不平衡的挑战,提出DinoAtten3D框架,利用预训练DINOv2提取2D切片特征并结合软注意力机制进行3D脑部MRI异常分类,采用复合损失函数增强特征区分度,在有限数据下实现了鲁棒的异常检测性能。

摘要:

医学图像中的异常检测和分类对于早期诊断至关重要,但由于标注数据有限、类别不平衡以及专家标注成本高昂,该任务仍具挑战性。在大量无标签数据集上预训练的新兴视觉基础模型(如DINOv2)提供了通用的特征表示,有望缓解这些限制。本研究提出一种专为3D医学图像异常分类定制的基于注意力的全局聚合框架。该方法利用自监督的DINOv2模型作为预训练特征提取器,处理脑部MRI的各个2D轴向切片,并通过软注意力机制分配自适应的切片级重要性权重。为进一步解决数据稀缺问题,我们采用结合监督对比学习与类方差正则化的复合损失函数,以增强类间可分性和类内一致性。我们在ADNI数据集和一个机构多类头痛队列上验证了该框架,结果表明,尽管数据可用性有限且存在显著的类别不平衡,该框架仍展现出强大的异常分类性能。研究结果突显了将预训练的2D基础模型与基于注意力的切片聚合相结合,在医学影像中进行鲁棒体积异常检测的有效性。我们的实现已在 https://github.com/Rafsani/DinoAtten3D.git 上公开。

英文摘要:

Anomaly detection and classification in medical imaging are critical for early diagnosis but remain challenging due to limited annotated data, class imbalance, and the high cost of expert labeling. Emerging vision foundation models such as DINOv2, pretrained on extensive, unlabeled datasets, offer generalized representations that can potentially alleviate these limitations. In this study, we propose an attention-based global aggregation framework tailored specifically for 3D medical image anomaly classification. Leveraging the self-supervised DINOv2 model as a pretrained feature extractor, our method processes individual 2D axial slices of brain MRIs, assigning adaptive slice-level importance weights through a soft attention mechanism. To further address data scarcity, we employ a composite loss function combining supervised contrastive learning with class-variance regularization, enhancing inter-class separability and intra-class consistency. We validate our framework on the ADNI dataset and an institutional multi-class headache cohort, demonstrating strong anomaly classification performance despite limited data availability and significant class imbalance. Our results highlight the efficacy of utilizing pretrained 2D foundation models combined with attention-based slice aggregation for robust volumetric anomaly detection in medical imaging. Our implementation is publicly available at https://github.com/Rafsani/DinoAtten3D.git.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑