REFINE: A Fine-Grained Medication Recommendation System Using Deep Learning and Personalized Drug Interaction Modeling
Suman Bhoi, Mong-Li Lee, Wynne Hsu, Ngiap Chuan Tan
Abstract
Patients with co-morbidities often require multiple medications to manage their conditions. However, existing medication recommendation systems only offer class-level medications and regard all interactions among drugs to have the same level of severity. This limits their ability to provide personalized and safe recommendations tailored to individual needs. In this work, we introduce a deep learning-based fine-grained medication recommendation system called REFINE, which is designed to improve treatment outcomes and minimize adverse drug interactions. In order to better characterize patient’s health conditions, we model the trend in medication dosage titrations and lab test responses, and adapt the vision transformer to obtain effective patient representations. We also model drug interaction severity levels as weighted graphs to learn safe drug combinations and design a balanced loss function to avoid overly conservative recommendations and miss medications that might be needed for certain conditions. Extensive experiments on two real-world datasets shows that REFINE outperforms state-of-the-art techniques.
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 7eafed37-093c-4d43-a3e8-4e24d13ec887Cited by top-tier papers3
- Leave No Patient Behind: Enhancing Medication Recommendation for Rare Disease PatientsZihao Zhao, Yi Jing, Fuli Feng, Jiancan Wu et al.SIGIR 2024 · 19 citations
- HI-DR: Exploiting Health Status-Aware Attention and an EHR Graph+ for Effective Medication RecommendationTaeri Kim, Jiho Heo, Hyunjoon Kim, Sang-Wook KimAAAI 2025 · 6 citations
- A New Approach to Backtracking Counterfactual Explanations: A Unified Causal Framework for Efficient Model InterpretabilityPouria Fatemi, Ehsan Sharifian, Mohammad Hossein YassaeeICML 2025
Builds on4
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- How Attentive are Graph Attention Networks?Shaked Brody, Uri Alon, Eran YahavICLR 2022 · 1,717 citations
- Conditional Generation Net for Medication RecommendationRui Wu, Zhaopeng Qiu, Jiacheng Jiang, Guilin Qi et al.WWW 2022 · 135 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
Related papers
- Context-Aware Safe Medication Recommendations with Molecular Graph and DDI Graph EmbeddingQianyu Chen, Xin Li, Kunnan Geng, Mingzhong WangAAAI 2023 · 38 citations
- Fine-grained List-wise Alignment for Generative Medication RecommendationChenxiao Fan, Chongming Gao, Wentao Shi, Yaxin Gong et al.NeurIPS 2025 · 14 citations
- Integrating Drug Substructures and Longitudinal Electronic Health Records for Personalized Drug RecommendationWenjie Du, Xuqiang Li, Jinke Feng, Shuai Zhang et al.NeurIPS 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
- Awaken the Giant: Activating LLMs via Deep Model Guidance for Boundary-aware Medication RecommendationHang Lv, Zixuan Guo, Yanchao Tan, Wanzi Shao et al.KDD 2026
