用原型特征提升乳腺癌影像病变检测的通用性与鲁棒性
Exemplar Med-DETR: Toward Generalized and Robust Lesion Detection in Mammogram Images and beyond
- 基于跨模态对比学习,用原型特征引导病变定位
- 在越南密集乳腺数据上达0.7 mAP(质量提升16个百分点)
- 适用于多种医学影像,适合临床部署与跨域应用
医学影像中的异常检测因特征表达差异和解剖结构与病灶间的复杂关系而面临挑战,尤其在乳腺钼靶中,致密乳腺组织会遮蔽病灶,增加放射科医生解读难度。现有方法虽利用解剖和语义上下文,但难以有效学习类别特异性特征,限制了其在不同任务和成像模态中的泛化能力。本文提出Exemplar Med-DETR,一种新型多模态对比检测器,支持基于特征的检测。该模型采用自生成的直观类别原型特征,并通过迭代训练策略优化。在四个公开数据集的三种不同成像模态上均取得当前最优性能:在越南密集乳腺钼靶数据上,肿块检测mAP达0.7,钙化检测达0.55,分别提升16个百分点;对来自中国外分布队列的100张乳腺图像进行放射科医生评估,检出率提升两倍。对于胸部X光和血管造影,肿块检测和狭窄检测的mAP分别为0.25和0.37,分别提升4和7个百分点。结果表明该方法可显著提升医学影像中稳健且通用的检测系统性能。
原文摘要 · Abstract (English)
Detecting abnormalities in medical images poses unique challenges due to differences in feature representations and the intricate relationship between anatomical structures and abnormalities. This is especially evident in mammography, where dense breast tissue can obscure lesions, complicating radiological interpretation. Despite leveraging anatomical and semantic context, existing detection methods struggle to learn effective class-specific features, limiting their applicability across different tasks and imaging modalities. In this work, we introduce Exemplar Med-DETR, a novel multi-modal contrastive detector that enables feature-based detection. It employs cross-attention with inherently derived, intuitive class-specific exemplar features and is trained with an iterative strategy. We achieve state-of-the-art performance across three distinct imaging modalities from four public datasets. On Vietnamese dense breast mammograms, we attain an mAP of 0.7 for mass detection and 0.55 for calcifications, yielding an absolute improvement of 16 percentage points. Additionally, a radiologist-supported evaluation of 100 mammograms from an out-of-distribution Chinese cohort demonstrates a twofold gain in lesion detection performance. For chest X-rays and angiography, we achieve an mAP of 0.25 for mass and 0.37 for stenosis detection, improving results by 4 and 7 percentage points, respectively. These results highlight the potential of our approach to advance robust and generalizable detection systems for medical imaging.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。