CIV-DG: Conditional Instrumental Variables for Domain Generalization in Medical Imaging
Not on the map. Why?
Abstract (arXiv)
Cross-site generalizability in medical AI is fundamentally compromised by selection bias, a structural mechanism where patient demographics (e.g., age, severity) non-randomly dictate hospital assignment. Conventional Domain Generalization (DG) paradigms, which predominantly target image-level distribution shifts, fail to address the resulting spurious correlations between site-specific variations and diagnostic labels. To surmount this identifiability barrier, we propose CIV-DG, a causal framework that leverages Conditional Instrumental Variables to disentangle pathological semantics from scanner-induced artifacts. By relaxing the strict random assignment assumption of standard IV methods, CIV-DG accommodates complex clinical scenarios where hospital selection is endogenously driven by patient demographics. We instantiate this theory via a Deep Generalized Method of Moments (DeepGMM) architecture, employing a conditional critic to minimize moment violations and enforce instrument-error orthogonality within demographic strata. Extensive experiments on the Camelyon17 benchmark and large-scale Chest X-Ray datasets demonstrate that CIV-DG significantly outperforms leading baselines, validating the efficacy of conditional causal mechanisms in resolving structural confounding for robust medical AI.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 16 of its 28 references that are in the map. First posted on arXiv 26 Mar 2026.
Builds on
16 of 28 references.
References · 16
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2020Shortcut Learning in Deep Neural Networkscited by 1,928
- 2019CheXpert: A Large Chest Radiograph Dataset with Uncertainty Labels and Expert Comparisoncited by 1,410
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 969
- 2021Domain Generalization: A Surveycited by 760
- 2022MedCLIP: Contrastive Learning from Unpaired Medical Images and Textcited by 564
- 2021SWAD: Domain Generalization by Seeking Flat Minimacited by 431
- 2022Last Layer Re-Training is Sufficient for Robustness to Spurious Correlationscited by 379
- 2022Improving Out-of-Distribution Robustness via Selective Augmentationcited by 202
- 2019Causality matters in medical imagingcited by 193
4 more
- 2021Causality-inspired Single-source Domain Generalization for Medical Image Segmentationcited by 118
- 2019Deep Generalized Method of Moments for Instrumental Variable Analysiscited by 88
- 2022Instrumental Variables in Causal Inference and Machine Learning: A Surveycited by 10
- 2025AdFair-CLIP: Adversarial Fair Contrastive Language-Image Pre-training for Chest X-rayscited by 1
Built on by
Not cited yet.