解决药物推荐中知识数据不均衡问题,提升推荐准确与安全
Combating the Bucket Effect:Multi-Knowledge Alignment for Medication Recommendation
- 设计跨模态编码器,统一文本、结构化等多源药物数据
- 在MIMIC-III/IV上实现更优推荐精度与安全性,缓解数据缺失影响
- 适合医疗AI研究者与临床辅助系统开发者参考
药物推荐在医疗中至关重要,基于电子健康记录(EHR)提供有效治疗方案。以往研究显示,整合更多药物相关知识可提升药物表征准确性。然而,并非所有药物同时具备多种类型的知识数据——例如部分药物仅有文本描述而无结构化信息。这种数据可用性的不平衡限制了现有模型性能,我们称之为“桶效应”。数据分析揭示了该问题的严重性。为此,我们提出一种跨模态药物编码器,能无缝对齐不同模态数据,并构建融合多类型知识的药物推荐框架MKMed。首先,在五种知识模态上通过对比学习预训练跨模态编码器,将数据映射至统一空间;随后,将多知识药物表征与患者记录结合进行推荐。在MIMIC-III和MIMIC-IV数据集上的大量实验表明,MKMed有效缓解了数据“桶效应”,显著优于当前最优基线,在推荐准确性和安全性上均有提升。
原文摘要 · Abstract (English)
Medication recommendation is crucial in healthcare, offering effective treatments based on patient's electronic health records (EHR). Previous studies show that integrating more medication-related knowledge improves medication representation accuracy. However, not all medications encompass multiple types of knowledge data simultaneously. For instance, some medications provide only textual descriptions without structured data. This imbalance in data availability limits the performance of existing models, a challenge we term the "bucket effect" in medication recommendation. Our data analysis uncovers the severity of the "bucket effect" in medication recommendation. To fill this gap, we introduce a cross-modal medication encoder capable of seamlessly aligning data from different modalities and propose a medication recommendation framework to integrate Multiple types of Knowledge, named MKMed. Specifically, we first pre-train a cross-modal encoder with contrastive learning on five knowledge modalities, aligning them into a unified space. Then, we combine the multi-knowledge medication representations with patient records for recommendations. Extensive experiments on the MIMIC-III and MIMIC-IV datasets demonstrate that MKMed mitigates the "bucket effect" in data, and significantly outperforms state-of-the-art baselines in recommendation accuracy and safety.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。