Region-of-Interest Augmentation for Mammography Classification under Patient-Level Cross-Validation

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Bigdeli, Farbod, Mohammadagha, Mohsen, Bigdeli, Ali
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866914074861764608
author Bigdeli, Farbod
Mohammadagha, Mohsen
Bigdeli, Ali
author_facet Bigdeli, Farbod
Mohammadagha, Mohsen
Bigdeli, Ali
contents Breast cancer screening with mammography remains central to early detection and mortality reduction. Deep learning has shown strong potential for automating mammogram interpretation, yet limited-resolution datasets and small sample sizes continue to restrict performance. We revisit the Mini-DDSM dataset (9,684 images; 2,414 patients) and introduce a lightweight region-of-interest (ROI) augmentation strategy. During training, full images are probabilistically replaced with random ROI crops sampled from a precomputed, label-free bounding-box bank, with optional jitter to increase variability. We evaluate under strict patient-level cross-validation and report ROC-AUC, PR-AUC, and training-time efficiency metrics (throughput and GPU memory). Because ROI augmentation is training-only, inference-time cost remains unchanged. On Mini-DDSM, ROI augmentation (best: p_roi = 0.10, alpha = 0.10) yields modest average ROC-AUC gains, with performance varying across folds; PR-AUC is flat to slightly lower. These results demonstrate that simple, data-centric ROI strategies can enhance mammography classification in constrained settings without requiring additional labels or architectural modifications.
format Preprint
id arxiv_https___arxiv_org_abs_2509_20585
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Region-of-Interest Augmentation for Mammography Classification under Patient-Level Cross-Validation
Bigdeli, Farbod
Mohammadagha, Mohsen
Bigdeli, Ali
Computer Vision and Pattern Recognition
Machine Learning
Breast cancer screening with mammography remains central to early detection and mortality reduction. Deep learning has shown strong potential for automating mammogram interpretation, yet limited-resolution datasets and small sample sizes continue to restrict performance. We revisit the Mini-DDSM dataset (9,684 images; 2,414 patients) and introduce a lightweight region-of-interest (ROI) augmentation strategy. During training, full images are probabilistically replaced with random ROI crops sampled from a precomputed, label-free bounding-box bank, with optional jitter to increase variability. We evaluate under strict patient-level cross-validation and report ROC-AUC, PR-AUC, and training-time efficiency metrics (throughput and GPU memory). Because ROI augmentation is training-only, inference-time cost remains unchanged. On Mini-DDSM, ROI augmentation (best: p_roi = 0.10, alpha = 0.10) yields modest average ROC-AUC gains, with performance varying across folds; PR-AUC is flat to slightly lower. These results demonstrate that simple, data-centric ROI strategies can enhance mammography classification in constrained settings without requiring additional labels or architectural modifications.
title Region-of-Interest Augmentation for Mammography Classification under Patient-Level Cross-Validation
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2509.20585