LLM2D
对抗桶效应:多知识对齐的药物推荐
Combating the Bucket Effect:Multi-Knowledge Alignment for Medication Recommendation
作者: Xiang Li, Haixu Ma, Guanyong Wu, Shi Mu, Chen Li, Shunpan Liang
发布日期: 4/28/2025
arXiv ID: oai:arXiv.org:2504.18096v1

摘要

arXiv:2504.18096v1 宣传类型: 新 摘要:药物推荐在医疗保健中至关重要,它可以根据患者的电子健康记录(EHR)提供有效的治疗方法。之前的研究所表明,整合更多的药物相关知识可以提高药物表示的准确性。然而,并非所有药物同时包含多种类型的知识数据。例如,有些药物仅提供文本描述而没有结构化数据。这种数据可用性的不平衡限制了现有模型的性能,我们将其在药物推荐中的这一挑战称为“桶效应”。我们的数据分析揭示了药物推荐中“桶效应”的严重性。为了解决这一问题,我们引入了一种跨模态药物编码器,它可以无缝地对齐不同模态的数据,并提出了一种集成多种类型知识的药物推荐框架,称为MKMed。具体来说,我们首先使用对比学习在五种知识模态上预训练了一个跨模态编码器,将它们对齐到一个统一的空间。然后,我们将多种知识的药物表示与患者记录结合起来进行推荐。在MIMIC-III和MIMIC-IV数据集上的 extensive 实验证明,MKMed 在缓解数据的“桶效应”方面表现出色,并且在推荐准确性和安全性方面显著优于现有最先进的基线模型。