用医学本体与图优化提升药物推荐鲁棒性,应对罕见病和数据缺失。
HiRef: Leveraging Hierarchical Ontology and Network Refinement for Robust Medication Recommendation
- 融合医学本体层级语义与真实病历共现关系,增强知识迁移能力。
- 在MIMIC-III/IV上表现优异,对未见用药代码仍保持高准确率。
- 适合医疗AI研究者及临床决策系统开发者参考。
药物推荐是辅助医生从纵向患者电子健康记录中及时决策的关键任务。然而,真实世界电子健康记录(EHR)存在罕见医疗实体和不完整记录,难以完整反映临床真实情况。尽管基于纵向EHR的数据驱动模型常取得良好实证性能,但在缺失或新出现条件下的泛化能力差,主要因其依赖于已观测到的共现模式。为此,我们提出一种统一框架HiRef,结合两种互补结构:(i) 编制医学本体中的层次语义,(ii) 从真实世界EHR中提炼出的优化共现模式。我们将本体实体嵌入双曲空间,自然捕捉树状关系,通过共享祖先实现知识迁移,从而提升对未见编码的泛化能力。为进一步提升鲁棒性,引入先验引导的稀疏正则化方案,抑制虚假边的同时保留临床有意义关联。模型在MIMIC-III和MIMIC-IV基准上表现强劲,并在模拟未见代码场景下保持高精度。大量实验与详尽消融研究证实,HiRef对未见医疗编码具有强韧性,学习到的稀疏化图结构与代码嵌入分析提供了有力支持。
原文摘要 · Abstract (English)
Medication recommendation is a crucial task for assisting physicians in making timely decisions from longitudinal patient medical records. However, real-world EHR data present significant challenges due to the presence of rarely observed medical entities and incomplete records that may not fully capture the clinical ground truth. While data-driven models trained on longitudinal Electronic Health Records often achieve strong empirical performance, they struggle to generalize under missing or novel conditions, largely due to their reliance on observed co-occurrence patterns. To address these issues, we propose Hierarchical Ontology and Network Refinement for Robust Medication Recommendation (HiRef), a unified framework that combines two complementary structures: (i) the hierarchical semantics encoded in curated medical ontologies, and (ii) refined co-occurrence patterns derived from real-world EHRs. We embed ontology entities in hyperbolic space, which naturally captures tree-like relationships and enables knowledge transfer through shared ancestors, thereby improving generalizability to unseen codes. To further improve robustness, we introduce a prior-guided sparse regularization scheme that refines the EHR co-occurrence graph by suppressing spurious edges while preserving clinically meaningful associations. Our model achieves strong performance on EHR benchmarks (MIMIC-III and MIMIC-IV) and maintains high accuracy under simulated unseen-code settings. Extensive experiments with comprehensive ablation studies demonstrate HiRef's resilience to unseen medical codes, supported by in-depth analyses of the learned sparsified graph structure and medical code embeddings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。