Exploring Accurate and Transparent Domain Adaptation in Predictive Healthcare via Concept-Grounded Orthogonal Inference
Pengfei Hu, Chang Lu, Feifan Liu, Yue Ning
Abstract
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, their "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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext a7ba695c-5a34-45e9-bd3e-b7d4af1a8d31Builds on9
- Sparse Autoencoders Find Highly Interpretable Features in Language ModelsRobert Huben, Hoagy Cunningham, Logan Riggs Smith, Aidan Ewart et al.ICLR 2024 · 1,072 citations
- Cycle Self-Training for Domain AdaptationHong Liu, Jianmin Wang, Mingsheng LongNeurIPS 2021 · 236 citations
- Safe Self-Refinement for Transformer-based Domain AdaptationTao Sun, Cheng Lu, Tianshuo Zhang, Haibin LingCVPR 2022 · 110 citations
- Connect, Not Collapse: Explaining Contrastive Learning for Unsupervised Domain AdaptationKendrick Shen, Robbie M. Jones, Ananya Kumar, Sang Michael Xie et al.ICML 2022 · 102 citations
- Style Neophile: Constantly Seeking Novel Styles for Domain GeneralizationJuwon Kang, Sohyun Lee, Namyup Kim, Suha KwakCVPR 2022 · 99 citations
Related papers
- ConCare: Personalized Clinical Feature Embedding via Capturing the Healthcare ContextLiantao Ma, Chaohe Zhang, Yasha Wang, Wenjie Ruan et al.AAAI 2020 · 190 citations
- AdaCare: Explainable Clinical Health Status Representation Learning via Scale-Adaptive Feature Extraction and RecalibrationLiantao Ma, Junyi Gao, Yasha Wang, Chaohe Zhang et al.AAAI 2020 · 156 citations
- EviCare: Enhancing Diagnosis Prediction with Deep Model-Guided Evidence for In-Context ReasoningHengyu Zhang, Xuyun Zhang, Pengxiang Zhan, Linhao Luo et al.KDD 2026
- An Iterative Self-Learning Framework for Medical Domain GeneralizationZhenbang Wu, Huaxiu Yao, David M. Liebovitz, Jimeng SunNeurIPS 2023 · 12 citations
- FlexCare: Leveraging Cross-Task Synergy for Flexible Multimodal Healthcare PredictionMuhao Xu, Zhenfeng Zhu, Youru Li, Shuai Zheng et al.KDD 2024 · 5 citations
