mlsanityData to 7 Oct 2026

GraphDiffMed: Knowledge-Constrained Differential Attention with Pharmacological Graph Priors for Medication Recommendation

arXiv 2605.20188 ↗First posted 21 Mar 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Recommending safe and effective medication combinations from electronic health records (EHRs) is a core clinical AI problem, yet it remains difficult because patient trajectories are long, noisy, and clinically heterogeneous. Existing methods typically excel at either temporal modeling across visits or pharmacological knowledge integration (e.g., drug-drug interactions, DDIs), but rarely achieve both while robustly suppressing noise. We present GraphDiffMed, a knowledge-constrained medication recommendation framework built on dual-scale Differential Attention v2. Differential attention is applied at both intra-visit and inter-visit levels to filter spurious signals within encounters and across longitudinal history, while pharmacological constraints are incorporated during learning. Experiments on MIMIC-III and ablation studies show that this design consistently improves recommendation quality and ranking over strong baselines while achieving a more favorable safety performance balance. We further find that the strongest-performing configuration uses only demographic auxiliary features under our experimental setting. Overall, GraphDiffMed demonstrates that combining noise-aware attention with pharmacological constraints yields more reliable and clinically meaningful medication recommendation. We open-source our code at https://github.com/saxenakrati09/GraphDiffMed.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 6 of its 18 references that are in the map. First posted on arXiv 21 Mar 2026.

Builds on

6 of 18 references.

References · 6

  1. 2017Attention Is All You Needcited by 59,932
  2. 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 344
  3. 2018GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combinationcited by 79
  4. 2024Large Language Model Distilling Medication Recommendation Modelcited by 25
  5. 2023SHAPE: A Sample-adaptive Hierarchical Prediction Network for Medication Recommendationcited by 7
  6. 2024CIDGMed: Causal Inference-Driven Medication Recommendation with Enhanced Dual-Granularity Learningcited by 3

Built on by

Not cited yet.