Cross-Dataset Generalization in Breast MRI Tumor Classification via Class-Wise Dataset Mixing
Not on the map. Why?
Abstract (arXiv)
Breast MRI is highly sensitive for detecting breast tumors, but exams contain many slices and require substantial reading time. Deep learning models often perform well on internal splits but can fail across institutions because of domain shift and dataset-origin bias. We study this failure mode for binary breast MRI tumor classification. EfficientNet-B3 and WaveViT-Small are trained using Duke Breast Cancer MRI and fastMRI, and evaluated only on the independent multi-center MAMA-MIA cohort. In a deliberately confounded setup, where label is perfectly correlated with dataset origin, external accuracy is near chance (0.5048--0.5265), despite very high recall. We then construct a mixed training set in which each class contains samples from both Duke and fastMRI, while preserving patient-level splitting, augmentation, and leakage controls. On MAMA-MIA, dataset mixing improves accuracy/F1 to 0.8463/0.8625 for WaveViT-Small and 0.8884/0.8994 for EfficientNet-B3. These results show that controlling dataset-origin bias is important for reliable breast MRI classification.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 6 of its 15 references that are in the map. First posted on arXiv 21 Jul 2026.
Builds on
6 of 15 references.
References · 6
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
- 2018fastMRI: An Open Dataset and Benchmarks for Accelerated MRIcited by 470
- 2018Confounding variables can degrade generalization performance of radiological deep learning modelscited by 379
- 2022Wave-ViT: Unifying Wavelet and Transformers for Visual Representation Learningcited by 78
- 2024A large-scale multicenter breast cancer DCE-MRI benchmark dataset with expert segmentationscited by 40
- 2024fastMRI Breast: A publicly available radial k-space dataset of breast dynamic contrast-enhanced MRIcited by 7
Built on by
Not cited yet.