[Objective] This paper proposes a new drug recommendation algorithm based on the graph neural network integrating patient signs and medication history, aiming to improve the illness diagnosis and treatments. [Methods] First, we constructed a transitive relationship model for abnormal signs and drugs based on the Graph Neural Network(GNN). Then, we designed a precise drug recommendation plan with sign perception and built a heterogeneous graph for the “sign-patient-drug” relationship. Third, our model learned the node representation with sign perception using the R-GCN encoder. Finally, we designed a sign-aware interaction decoder, which integrated the abnormal signs to recommend drugs accurately. [Results] We examined the proposed model with diagnosis and treatment records of three types of diseases from the MIMIC-Ⅲ dataset. Compared with the SVD, NeuMF and NGCF models, the proposed method’s Recall@20 value increased by 5.76, 5.33 and 0.91 percentage point, respectively. Meanwhile, it increased the NDCG@20 value by 5.03, 4.25 and 2.67 percentage point. [Limitations] Our method did not include the dynamic changes of patients’ drug use due to the developments of diseases. [Conclusions] The proposed drug recommendation method is effective and feasible. This model could perceive the impacts of patient signs on medication, which lays foundations for precise drug recommendation algorithm integrating multi-dimensional information.
成全, 佘德昕. 融合患者体征与用药数据的图神经网络药物推荐方法研究*[J]. 数据分析与知识发现, 2022, 6(9): 113-124.
Cheng Quan, She Dexin. Drug Recommendation Based on Graph Neural Network with Patient Signs and Medication Data. Data Analysis and Knowledge Discovery, 2022, 6(9): 113-124.
( Huang Liwei, Jiang Bitao, Lv Shouye, et al. Survey on Deep Learning Based Recommender Systems[J]. Chinese Journal of Computers, 2018, 41(7): 1619-1647.)
( Peng Chunting, Ruan Bing. Diagnosis Value of Laboratory Tests for Infectious Fever[J]. Chinese Journal of Practical Internal Medicine, 2016, 36(12): 1025-1028.)
doi: 10.7504/nk2016110102
( Li Pengfei, Lu Faming, Bao Yunxia, et al. Drug Recommendation Method Based on Medical Process Mining and Patient Signs[J]. Computer Integrated Manufacturing Systems, 2020, 26(6): 1668-1678.)
[5]
Almirall D, Compton S N, Gunlicks-Stoessel M, et al. Designing a Pilot Sequential Multiple Assignment Randomized Trial for Developing an Adaptive Treatment Strategy[J]. Statistics in Medicine, 2012, 31(17): 1887-1902.
doi: 10.1002/sim.4512
pmid: 22438190
[6]
Norén G N, Bate A, Hopstadius J, et al. Temporal Pattern Discovery for Trends and Transient Effects: Its Application to Patient Records[C]// Proceedings of the 14th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 2008: 963-971.
[7]
Wright A, Chen E S, Maloney F L. An Automated Technique for Identifying Associations Between Medications, Laboratory Results and Problems[J]. Journal of Biomedical Informatics, 2010, 43(6): 891-901.
doi: 10.1016/j.jbi.2010.09.009
pmid: 20884377
[8]
Wright A P, Wright A T, McCoy A B, et al. The Use of Sequential Pattern Mining to Predict Next Prescribed Medications[J]. Journal of Biomedical Informatics, 2015, 53: 73-80.
doi: 10.1016/j.jbi.2014.09.003
pmid: 25236952
[9]
Wang H Q, Wu Y Y, Gao C, et al. Medication Combination Prediction Using Temporal Attention Mechanism and Simple Graph Convolution[J]. IEEE Journal of Biomedical and Health Informatics, 2021, 25(10): 3995-4004.
doi: 10.1109/JBHI.2021.3082548
[10]
Choi E, Bahadori M T, Schuetz A, et al. Doctor AI: Predicting Clinical Events via Recurrent Neural Networks[J]. JMLR Workshop and Conference Proceedings, 2016, 56: 301-318.
pmid: 28286600
[11]
Zhang Y T, Chen R, Tang J, et al. LEAP: Learning to Prescribe Effective and Safe Treatment Combinations for Multimorbidity[C]// Proceedings of the 23rd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining. 2017: 1315-1324.
[12]
Song J, Wang Y Y, Tang S L, et al. Local-Global Memory Neural Network for Medication Prediction[J]. IEEE Transactions on Neural Networks and Learning Systems, 2021, 32(4): 1723-1736.
doi: 10.1109/TNNLS.2020.2989364
[13]
An Y, Zhang L, You M, et al. MeSIN: Multilevel Selective and Interactive Network for Medication Recommendation[J]. Knowledge-Based Systems, 2021, 233: 107534.
doi: 10.1016/j.knosys.2021.107534
[14]
Zhou J, Cui G Q, Hu S D, et al. Graph Neural Networks: A Review of Methods and Applications[J]. AI Open, 2020, 1: 57-81.
doi: 10.1016/j.aiopen.2021.01.001
[15]
Hamilton W L, Ying R, Leskovec J. Inductive Representation Learning on Large Graphs[C]// Proceedings of the 31st International Conference on Neural Information Processing Systems. 2017: 1025-1035.
[16]
Ying R, He R N, Chen K F, et al. Graph Convolutional Neural Networks for Web-Scale Recommender Systems[C]// Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining. 2018: 974-983.
[17]
Wang X, Chai Y B, Li H, et al. Link Prediction in Heterogeneous Information Networks: An Improved Deep Graph Convolution Approach[J]. Decision Support Systems, 2021, 141: 113448.
doi: 10.1016/j.dss.2020.113448
[18]
van den Berg R, Kipf T N, Welling M. Graph Convolutional Matrix Completion[OL]. arXiv Preprint, arXiv: 1706.02263.
( Su Jing, Xu Tianqi, Zhang Xiankun, et al. Collaborative Filtering Recommendation Model Based on Graph Convolution and Cross Product[J]. Application Research of Computers, 2021, 38(10): 3044-3048.)
[20]
Wang X, He X N, Wang M, et al. Neural Graph Collaborative Filtering[C]// Proceedings of the 42nd International ACM SIGIR Conference on Research and Development in Information Retrieval. 2019: 165-174.
[21]
Shang J Y, Xiao C, Ma T F, et al. GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combination[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 2019, 33: 1126-1133.
doi: 10.1609/aaai.v33i01.33011126
[22]
Jin Y Y, Zhang W, He X N, et al. Syndrome-Aware Herb Recommendation with Multi-Graph Convolution Network[C]// Proceedings of the 36th International Conference on Data Engineering. IEEE, 2020: 145-156.
[23]
Mao C, Yao L, Luo Y,. MedGCN: Graph Convolutional Networks for Multiple Medical Tasks[OL]. arXiv Preprint, arXiv: 1904.00326.
[24]
Liu S C, Li T, Ding H Y, et al. A Hybrid Method of Recurrent Neural Network and Graph Neural Network for Next-Period Prescription Prediction[J]. International Journal of Machine Learning and Cybernetics, 2020, 11(12): 2849-2856.
doi: 10.1007/s13042-020-01155-x
[25]
Schlichtkrull M, Kipf T N, Bloem P, et al. Modeling Relational Data with Graph Convolutional Networks[C]// Proceedings of the 15th Extended Semantic Web Conference. 2018: 593-607.
[26]
Zheng Y, Gao C, He X N, et al. Price-Aware Recommendation with Graph Convolutional Networks[C]// Proceedings of the 36th International Conference on Data Engineering. IEEE, 2020: 133-144.
[27]
Johnson A E W, Pollard T J, Shen L, et al. MIMIC-III, a Freely Accessible Critical Care Database[J]. Scientific Data, 2016, 3: 160035.
doi: 10.1038/sdata.2016.35
[28]
He X N, Liao L Z, Zhang H W, et al. Neural Collaborative Filtering[C]// Proceedings of the 26th International Conference on World Wide Web. 2017: 173-182.
