Benchmarking Transfer Learning for Dense Breast Tissue Segmentation on Small Mammogram Datasets
Qu, B.; Liu, W.; Zhou, L.; Guo, X.; Malin, B.; Yin, Z.
Show abstract
Dense breast tissue diminishes the sensitivity of mammographic screening and is a key cancer risk factor, which motivates accurate segmentation under scarce and expensive expert annotations in the medical imaging domain. Here, we benchmark the effect of backbone architecture, self-supervised pre-training (SSL), fine-tuning strategy, and loss design for dense-tissue segmentation on a small expert-labeled dataset (596 images) and an in-domain unlabeled corpus (20, 000 images), reflecting the lack of large public pixel-level density datasets. CNNs (EfficientNet, Xception, nnUNet) clearly outperform transformer and Medical-SAM2 models, and full or layer-wise fine-tuning reliably exceeds parameter-efficient updates. Generic image-only SSL (MIM, SimCLR, Barlow Twins) often yields negligible or negative gains over ImageNet initialization, whereas a simple multi-view contrastive SSL and a hybrid segmentation-density loss provide the best accuracy and calibration (e.g., MAE from 14.8% to 11.8%, Spearman with the four BI-RADS breast density categories from 0.42 to 0.51 on VinDr). We also quantify GPU hours for different SSL and fine-tuning choices, showing that only a small set of protocols, such as EfficientNet with multi-view SSL, hybrid loss, and full fine-tuning, offers favorable accuracy-efficiency trade-offs. These findings provide practical defaults for annotation-limited mammography studies and support compute-conscious deployment of automatic breast density assessment in web-based screening workflows.
Matching journals
The top 8 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Clinical Validation of Saliency Maps for Understanding Deep Neural Networks in Ophthalmology 93%
- A Framework for Falsifiable Explanations of Machine Learning Models with an Application in Computational Pathology 93%
- STAMP: Simultaneous Training and Model Pruning for Low Data Regimes in Medical Image Segmentation 93%
Similar papers in this journal
- Generative AI Enables Medical Image Segmentation in Ultra Low-Data Regimes 95%
- Weakly supervised classification of rare aortic valve malformations using unlabeled cardiac MRI sequences 92%
- Features fusion or not: harnessing multiple pathological foundation models using Meta-Encoder for downstream tasks fine-tuning 92%
Similar papers in this journal
- Dual Adversarial Deconfounding Autoencoder for joint batch-effects removal from multi-center and multi-scanner radiomics data 94%
- Spatial Transcriptomics Inferred from Pathology Whole-Slide Images Links Tumor Heterogeneity to Survival in Breast and Lung Cancer 93%
- Test-time augmentation for deep learning-based cell segmentation on microscopy images 93%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.