English

BreastMammo and DenseMammo: Benchmarks for Mammography Domain Generalization

Image and Video Processing 2026-08-10 v1 Machine Learning

Abstract

Breast density classification is a critical component of breast cancer risk assessment, yet AI models often struggle to generalize across clinical sites due to vendor-specific acquisition styles. In this work, we introduce two new datasets, BreastMammo and DenseMammo, to facilitate robust multi-view mammography research. We propose a domain generalization framework that utilizes a foreground-only histogram matching protocol to resolve the domain shift issue arising from disparate clinical sources. Internal evaluation using a 5-fold cross-validation protocol demonstrates the efficacy of our approach, with the Swin Transformer backbone achieving a peak AUC of 98.32% for density classification. External evaluation on the TNMammo and LUMINA datasets demonstrates that the proposed approach consistently reduces domain shift, significantly outperforming prominent domain generalization paradigms, including MixStyle and Discrete-Fourier-Transform-based frameworks.

Keywords

Cite

@article{arxiv.2608.10271,
  title  = {BreastMammo and DenseMammo: Benchmarks for Mammography Domain Generalization},
  author = {Hongyi Pan and Gorkem Durak and Halil Ertugrul Aktas and Andrea Mia Bejar and Mustafa Ege Seker and Nebile Alibeyoglu and Rumeysa Guclu and Rana Gunoz Comert Bozkurt and Sibel Ozkan Gurdal and Neslihan Cabioglu and Beyza Ozcinar and Ravza Yilmaz and Vahit Ozmen and Erkin Aribal and Sukru Mehmet Erturk and Yalda Zafari and Mohamed Mabrok and Kayhan Batmanghelich and Mohammad Yaqub and Ziyue Xu and Ulas Bagci},
  journal= {arXiv preprint arXiv:2608.10271},
  year   = {2026}
}

Comments

This paper was accepted to the MICCAI 2026 workshop Deep-Brea3th