EDRMM: enhancing drug recommendation via multi-granularity and multi-attribute representation

作者：Feiyan Liu, Wenhao Wang, Jiawei Zheng, Yibo Xie, Xiaoli Wang, Dongxiang Zhang · 发表于：BMC Bioinformatics · 年份：2025 · DOI：10.1186/s12859-025-06167-4 · 被引用次数：6 · 研究领域：Machine Learning in Healthcare、Topic Modeling、Artificial Intelligence in Healthcare and Education

BACKGROUND: Drug recommendation is a crucial application of artificial intelligence in medical practice. Although many models have been proposed to solve this task, two challenges remain unresolved: (i) most existing models use all historical visits as input, overlooking fine-grained correlations between historical and current information; (ii) Electronic Health Records (EHRs) are underutilized, with only partial information considered to describe patient conditions. To tackle the challenges, we propose a novel drug recommendation model, denoted by EDRMM, which incorporates multi-granularity and multi-attribute information into representation learning. We develop a longitudinal attribute-level history selection mechanism to effectively identify fine-grained historical information that is highly relevant to a patient's current clinical conditions. We analyze the impact of key Electronic Health Record (EHR) attributes, demonstrating that incorporating such attributes into patient representations can further boost performance. We also design an adaptive global Drug-Drug Interaction (DDI) risk regularization term for the DDI loss function to better balance accuracy and safety during training. RESULTS: Experimental results show that our model achieves state-of-the-art performance on a widely used MIMIC-III dataset. CONCLUSIONS: EDRMM overcomes two key drug recommendation limitations through three innovations: (1) Dynamic attribute-level history selection, which retrieves relevant ...