Incomplete Multi-View Drug Recommendation via Multi-Level Representation Learning and Curriculum Learning
The drug recommendation task aims to provide effective and safe prescription decision support for clinical treatment based on patients' past Electronic Health Records (EHR). However, the prevalent phenomenon of missing views in multi-source heterogeneous EHR data may cause performance degradation. This is due to the lack of sufficient information and increased learning difficulties, which limit the practical effectiveness of drug recommendation models in medical applications. In this paper, we emphasize the problems of incompleteness in practical drug recommendation and propose the Incomplete Multi-View Drug Recommendation model via Multi-Level Representation Learning and Curriculum Learning named IMDR. In particular, IMDR employs a Multi-Level Representation Learning architecture equipped with a Medical Code-Level Drug Knowledge Infusion Module and a Visit-Level Cross-View Information Module for patient representation learning to overcome the information loss caused by incomplete data. And then, a Gaussian-guided curriculum learning strategy is proposed to assist the learning process of IMDR with a novel difficulty measure to achieve effective progressive learning under missing medical views. Systematic evaluation on two large-scale real-world medical datasets, MIMIC-III and MIMIC-IV, demonstrates that IMDR reduces the Drug-Drug Interaction (DDI) rate by 2.97% compared to existing state-of-the-art drug recommendation baselines, while achieving significant improvements of 3.29% and 1.97% in Jaccard similarity scores and F1 score, respectively. Furthermore, compared to advanced incomplete multi-view learning (IML) models, IMDR's advantages in Jaccard similarity scores and F1 score further expand to 4.03% and 2.41%.
Authors
- Haitao Yuan (ORCID: https://orcid.org/0000-0001-6721-065X)
- Lili Jiang (ORCID: https://orcid.org/0000-0002-7788-3986)
- Hongtao Lv (ORCID: https://orcid.org/0000-0002-5451-693X)
- Yunsen Tang
- Zhen Li (ORCID: https://orcid.org/0000-0002-9783-169X)
- Jianyong Wang (ORCID: https://orcid.org/0000-0002-7555-170X)
- Wei Zhang (ORCID: https://orcid.org/0000-0001-6763-8146)
- Ning Liu (ORCID: https://orcid.org/0009-0008-5231-1835)
Institutions
- Shandong University (CN)
- Nanyang Technological University (SG)
- Qilu Hospital of Shandong University (CN)
- East China Normal University (CN)
- Umeå University (SE)
- Tsinghua University (CN)
Publication Details
- Published
- 2025-08-03
- DOI
- https://doi.org/10.1145/3711896.3737236
- Primary Topic
- Machine Learning in Healthcare
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Science Foundation
- Natural Science Foundation of Shandong Province
- National Key Research and Development Program of China