Stroke remains a leading cause of mortality and disability worldwide, requiring timely therapeutic decisions. Existing content-based drug recommendation approaches often rely on static similarity measures and multi-stage pipelines (e.g., clustering followed by instance-based retrieval), which may limit scalability and generalization. We propose CRGANNC-GD, an end-to-end recommendation framework that models patient profiles and medication classes as a heterogeneous graph and performs representation learning through graph diffusion to capture clinically meaningful relationships between symptoms, comorbidities, demographics, and drug classes. Unlike traditional KNN-centric pipelines, CRGANNC-GD learns a task-optimized latent space and outputs calibrated scores for multiple drug classes (antihypertensive, anticoagulant, fibrate). We further incorporate a ranking-aware loss to directly optimize top-k recommendation quality and introduce an explainability module that highlights influential clinical attributes and graph neighborhoods contributing to each recommendation. Experiments on a stroke dataset with 9691 records demonstrate that CRGANNC-GD improves predictive performance and recommendation quality over classical content-based baselines, while maintaining fast inference suitable for clinical decision support and telemedicine settings. Our results suggest that graph-diffusion representation learning provides a scalable and robust alternative for medication class recommendation in stroke management, with a clear pathway to extend the framework toward novel drug discovery via drug-drug and drug-target similarity integration.
References
[1]
de Andrade, J.B.C., Fagundes, T.P., Katsuyama, E. and Silva, G.S. (2026) Digital Health in Low-Resource Settings: Comprehensive Challenges and Opportunities with a Focus on Stroke Care. Stroke, 57, 245-253. https://doi.org/10.1161/strokeaha.125.050448
[2]
Nayyar, S. and Brown, P. (2026) Medications for Anti-Coagulation in Stroke Prevention. In: Thomas, R.E., Ed., MedicationsforOlderPersons, Elsevier, 141-154. https://doi.org/10.1016/b978-0-443-29025-1.00005-9
[3]
Kücking, F., Busch, D.A., Przysucha, M., Kutza, J., Hannemann, N., Hüsers, J., et al. (2026) Impact of AI Recommendation Correctness on Diagnostic Accuracy in Clinical Decision-Making. InternationalJournalofMedicalInformatics, 207, Article ID: 106223. https://doi.org/10.1016/j.ijmedinf.2025.106223
[4]
Ayata, C., Bath, P.M., Planas, A.M., Allan, S.M., Boltze, J., Cabeen, R.P., et al. (2026) Preclinical Ischemic Stroke Multicenter Trials (PRISM) Collective Statement: Opportunities, Challenges, and Recommendations for a New Era. Stroke, 57, e12-e40. https://doi.org/10.1161/strokeaha.125.052056
[5]
Tene Ceskoutse, R.F., et al. (2024) Crgannc: A Clustering-Based Content Recommendation Model for Drug Class Detection in Stroke Patients. Expert Systems with Ap-plications.
[6]
Liu, Y., Wen, Z.J., Wang, Y.R., Zhong, Y.X., Wang, J.X., Hu, Y.H., Zhou, P. and Guo, S.M. (2024) Artificial Intelligence in Ischemic Stroke Images: Current Applications and Future Directions. Frontiers in Neurology, 15, Article 1418060. https://doi.org/10.3389/fneur.2024.1418060
[7]
Amann, J. (2021) Machine Learning (ML) in Stroke Medicine: Opportunities and Challenges for Risk Prediction and Prevention. In: Jotterand, F. and Ienca M., Eds., Artificial Intelligence in Brain and Mental Health: Philosophical, Ethical & Policy Issues, Springer International Publishing, 57-71. https://doi.org/10.1007/978-3-030-74188-4_5
[8]
Basem, J., Mani, R., Sun, S., Gilotra, K., Dianati-Maleki, N. and Dashti, R. (2025) Clinical Applications of Artificial Intelligence and Machine Learning in Neurocardiology: A Comprehensive Review. Frontiers in Cardiovascular Medicine, 12, Article 1525966. https://doi.org/10.3389/fcvm.2025.1525966
[9]
Hancox, Z., Pang, A., Conaghan, P.G., Kingsbury, S.R., Clegg, A. and Relton, S.D. (2024) A Systematic Review of Networks for Prognostic Prediction of Health Outcomes and Diagnostic Prediction of Health Conditions within Electronic Health Records. Artificial Intelligence in Medicine, 158, Article 102999. https://doi.org/10.1016/j.artmed.2024.102999
[10]
Vaida, M. and Huang, Z.Y. (2026) Multimodal Graph Neural Networks in Healthcare: A Review of Fusion Strategies across Biomedical Domains. Frontiers in Artificial Intelligence, 8, Article 1716706. https://doi.org/10.3389/frai.2025.1716706
[11]
Al Khatib, H.S., Neupane, S., Manchukonda, H.K., Golilarz, N.A., Mittal, S., Amirlatifi, A. and Rahimi, S. (2024) Patient-Centric Knowledge Graphs: A Survey of Current Methods, Challenges, and Applications. Frontiers in Artificial Intelligence, 7, Article 1388479. https://doi.org/10.3389/frai.2024.1388479
[12]
Ali, Z., Huang, Y., Ullah, I., Feng, J.L., Deng, C., Thierry, N., et al. (2022) Deep Learning for Medication Recommendation: A Systematic Survey. Data Intelligence, 5, 303-354. https://doi.org/10.1162/dint_a_00197
[13]
Mishra, P. and Shridevi, S. (2024) Knowledge Graph-Driven Medicine Recommendation System Using Graph Neural Networks on Longitudinal Medical Records. Scientific Reports, 14, Article 22611. https://doi.org/10.1038/s41598-024-75784-5
[14]
Liu, F., Wang, W., Zheng, J., Xie, Y., Wang, X. and Zhang, D. (2025) EDRMM: Enhancing Drug Recommendation via Multi-Granularity and Multi-Attribute Representation. BMC Bioinformatics, 26, Article No. 173. https://doi.org/10.1186/s12859-025-06167-4
[15]
Wang, Y.G. and Lu, B.X. (2024) Drug Recommendation Combining Attention Mechanism and Graph Encoder. Proceedings of the 2023 International Conference on Information Education and Artificial Intelligence, New York, 22-24 December 2023, 750-755. Association for Computing Machinery. https://doi.org/10.1145/3660043.3660177
[16]
Jiang, P.C., Xiao, C., Cross, A. and Sun, J.M. (2024) Graphcare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs.
[17]
Rivera-Juzga, C.Y., Mistry, D., Knowles, A.T., Lip, G.Y.H., Ortega-Martorell, S. and Olier, I. (2025) Graph Neural Networks for the Enhanced Prediction of New Atrial Fibrillation Episodes after Stroke. Computers in Biology and Medicine, 197, Article 111095. https://doi.org/10.1016/j.compbiomed.2025.111095