用药物知识约束注意力机制,提升电子病历中用药推荐的准确与安全。
GraphDiffMed: Knowledge-Constrained Differential Attention with Pharmacological Graph Priors for Medication Recommendation

- 双尺度差异注意力过滤就诊内和跨时间的噪声信号
- 在MIMIC-III上优于强基线,且安全性表现更优
- 仅用人口统计特征即达最佳效果,适合临床部署
从电子健康记录(EHR)中推荐安全有效的药物组合是核心临床人工智能问题,但因患者轨迹长、噪声多且临床异质性强而困难。现有方法通常在时间建模或药物知识整合(如药物-药物相互作用,DDIs)中表现优异,却难以兼顾二者且稳健去噪。我们提出GraphDiffMed,基于双尺度Differential Attention v2构建的知识约束用药推荐框架。在就诊内与跨访视层级应用差异注意力,以过滤干扰信号;同时在学习过程中融入药理学约束。在MIMIC-III上的实验及消融研究显示,该设计持续提升推荐质量与排序性能,并实现更优的安全性平衡。我们进一步发现,在实验设置下,仅使用人口统计辅助特征的配置表现最强。总体表明,将噪声感知注意力与药理约束结合可生成更可靠、更具临床意义的用药推荐。代码已开源:https://github.com/saxenakrati09/GraphDiffMed。
原文摘要 · Abstract (English)
Recommending safe and effective medication combinations from electronic health records (EHRs) is a core clinical AI problem, yet it remains difficult because patient trajectories are long, noisy, and clinically heterogeneous. Existing methods typically excel at either temporal modeling across visits or pharmacological knowledge integration (e.g., drug-drug interactions, DDIs), but rarely achieve both while robustly suppressing noise. We present GraphDiffMed, a knowledge-constrained medication recommendation framework built on dual-scale Differential Attention v2. Differential attention is applied at both intra-visit and inter-visit levels to filter spurious signals within encounters and across longitudinal history, while pharmacological constraints are incorporated during learning. Experiments on MIMIC-III and ablation studies show that this design consistently improves recommendation quality and ranking over strong baselines while achieving a more favorable safety performance balance. We further find that the strongest-performing configuration uses only demographic auxiliary features under our experimental setting. Overall, GraphDiffMed demonstrates that combining noise-aware attention with pharmacological constraints yields more reliable and clinically meaningful medication recommendation. We open-source our code at https://github.com/saxenakrati09/GraphDiffMed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。