Exploring Accurate and Transparent Domain Adaptation in Predictive Healthcare via Concept-Grounded Orthogonal Inference
About
Deep learning models for clinical event prediction on electronic health records (EHR) often suffer performance degradation when deployed under different data distributions. While domain adaptation (DA) methods can mitigate such shifts, its "black-box" nature prevents widespread adoption in clinical practice where transparency is essential for trust and safety. We propose ExtraCare to decompose patient representations into invariant and covariant components. By supervising these two components and enforcing their orthogonality during training, our model preserves label information while exposing domain-specific variation at the same time for more accurate predictions than most feature alignment models. More importantly, it offers human-understandable explanations by mapping sparse latent dimensions to medical concepts and quantifying their contributions via targeted ablations. ExtraCare is evaluated on two real-world EHR datasets across multiple domain partition settings, demonstrating superior performance along with enhanced transparency, as evidenced by its accurate predictions and explanations from extensive case studies.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Diagnosis prediction | eICU spatial domain shift, Midwest as target (test) | Weighted F168.61 | 21 | |
| Heart failure prediction | eICU spatial domain shift 2014-2015 (test) | AUROC91.88 | 15 | |
| Heart failure prediction | OCHIN temporal domain shift 2012-2023 (test) | AUROC95.48 | 15 | |
| Diagnosis prediction | OCHIN temporal domain shift 2012-2023 (test) | w-F174.05 | 13 | |
| Heart failure prediction | OCHIN spatial gap | AUROC94.52 | 11 | |
| Diagnosis prediction | OCHIN spatial gap | w-F174.97 | 9 |