Improving the generalization of deep learning models in the segmentation of mammography images

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Hurtado, Jan, Maia, Joao P., Sierra-Franco, Cesar A., Raposo, Alberto
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916664672518144
author Hurtado, Jan
Maia, Joao P.
Sierra-Franco, Cesar A.
Raposo, Alberto
author_facet Hurtado, Jan
Maia, Joao P.
Sierra-Franco, Cesar A.
Raposo, Alberto
contents Mammography stands as the main screening method for detecting breast cancer early, enhancing treatment success rates. The segmentation of landmark structures in mammography images can aid the medical assessment in the evaluation of cancer risk and the image acquisition adequacy. We introduce a series of data-centric strategies aimed at enriching the training data for deep learning-based segmentation of landmark structures. Our approach involves augmenting the training samples through annotation-guided image intensity manipulation and style transfer to achieve better generalization than standard training procedures. These augmentations are applied in a balanced manner to ensure the model learns to process a diverse range of images generated by different vendor equipments while retaining its efficacy on the original data. We present extensive numerical and visual results that demonstrate the superior generalization capabilities of our methods when compared to the standard training. For this evaluation, we consider a large dataset that includes mammography images generated by different vendor equipments. Further, we present complementary results that show both the strengths and limitations of our methods across various scenarios. The accuracy and robustness demonstrated in the experiments suggest that our method is well-suited for integration into clinical practice.
format Preprint
id arxiv_https___arxiv_org_abs_2503_22052
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Improving the generalization of deep learning models in the segmentation of mammography images
Hurtado, Jan
Maia, Joao P.
Sierra-Franco, Cesar A.
Raposo, Alberto
Image and Video Processing
Computer Vision and Pattern Recognition
Mammography stands as the main screening method for detecting breast cancer early, enhancing treatment success rates. The segmentation of landmark structures in mammography images can aid the medical assessment in the evaluation of cancer risk and the image acquisition adequacy. We introduce a series of data-centric strategies aimed at enriching the training data for deep learning-based segmentation of landmark structures. Our approach involves augmenting the training samples through annotation-guided image intensity manipulation and style transfer to achieve better generalization than standard training procedures. These augmentations are applied in a balanced manner to ensure the model learns to process a diverse range of images generated by different vendor equipments while retaining its efficacy on the original data. We present extensive numerical and visual results that demonstrate the superior generalization capabilities of our methods when compared to the standard training. For this evaluation, we consider a large dataset that includes mammography images generated by different vendor equipments. Further, we present complementary results that show both the strengths and limitations of our methods across various scenarios. The accuracy and robustness demonstrated in the experiments suggest that our method is well-suited for integration into clinical practice.
title Improving the generalization of deep learning models in the segmentation of mammography images
topic Image and Video Processing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.22052