Cross-cohort Mamography]{Representation-dependent domain adaptation for cross-cohort generalization of mammography diagnostic models
Pandey, A. K.; Ahmad, S.
Show abstract
Background: Artificial intelligence (AI) models for mammography can achieve high diagnostic performance when training and test data originate from similar populations and imaging environments, but performance often deteriorates under cross- cohort domain shift. Differences in acquisition systems, image characteristics, preprocessing, population composition, and disease prevalence can introduce distributional variation that is poorly captured by conventional within- dataset validation. Recent studies have further shown that Foundation Model representations can retain substantial dataset-specific information. However, it remains unclear how image representation influences diagnostic transfer to completely unseen cohorts and whether domain-adversarial learning provides comparable benefits across different representation types. Methods: We systematically evaluated cross-cohort mammography classification using seven independent datasets spanning diverse geographic, institutional, scanner, and imaging environments. Three image representations were compared: handcrafted radiomic features, EfficientNet-B0-derived features, and embeddings from the medical vision-language model MedSigLIP-448. Each was evaluated under three preprocessing conditions: unprocessed images, Otsu-based breast-region extraction, and Otsu extraction combined with Contrast Limited Adaptive Histogram Equalization (CLAHE). Patient-level classification was performed using a mean-pooled multilayer perceptron (MLP) baseline and an attention-based Domain-Adversarial Neural Network (DANN). Generalization was evaluated using a multi-source Leave-One-Domain-Out (LODO) framework in which six datasets were used for model development and the seventh was completely withheld as an unseen test cohort, iteratively across all seven datasets. Performance was primarily assessed by area under the receiver operating characteristic curve (AUC-ROC). Results: Cross-cohort performance depended strongly on image representation and model architecture. MedSigLIP consistently provided the highest performance on unseen cohorts. With MedSigLIP features, mean LODO AUC-ROC increased from 0.598 with MLP to 0.685 with DANN for unprocessed images, from 0.607 to 0.678 after Otsu preprocessing, and from 0.603 to 0.659 after CLAHE+Otsu preprocessing. When DANN-MLP differences were evaluated across held-out datasets and preprocessing conditions, a significant gain was observed for MedSigLIP after correction for multiple comparisons, whereas corresponding gains were not significant for Radiomics or EfficientNet-B0. A mixed-effects analysis similarly showed a positive MedSigLIP X DANN interaction, although this narrowly missed conventional statistical significance. Unprocessed images produced the highest numerical mean performance for MedSigLIP-DANN, but preprocessing itself had no significant overall effect. Notably, linear probing showed strong dataset discriminability in MedSigLIP and EfficientNet representations but little linearly recoverable dataset information in Radiomics, indicating that low dataset separability alone did not predict better diagnostic generalization. Conclusion: Cross-cohort mammography generalization depends strongly on the choice of image representation and its interaction with the classification strategy. MedSigLIP provided the most transferable representation in our multi-source unseen-domain evaluation, and the performance gain associated with DANN was concentrated in MedSigLIP rather than consistently observed across representation types. These findings suggest that the effectiveness of domain-adversarial learning is representation-dependent and that neither conventional preprocessing nor low dataset discriminability alone ensures robust transfer to unseen clinical cohorts.
Matching journals
The top 8 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Classification performance bias between training and test sets in a limited mammography dataset 94%
- BioFuse: An Embedding Fusion Framework for Biomedical Foundation Models 92%
- Enhancing Semantic Segmentation in Chest X-Ray Images through Image Preprocessing: ps-KDE for Pixel-wise Substitution by Kernel Density Estimation 92%
Similar papers in this journal
- Reproducible And Clinically Translatable Deep Neural Networks For Cervical Screening 94%
- Automated and Manual Quantification of Tumour Cellularity in Digital Slides for Tumour Burden Assessment 93%
- Dual Adversarial Deconfounding Autoencoder for joint batch-effects removal from multi-center and multi-scanner radiomics data 92%
Similar papers in this journal
- A human-in-the-loop explanation framework for morphologically transparent AI predictions from whole-slide images 94%
- Diagnostic Accuracy of Artificial Intelligence in Classifying HER2 Status in Breast Cancer Immunohistochemistry Slides and Implications for HER2-Low Cases: A Systematic Review and Meta-Analysis 91%
- Machine Learning Generalizability Across Healthcare Settings: Insights from multi-site COVID-19 screening 90%
Similar papers in this journal
- Deep Learning Image Analysis of Benign Breast Disease to Identify Subsequent Risk of Breast Cancer 93%
- Tumor localization strategies of multi-cancer early detection tests: a quantitative assessment 90%
- Relation of Quantitative Histologic and Radiologic Breast Tissue Composition Metrics with Invasive Breast Cancer Risk 87%
"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.