arXiv · 2506.06977
Discovering Hierarchy-Grounded Domains with Adaptive Granularity for Clinical Domain Generalization
Abstract
Domain generalization has become a critical challenge in predictive healthcare, where different patient groups often exhibit shifting data distributions that degrade model performance. Still, regular domain generalization approaches often struggle in clinical settings due to (1) the absence of domain labels and (2) the lack of clinical insight integration. To address these challenges in healthcare, we aim to explore how medical ontologies can be used to discover dynamic yet hierarchy-grounded patient domains, a partitioning strategy that remains under-explored in prior work. Hence, we introduce UdonCare, a hierarchy-pruning method that iteratively divides patients into latent domains and retrieve domain-invariant (label) information from patient data. On two public datasets, UdonCare shows superiority over eight baselines across four representative clinical prediction tasks with substantial domain gaps, highlighting the potential of medical knowledge for enhancing model generalization.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Pengfei Hu, Xiaoxue Han, Fei Wang, Yue Ning. 2025-06-08. Discovering Hierarchy-Grounded Domains with Adaptive Granularity for Clinical Domain Generalization. https://arxiv.org/abs/2506.06977
Cite the original work for its findings. Save a collection to share your selection of sources.