cs.AIAug 10, 2026

Coupled Graph--Policy Distillation for Personalized Medication Safety in Older Adults with Multimorbidity

Authors: Zihan WangAnglin LiuRongyi WangDantong LiYi LuSiqing YuanHongxia XuZhongtian Long+1 more

Organizations: 1Artificial Intelligence Thrust, The Hong Kong University of Science and Technology (Guangzhou) · Faculty of Engineering, University of New South Wales · 3Medical Big Data Center, Guangdong Provincial People’s Hospital, Southern Medical University · 4Transvascular Implantation Devices Research Institute, Zhejiang University · School of Computer Science and Technology, Huazhong University of Science and Technology, Wuhan 430074, China.

Abstract

Large language model (LLM) agents can support medication review between clinical visits, but safe choices for older adults with multimorbidity depend on conditions, medications, and geriatric risks that users may omit. We introduce ATLAS, a coupled graph--policy distillation framework for patient-adaptive medication safety. ATLAS structures guideline evidence as a medication-safety graph. Targeted questions update the patient state and distill relevant relations into a patient-specific medication conflict graph (PMCG). A risk-first multi-agent policy uses the PMCG to screen contraindications, assess cautions and monitoring needs, identify safer alternatives, and verify the final medication plan. We also introduce GeriMedBench, an interactive benchmark that tests safety-critical information acquisition and evidence-based decision revision. Across a European non-interactive multimorbidity benchmark, an Asian interactive multimorbidity benchmark, and an Asian non-interactive cross-guideline benchmark, ATLAS achieves the strongest complete-decision performance among the compared systems. On the European non-interactive multimorbidity benchmark, it exceeds the strongest proprietary LLM baseline by 53.73 points in Strict Success Rate and 14.63 points in overall safety reasoning score (OSRS), with no unsafe recommendations under the automated evaluator. A blinded clinician evaluation gives ATLAS higher mean ratings across all five criteria and flags potentially unsafe recommendations in one ATLAS case and two Gemini cases.

Explore similar work

CardsList