Integrating Drug Substructures and Longitudinal Electronic Health Records for Personalized Drug Recommendation
Wenjie Du, Xuqiang Li, Jinke Feng, Shuai Zhang, Wen Zhang, Yang Wang
Abstract
Drug recommendation systems aim to identify optimal drug combinations for patient care, balancing therapeutic efficacy and safety. Advances in large-scale longitudinal EHRs have enabled learning-based approaches that leverage patient histories such as diagnoses, procedures, and previously prescribed drugs, to model complex patient-drug relationships. Yet, many existing solutions overlook standard clinical practices that favor certain drugs for specific conditions and fail to fully integrate the influence of molecular substructures on drug efficacy and safety. In response, we propose SubRec , a unified framework that integrates representation learning across both patient and drug spaces. Specifically, SubRec introduces a conditional information bottleneck to extract core drug substructures most relevant to patient conditions, thereby enhancing interpretability and clinical alignment. Meanwhile, an adaptive vector quantization mechanism is designed to generate patient–drug interaction patterns into a condition-aware codebook which reuses clinically meaningful patterns, reduces training overhead, and provides a controllable latent space for recommendation. Crucially, the synergy between condition-specific substructure learning and discrete patient prototypes allows SubRec to make accurate and personalized drug recommendations. Experimental results on the real-world MIMIC III and IV demonstrate our model’s advantages. The source code is available at https://DrugRecommendation/.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext ee86295b-0e90-4595-96ca-086dba0675bbCited by top-tier papers1
Ask how each one uses itBuilds on8
- Graph Information BottleneckTailin Wu, Hongyu Ren, Pan Li, Jure LeskovecNeurIPS 2020 · 366 citations
- Conditional Generation Net for Medication RecommendationRui Wu, Zhaopeng Qiu, Jiacheng Jiang, Guilin Qi et al.WWW 2022 · 135 citations
- Mole-BERT: Rethinking Pre-training Graph Neural Networks for MoleculesJun Xia, Chengshuai Zhao, Bozhen Hu, Zhangyang Gao et al.ICLR 2023 · 119 citations
- Drug Package Recommendation via Interaction-aware Graph InductionZhi Zheng, Chao Wang, Tong Xu, Dazhong Shen et al.WWW 2021 · 78 citations
- Improving Subgraph Recognition with Variational Graph Information BottleneckJunchi Yu, Jie Cao, Ran HeCVPR 2022 · 56 citations
Related papers
- MedAlign: Enhancing Combinatorial Medication Recommendation with Multi-modality AlignmentHang Lv, Zixuan Guo, Zijie Wu, Yanchao Tan et al.ACM MM 2025 · 4 citations
- Debiased, Longitudinal and Coordinated Drug Recommendation through Multi-Visit Clinic RecordsHongda Sun, Shufang Xie, Shuqi Li, Yuhan Chen et al.NeurIPS 2022 · 57 citations
- Iterative Substructure Extraction for Molecular Relational Learning with Interactive Graph Information BottleneckShuai Zhang, Junfeng Fang, Xuqiang Li, Hongxin Xiang et al.ICLR 2025
- Multimodal Medical Code TokenizerXiaorui Su, Shvat Messica, Yepeng Huang, Ruth Johnson et al.ICML 2025 · 2 citations
- RES-MR: Risk-Aware Reasoning for Explainable and Safe Medication RecommendationCong Wang, Jin Li, Shoujin Wang, Yishuo Li et al.SIGIR 2026 · 1 citation
