Domain generalization in deep learning-based mass detection in mammography: A large-scale multi-center study
Computer-aided detection systems based on deep learning have shown great potential in breast cancer detection. However, the lack of domain generalization of artificial neural networks is an important obstacle to their deployment in changing clinical environments. In this work, we explore the domain generalization of deep learning methods for mass detection in digital mammography and analyze in-depth the sources of domain shift in a large-scale multi-center setting. To this end, we compare the performance of eight state-of-the-art detection methods, including Transformer-based models, trained in a single domain and tested in five unseen domains. Moreover, a single-source mass detection training pipeline is designed to improve the domain generalization without requiring images from the new domain. The results show that our workflow generalizes better than state-of-the-art transfer learning-based approaches in four out of five domains while reducing the domain shift caused by the different acquisition protocols and scanner manufacturers. Subsequently, an extensive analysis is performed to identify the covariate shifts with bigger effects on the detection performance, such as due to differences in patient age, breast density, mass size, and mass malignancy. Ultimately, this comprehensive study provides key insights and best practices for future research on domain generalization in deep learning-based breast cancer detection.
Code (0)
등록된 구현이 없습니다.
Tasks
Breast Cancer DetectionDeep LearningDomain GeneralizationTransfer LearningSimilar Papers 제목 키워드 기반
High-resolution synthesis of high-density breast mammograms: Application to improved fairness in deep learning based mass detection
Computer-aided detection systems based on deep learning have shown good performance in breast cancer detection. However, high-density breasts show poorer detection performance since dense tissues can mask or even simulat…
Breast Cancer DetectionData AugmentationDomain AdaptationDomain Generalization+3Improving Mass Detection in Mammography Images: A Study of Weakly Supervised Learning and Class Activation Map Methods
In recent years, weakly supervised models have aided in mass detection using mammography images, decreasing the need for pixel-level annotations. However, most existing models in the literature rely on Class Activation M…
Weakly-supervised LearningMammography Dual View Mass Correspondence
Standard breast cancer screening involves the acquisition of two mammography X-ray projections for each breast. Typically, a comparison of both views supports the challenging task of tumor detection and localization. We …
Template MatchingTransfer LearningDomain Generalization for Mammographic Image Analysis with Contrastive Learning
The deep learning technique has been shown to be effectively addressed several image analysis tasks in the computer-aided diagnosis scheme for mammography. The training of an efficacious deep learning model requires larg…
breast density classificationContrastive LearningDeep LearningDiversity+3In-context Cross-Density Adaptation on Noisy Mammogram Abnormalities Detection
This paper investigates the impact of breast density distribution on the generalization performance of deep-learning models on mammography images using the VinDr-Mammo dataset. We explore the use of domain adaptation tec…
Deep LearningDomain Adaptationobject-detectionObject Detection