|
|
Drug Recommendation Based on Graph Neural Network with Patient Signs and Medication Data |
Cheng Quan( ),She Dexin |
School of Economics and Management, Fuzhou University, Fuzhou 350108, China |
|
|
Abstract [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.
|
Received: 25 December 2021
Published: 26 October 2022
|
|
Fund:National Social Science Fund of China(19BTQ072) |
Corresponding Authors:
Cheng Quan,ORCID:0000-0002-7302-4527
E-mail: chengquan@fzu.edu.cn
|
[1] |
陈耶拉, 耿秀丽. 基于改进协同过滤的个性化产品服务系统方案推荐[J]. 计算机集成制造系统, 2021, 27(1): 240-248.
|
[1] |
( Chen Yela, Geng Xiuli. Recommendation of Personalized Product-Service System Scheme Based on Improved Collaborative Filtering[J]. Computer Integrated Manufacturing Systems, 2021, 27(1): 240-248.)
|
[2] |
黄立威, 江碧涛, 吕守业, 等. 基于深度学习的推荐系统研究综述[J]. 计算机学报, 2018, 41(7): 1619-1647.
|
[2] |
( 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.)
|
[3] |
彭春婷, 阮冰. 实验室检查对感染性发热的诊断价值[J]. 中国实用内科杂志, 2016, 36(12): 1025-1028.
doi: 10.7504/nk2016110102
|
[3] |
( 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
|
[4] |
李鹏飞, 鲁法明, 包云霞, 等. 基于医疗过程挖掘与患者体征的药物推荐方法[J]. 计算机集成制造系统, 2020, 26(6): 1668-1678.
|
[4] |
( 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.
|
[19] |
苏静, 许天琪, 张贤坤, 等. 基于图卷积与外积的协同过滤推荐模型[J]. 计算机应用研究, 2021, 38(10): 3044-3048.
|
[19] |
( 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.
|
|
Viewed |
|
|
|
Full text
|
|
|
|
|
Abstract
|
|
|
|
|
Cited |
|
|
|
|
|
Shared |
|
|
|
|
|
Discussed |
|
|
|
|