动态因果注意力融合,提升药物推荐精准与安全
CafeMed: Causal Attention Fusion Enhanced Medication Recommendation
- 用动态权重模拟患者状态对用药影响,替代静态因果关系
- 在MIMIC-III/IV上准确率超基线,且药物相互作用率更低
- 适合临床辅助决策系统开发,尤其关注个性化与安全
药物推荐系统在辅助临床个性化治疗决策中至关重要。现有方法虽在药物表征学习上取得进展,但仍存在两大局限:(i) 将医疗实体视为独立特征,未建模其对用药选择的协同效应;(ii) 采用静态因果关系,无法适应患者个体化情境与健康状态。为此,我们提出CafeMed框架,通过动态因果推理与跨模态注意力融合,实现安全、精准的药物推荐。CafeMed引入两个核心组件:基于患者状态动态生成调节权重的因果权重生成器(CWG),以及捕获诊断与操作间复杂依赖关系的通道调和注意力精炼模块(CHARM)。该设计使系统能建模多种疾病共同影响治疗决策的过程,同时满足用药安全约束。在MIMIC-III与MIMIC-IV数据集上的大量实验表明,CafeMed显著优于当前最优基线,在药物预测准确率上表现更优,且药物-药物相互作用率更低。结果表明,融入动态因果关系与跨模态协同效应可带来更符合临床需求的个性化推荐。代码已开源:https://github.com/rkl71/CafeMed。
原文摘要 · Abstract (English)
Medication recommendation systems play a crucial role in assisting clinicians with personalized treatment decisions. While existing approaches have made significant progress in learning medication representations, they suffer from two fundamental limitations: (i) treating medical entities as independent features without modeling their synergistic effects on medication selection; (ii) employing static causal relationships that fail to adapt to patient-specific contexts and health states. To address these challenges, we propose CafeMed, a framework that integrates dynamic causal reasoning with cross-modal attention for safe and accurate medication recommendation. CafeMed introduces two key components: the Causal Weight Generator (CWG) that transforms static causal effects into dynamic modulation weights based on individual patient states, and the Channel Harmonized Attention Refinement Module (CHARM) that captures complex interdependencies between diagnoses and procedures. This design enables CafeMed to model how different medical conditions jointly influence treatment decisions while maintaining medication safety constraints. Extensive experiments on MIMIC-III and MIMIC-IV datasets demonstrate that CafeMed significantly outperforms state-of-the-art baselines, achieving superior accuracy in medication prediction while maintaining the lower drug--drug interaction rates. Our results indicate that incorporating dynamic causal relationships and cross-modal synergies leads to more clinically-aligned and personalized medication recommendations. Our code is released publicly at https://github.com/rkl71/CafeMed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。