Overview
- Field: Computer Vision (medical imaging)
- Authors: Xuan Liu, Derek L. Nguyen, Emily C. Barre
- Published: 2025-07-09
- arXiv: 2507.06826
- Deep learning-based computer-aided diagnosis (CAD) systems perform strongly in breast cancer diagnosis, especially mammography classification, but domain shifts across multi-site datasets remain a major challenge when models are applied to unseen domains.
- The authors propose a calcification classification framework for malignant vs. benign breast disease classification with two components: 1. An unsupervised domain adaptation module based on style transfer models (AdaIN and CycleGAN), generating vendor-specific and technique-specific training samples without additional annotations. 2. A supervised classification module using Swin Transformer V2 as the backbone.
- Cross-validation: OPTIMAM (UK National Health Service; n=2994)
- External validation: EMBED (Emory University; n=125) and Duke Calcification Dataset v1 (n=788)
- Datasets span multiple vendors, including full-field digital mammography and synthetic 2D images derived from digital breast tomosynthesis.
- EMBED: AUC improved from 0.68 to 0.72
- Duke Calcification Dataset: AUC improved from 0.68 to 0.73
Key points
Evaluation
Results
Conclusion
Unsupervised domain adaptation via style transfer reduces domain shift and improves generalization of calcification classification across multi-site mammography datasets.
Original abstract
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 eval...
--- *Auto-collected on 2026-07-09*