通过无监督域适应提升乳腺钼靶图像中钙化病灶的跨机构分类性能。
Unsupervised Domain Adaptation for Calcification Classification in Mammography Across Multi-Site Datasets

- 利用风格迁移生成不同设备和技术的训练样本,无需额外标注。
- 在三个数据集上验证,外推性能AUC从0.68提升至0.73。
- 适合需要跨机构部署乳腺癌辅助诊断系统的研究者使用。
基于深度学习的计算机辅助诊断(CAD)系统在乳腺癌诊断中表现优异,尤其在乳腺钼靶图像的分类任务中。然而,多中心数据集间的域偏移问题仍是挑战,尤其是在模型应用于未见数据时。本文提出一种钙化分类框架,以提升跨多机构乳腺钼靶数据集的良恶性病变分类性能。该框架包含两部分:(1)基于AdaIN和CycleGAN的无监督域适应模块,生成具有设备与技术特性的训练样本,无需额外标注;(2)采用Swin Transformer V2作为主干的有监督分类模块。在三个数据集上进行评估:在英国国家医疗服务体系(NHS)的OPTIMAM数据集上进行交叉验证(n=2994),随后在埃默里大学的EMBED数据集(n=125)和杜克钙化数据集v1(n=788)上进行外部验证。这些数据集涵盖多种设备厂商,并包括全视野数字乳腺摄影和由数字乳腺断层成像生成的合成2D图像。所提框架在EMBED数据集上将AUC从0.68提升至0.72,在杜克钙化数据集上从0.68提升至0.73。结果表明,域适应可有效缓解域偏移,增强钙化分类的泛化能力。
原文摘要 · Abstract (English)
Deep learning-based computer-aided diagnosis (CAD) systems have shown strong performance in breast cancer diagnosis, particularly for classification tasks in mammography. However, domain shifts across multi-site datasets remain a challenge, especially when models are applied to unseen domains. In this work, we proposed a calcification classification framework to improve malignant versus benign breast disease classification across multi-site mammography datasets. The framework consisted of two components: (1) an unsupervised domain adaptation module based on style transfer models (AdaIN and CycleGAN) to generate vendor-specific and technique-specific training samples without additional annotations, and (2) a supervised classification module using Swin Transformer V2 as the backbone. We evaluated the proposed method on three datasets: cross-validation on OPTIMAM (National Health Service, United Kingdom; n=2994), followed by external validation on EMBED (Emory University; n=125), and Duke Calcification Dataset v1 (n=788). These datasets cover multiple vendors and include both full-field digital mammography and synthetic 2D images derived from digital breast tomosynthesis. The proposed framework improved cross-site performance for both EMBED (AUC 0.68 to 0.72) and the Duke Calcification Dataset (AUC 0.68 to 0.73). These findings indicate that domain adaptation can reduce domain shifts and improve the generalization for calcification classification across multi-site datasets.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。