LUMINA: A Multi-Vendor Mammography Benchmark with Energy Harmonization Protocol
Organizations: Department of Radiology, Northwestern University, Chicago, IL, USA · Department of Radiation Oncology, University of Health Sciences Prof. Dr. Cemil Tascioglu City Hospital, Istanbul, Turkey · Department of Radiology, Istanbul University, Istanbul, Turkey
Abstract
Publicly available full-field digital mammography (FFDM) datasets remain limited in size, clinical annotations, and vendor diversity, hindering the development of robust models. We introduce LUMINA, a curated, multi-vendor FFDM dataset that explicitly encodes acquisition energy and vendor metadata to capture clinically relevant appearance variations often overlooked in existing benchmarks. This dataset contains 1824 images from 468 patients (960 benign, 864 malignant), with pathology-confirmed labels, BI-RADS assessments, and breast-density annotations. LUMINA spans six acquisition systems and includes both high- and low-energy imaging styles, enabling systematic analysis of vendor- and energy-induced domain shifts. To address these variations, we propose a foreground-only pixel-space alignment method (''energy harmonization'') that maps images to a low-energy reference while preserving lesion morphology. We benchmark CNN and transformer models on three clinically relevant tasks: diagnosis (benign vs. malignant), BI-RADS classification, and density estimation. Two-view models consistently outperform single-view models. EfficientNet-B0 achieves an AUC of 93.54% for diagnosis, while Swin-T achieves the best macro-AUC of 89.43% for density prediction. Harmonization improves performance across architectures and produces more localized Grad-CAM responses. Overall, LUMINA provides (1) a vendor-diverse benchmark and (2) a model-agnostic harmonization framework for reliable and deployable mammography AI.
Figures & tables
| Pathology | BI-RADS | Breast | Acquisition | ||||
| Dataset | Origin | Patients | Images | Labels | Assessment | Density | Technique |
| MIAS [ 30 ] | UK | 161 | 322 | Yes | No | No | SFM |
| DDSM [ 4 ] | USA | 2,620 | 10,480 | Yes | Yes | Yes | SFM |
| CBIS-DDSM [ 14 ] | USA | 2,620 | 10,239 | Yes | Yes | Yes | SFM |
| INbreast [ 19 ] | Portugal | 115 | 410 | No | Yes | Yes | FFDM |
| VinDR-Mammo [ 22 ] | Vietnam | 5,000 | 20,000 | No | Yes | Yes | FFDM |
| Vendor | # Patients | # Images | Energy |
|---|---|---|---|
| IMS | 341 | 1326 | High |
| Metaltronica | 91 | 354 | Low |
| FUJIFILM Corporation | 29 | 116 | Low |
| SIEMENS | 4 | 16 | Low |
| Carestream Health | 1 | 4 | Low |
| GE MEDICAL SYSTEMS | 2 | 8 | High |
| View | Benign | Malignant | Total |
|---|---|---|---|
| Single | 592 (592) | 344 (344) | 936 (936) |
| Two | 296 (592) | 172 (344) | 468 (936) |
| View | Model | Input | Params | Flops | ACC(%) | AUC(%) | Precision(%) | Recall(%) | F1(%) | Specificity(%) |
|---|---|---|---|---|---|---|---|---|---|---|
| Single | ResNet-50 | 23.51M | 4.13G | 79.49 5.99 | 87.42 4.21 | 81.29 10.90 | 61.97 18.17 | 67.52 13.09 | 89.68 9.63 | |
| DenseNet-121 | 6.95M | 2.90G | 79.49 3.37 | 87.14 3.08 | 80.45 9.45 | 62.55 17.56 | 67.79 9.57 | 89.38 7.52 | ||
| EfficientNet-B0 | 4.01M | 0.41G | 83.12 4.13 | 90.66 4.21 | 79.78 6.31 | 73.89 15.77 | 75.50 7.95 | 88.51 5.00 | ||
| Swin-T | 27.52M | 3.13G | 81.84 5.92 | 88.62 5.18 | 78.27 8.63 | 71.83 12.10 | 74.16 8.36 | 87.66 7.45 | ||
| ResNet-50 | 23.51M | 21.58G | 77.68 7.18 | 86.69 4.37 | 82.79 14.09 | 59.36 28.73 | 61.41 22.78 | 88.38 12.29 | ||
| DenseNet-121 | 6.95M | 15.14G | 79.38 5.55 | 90.52 3.61 | 81.32 16.39 | 68.09 21.96 | 69.22 11.98 | 85.99 14.42 |
| Two-Class (BI-RADS 1/2/3 VS 4/5/6) | Three-Class (BI-RADS 1/2 VS 3/4 VS 5/6) | ||||||
|---|---|---|---|---|---|---|---|
| Model | Input | ACC(%) | AUC(%) | F1(%) | ACC(%) | AUC(%) | F1(%) |
| ResNet-50 | 71.98 12.34 | 88.23 3.52 | 62.45 10.33 | 69.02 5.14 | 79.58 3.00 | 55.36 1.97 | |
| DenseNet-121 | 77.55 3.67 | 87.89 3.52 | 58.98 14.70 | 68.15 3.12 | 80.90 3.74 | 48.63 3.49 | |
| EfficientNet-B0 | 83.56 4.76 | 92.80 4.13 | 74.18 9.32 | 71.78 5.03 | 83.27 4.91 | 55.15 4.65 | |
| Swin-T | 79.29 9.01 | 90.80 4.61 | 60.82 31.02 | 70.51 4.45 | 81.69 5.55 | 49.55 6.85 | |
| ResNet-50 | 75.22 4.46 | 89.08 5.42 | 64.78 9.96 | 66.66 3.10 | 78.92 4.91 | 50.71 3.73 | |
| Model | Input | ACC(%) | AUC(%) | F1(%) |
|---|---|---|---|---|
| ResNet-50 | 64.74 3.33 | 85.66 1.62 | 56.94 6.43 | |
| DenseNet-121 | 66.29 4.20 | 86.39 2.92 | 56.70 2.97 | |
| EfficientNet-B0 | 59.43 5.23 | 84.72 3.54 | 48.25 5.68 | |
| Swin-T | 70.35 4.60 | 89.43 2.22 | 65.28 6.48 | |
| ResNet-50 | 65.37 4.07 | 87.06 2.10 | 58.29 7.57 | |
| DenseNet-121 | 66.31 5.31 | 86.60 3.76 | 61.42 7.62 |
| Backbone | Input | Params | ACC(%) | AUC(%) | F1(%) |
|---|---|---|---|---|---|
| Indepen- | 8.34M | 84.82 3.00 | 93.54 2.38 | 77.41 6.13 | |
| dent | 8.34M | 88.46 4.91 | 92.71 4.63 | 82.19 8.61 | |
| Shared | 4.34M | 86.11 2.47 | 92.99 3.04 | 79.39 4.18 | |
| 4.34M | 85.04 5.92 | 93.54 3.88 | 75.75 12.85 |
| Family | Examples | Labels/Supervision | Architecture/Objective | Limitations vs. LUMINA |
|---|---|---|---|---|
| Film (SFM) datasets | MIAS [ 30 ] , DDSM [ 4 ] , CBIS-DDSM [ 14 ] | Pathology (yes), BI-RADS (varies) | Early CNNs / CAD; single-task classification | Film scans, lower resolution; limited modern FFDM relevance. |
| Digital (FFDM) datasets | INbreast [ 19 ] , VinDR-Mammo [ 22 ] , RSNA [ 6 ] , CMMD [ 5 ] , KAU-BCMD [ 1 ] | Pathology or BI-RADS; density varies | CNNs/ViTs; single- or dual-task | Often single vendor/system; partial labels; limited multi-task scope. |
| Harmonization (feature space) | ComBat [ 21 ] | Batch-effect correction on features | Empirical Bayes (location/scale) on radiomics/latent features | Requires feature extraction; not pixel-space; modality-agnostic assumptions. |
| Harmonization (federated learning) | HarmoFL [ 12 ] | Federated frequency‑domain drift normalization | Joint optimization across clients | Heavy infra; task/model coupling; not a simple preproc. |
| LUMINA (Ours) | Multi-vendor FFDM (6 systems); energy metadata | Pathology + BI-RADS + density | Foreground-only pixel-space CDF matching + unified 3‑task benchmark; single-/two-/four‑view baselines | Vendor/energy diversity; three tasks; consistent harmonization gains; improved attention focus. |
| Backbone | Pre-trained | Learning Rate | Weight Decay | Epochs | Learning Rate scheduler | Batch | Augmentation |
|---|---|---|---|---|---|---|---|
| ResNet-50 | ImageNet-1K | 100 | every 30 | 32 | flip, resize | ||
| DenseNet-121 | ImageNet-1K | 100 | every 30 | 32 | flip, resize | ||
| EfficientNet-B0 | ImageNet-1K | 100 | every 30 | 32 | flip, resize | ||
| Swin-T | ImageNet-1K | 100 | every 30 | 32 | flip, resize |
| Stage | Operation (applied to all models equally) |
|---|---|
| DICOM handling | Convert MONOCHROME1 to MONOCHROME2; remove text burn-ins. |
| Foreground mask | Define ; exclude background from histograms. |
| Harmonization | Foreground CDF matching to low-energy reference (Eqs. 1–8); training-free, model-agnostic. |
| Resize/replicate | Resize to or ; replicate grayscale to 3 channels for ImageNet backbones (Sec. 6.1 ). |
| Augment | Random horizontal flip (all backbones, all tasks). |
| Normalization | As in backbone defaults; no per-model special tuning beyond LR noted in Sec. 6.1 . |