Safe and personalized drug recommendation via heterogeneous graph learning: integrating graphSAGE representations with DDI-aware cross-attention
摘要
Personalized medication recommendation must ensure both efficacy and safety, particularly by avoiding drug–drug interactions (DDIs), a major cause of preventable adverse events. We introduce a heterogeneous graph learning framework that unifies electronic health records (EHRs) using explicit patient–diagnosis, patient–drug, and diagnosis–drug edges together with curated DDI interaction edges. Patient- and diagnosis-specific embeddings are learned inductively via GraphSAGE, while pharmacological risk is modeled through a graph convolutional network (GCN) on the DDI subgraph. A DDI-aware cross-attention mechanism fuses these representations, aligning patient context with safety signals and penalizing unsafe co-prescriptions during training. Evaluation on the MIMIC-III dataset demonstrates that our model consistently outperforms strong baselines, achieving superior predictive accuracy (Jaccard 0.5281, F1 0.6701, PR-AUC 0.7754) while reducing unsafe recommendations (DDI Rate 0.0581). All improvements are statistically significant (p