MedGCN: Medication recommendation and lab test imputation via graph convolutional networks
Open accessAbstract
Laboratory testing and medication prescription are two of the most important routines in daily clinical practice. Developing an artificial intelligence system that can automatically make lab test imputations and medication recommendations can save costs on potentially redundant lab tests and inform physicians of a more effective prescription. We present an intelligent medical system (named MedGCN) that can automatically recommend the patients' medications based on their incomplete lab tests, and can even accurately estimate the lab values that have not been taken. In our system, we integrate the complex relations between multiple types of medical entities with their inherent features in a heterogeneous graph. Then we model the graph to learn a distributed representation for each entity in the graph based on graph convolutional networks (GCN). By the propagation of graph convolutional networks, the entity representations can incorporate multiple types of medical information that can benefit multiple medical tasks. Moreover, we introduce a cross regularization strategy to reduce overfitting for multi-task training by the interaction between the multiple tasks. In this study, we construct a graph to associate 4 types of medical entities, i.e., patients, encounters, lab tests, and medications, and applied a graph neural network to learn node embeddings for medication recommendation and lab test imputation. we validate our MedGCN model on two real-world datasets: NMEDW and MIMIC-III. The experimental results on both datasets demonstrate that our model can outperform the state-of-the-art in both tasks. We believe that our innovative system can provide a promising and reliable way to assist physicians to make medication prescriptions and to save costs on potentially redundant lab tests.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
A cell-to-tissue graph can capture cell morphology and interactions, tissue morphology and spatial distribution, cell-to-tissue hierarchies, and spatial distribution of cells with respect to tissues, and a hierarchical GNN can learn representations from these modalities.
“Further, cell morphology and interactions, tissue morphology and spatial distribution, cell-to-tissue hierarchies, and spatial distribution of cells with respect to tissues can be captured in a cell-to-tissue graph, upon which a hierarchical GNN can learn representations using these different data modalities [189].”
From Graph representation learning in biomedicine and healthcare (2022) · cites this paper as [189]
Deep graph learning has been shown to detect subtle signs of disease progression in medical images while integrating other modalities such as tissue localization and genomic features.
“Deep graph learning has been shown to detect subtle signs of disease progression in the images while integrating other modalities (e.g., tissue localization [189] and genomic features [8]) to improve medical image processing.”
From Graph representation learning in biomedicine and healthcare (2022) · cites this paper as [189]
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Breiman, L. (2001) · Machine Learning · doi:10.1023/a:1010933404324 · Cited by 133,538 · see citing works
Adam: A Method for Stochastic Optimization
Kingma, D. P., & Ba, J. (2014) · arXiv · doi:10.48550/arxiv.1412.6980 · Cited by 98,946 · see citing works
Scikit-learn: Machine Learning in Python
Pedregosa, F., Varoquaux, G., Gramfort, A., Michel, V., Thirion, B., Grisel, O., Blondel, M., Müller, A., Nothman, J., Louppe, G., et al. (2012) · arXiv · doi:10.48550/arxiv.1201.0490 · Cited by 71,016 · see citing works
A Comprehensive Survey on Graph Neural Networks
Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C., & Yu, P. S. (2021) · IEEE Transactions on Neural Networks and Learning Systems · doi:10.1109/tnnls.2020.2978386 · Cited by 11,000 · see citing works
Libsvm : A library for support vector machines
Ferrari, V. (2008) · Medical Entomology and Zoology · Cited by 10,124 · see citing works
MIMIC-III, a freely accessible critical care database
Johnson, A. E. W., Pollard, T. J., Shen, L., Lehman, L.-W. H., Feng, M., Ghassemi, M., Moody, B., Szolovits, P., Anthony Celi, L., & Mark, R. G. (2016) · Scientific Data · doi:10.1038/sdata.2016.35 · Cited by 8,141 · see citing works
Friedman, J. H. (2002) · Computational Statistics & Data Analysis · doi:10.1016/s0167-9473(01)00065-2 · Cited by 7,067 · see citing works
LIBLINEAR: A Library for Large Linear Classification
Fan, R.-E., Chang, K., Hsieh, C., Xiang-rui, W., & Lin, C. (2008) · doi:10.5555/1390681.1442794 · Cited by 6,680 · see citing works
MissForest—Non-parametric missing value imputation for mixed-type data
Stekhoven, D. J., & Bühlmann, P. (2012) · Bioinformatics · doi:10.1093/bioinformatics/btr597 · Cited by 6,156 · see citing works
(2010) · University of Twente Research Information · Cited by 6,037 · see citing works
Modeling Relational Data with Graph Convolutional Networks
Schlichtkrull, M., Kipf, T. N., Bloem, P., van den Berg, R., Titov, I., & Welling, M. (2018) · Lecture Notes in Computer Science · doi:10.1007/978-3-319-93417-4_38 · Cited by 5,570 · see citing works
Targeted Branching for the Maximum Independent Set Problem Using Graph Neural Networks
Silva, G., Rodrigues, M., Teixeira, A., et al. (2024) · DROPS (Schloss Dagstuhl – Leibniz Center for Informatics) · doi:10.4230/lipics.sea.2024.20 · Cited by 5,266 · see citing works
Wang, X., Peng, Y., Lu, L., Lu, Z., Bagheri, M., & Summers, R. M. (2017) · 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr.2017.369 · Cited by 4,616 · see citing works
An Overview of Multi-Task Learning in Deep Neural NetworksPreprint
Ruder, S. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1706.05098 · Cited by 3,195 · see citing works
Heterogeneous Graph Attention Network
Wang, X., Ji, H., Shi, C., Wang, B., Ye, Y., Cui, P., & Yu, P. S. (2019) · The World Wide Web Conference · doi:10.1145/3308558.3313562 · Cited by 3,191 · see citing works
A Comprehensive Survey on Graph Neural Networks
Wu, Z., Pan, S., Chen, F., Long, G., Zhang, C., & Yu, P. (2020) · UTS ePRESS (University of Technology Sydney) · Cited by 3,174 · see citing works
A Survey on Multi-Task Learning
Zhang, Y., & Yang, Q. (2017) · arXiv · doi:10.48550/arxiv.1707.08114 · Cited by 2,910 · see citing works
Deeper Insights Into Graph Convolutional Networks for Semi-Supervised Learning
Li, Q., Han, Z., & Wu, X.-M. (2018) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v32i1.11604 · Cited by 2,872 · see citing works
Classifier chains for multi-label classification
Read, J., Pfahringer, B., Holmes, G., & Frank, E. (2011) · Machine Learning · doi:10.1007/s10994-011-5256-5 · Cited by 2,312 · see citing works
Graph Convolutional Networks for Text Classification
Yao, L., Mao, C., & Luo, Y. (2019) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v33i01.33017370 · Cited by 2,171 · see citing works
Self-citationMiotto, R., Li, L., Kidd, B. A., & Dudley, J. T. (2016) · Scientific Reports · doi:10.1038/srep26094 · Cited by 1,710 · see citing works
Modeling polypharmacy side effects with graph convolutional networks
Zitnik, M., Agrawal, M., & Leskovec, J. (2018) · Bioinformatics · doi:10.1093/bioinformatics/bty294 · Cited by 1,501 · see citing works
Graph Convolutional Matrix CompletionPreprint
van den Berg, R., Kipf, T., & Welling, M. (2017) · arXiv (Cornell University) · doi:10.48550/arxiv.1706.02263 · Cited by 1,023 · see citing works
RhizoDB: A Bounded Activation-Flow Architecture for Graph-Based Memory Systems
Guberte, J. A. (2026) · Zenodo (CERN European Organization for Nuclear Research) · doi:10.5281/zenodo.20695702 · Cited by 897 · see citing works
(2016) · PubMed · Cited by 576 · see citing works
Geometric Matrix Completion with Recurrent Multi-Graph Neural NetworksPreprint
Monti, F., Bronstein, M., & Bresson, X. (2017) · Neural Information Processing Systems · doi:10.48550/arxiv.1704.06803 · Cited by 558 · see citing works
Comparison of imputation methods for missing laboratory data in medicine
Waljee, A. K., Mukherjee, A., Singal, A. G., Zhang, Y., Warren, J. S., Balis, U. J., Marrero, J. A., Zhu, J., & Higgins, P. (2013) · BMJ Open · doi:10.1136/bmjopen-2013-002847 · Cited by 495 · see citing works
Multi-layer Representation Learning for Medical Concepts
Choi, E., Bahadori, M. T., Searles, E., Coffey, C., Thompson, M., Bost, J., Tejedor-Sojo, J., & Sun, J. (2016) · Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining · doi:10.1145/2939672.2939823 · Cited by 469 · see citing works
GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combination
Shang, J., Xiao, C., Ma, T., Li, H., & Sun, J. (2019) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v33i01.33011126 · Cited by 345 · see citing works
Pre-training of Graph Augmented Transformers for Medication Recommendation
Shang, J., Ma, T., Xiao, C., & Sun, J. (2019) · Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence · doi:10.24963/ijcai.2019/825 · Cited by 311 · see citing works
Zhang, Y., Chen, R. F., Tang, J., Stewart, W. F., & Sun, J. (2017) · ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) · doi:10.1145/3097983.3098109 · Cited by 224 · see citing works
Computational Medicine: Translating Models to Clinical Care
Winslow, R. L., Trayanova, N. A., Geman, D., & Miller, M. I. (2012) · Science Translational Medicine · doi:10.1126/scitranslmed.3003528 · Cited by 202 · see citing works
Tran, T., Nguyen, T. D., Phung, D., & Venkatesh, S. (2015) · Journal of Biomedical Informatics · doi:10.1016/j.jbi.2015.01.012 · Cited by 172 · see citing works
Weisfeiler and Leman Go Neural: Higher-Order Graph Neural Networks
Morris, C., Ritzert, M., Fey, M., Hamilton, W. L., Lenssen, J. E., Rattan, G., & Grohe, M. (2019) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v33i01.33014602 · Cited by 172 · see citing works
Ten things we have to do to achieve precision medicine
Kohane, I. S. (2015) · Science · doi:10.1126/science.aab1328 · Cited by 162 · see citing works
Using Machine Learning to Predict Laboratory Test Results
Luo, Y., Szolovits, P., Dighe, A. S., & Baron, J. M. (2016) · American Journal of Clinical Pathology · doi:10.1093/ajcp/aqw064 · Cited by 162 · see citing works
Self-citationAssociation of Albuminuria and Cancer Incidence
Jørgensen, L., Heuch, I., Jenssen, T. G., & Jacobsen, B. K. (2008) · Journal of the American Society of Nephrology · doi:10.1681/asn.2007060712 · Cited by 159 · see citing works
Dual Memory Neural Computer for Asynchronous Two-view Sequential Learning
Lê, H., Tran, T., & Venkatesh, S. (2018) · ACM SIGKDD Conference on Knowledge Discovery and Data Mining (KDD) · doi:10.1145/3219819.3219981 · Cited by 121 · see citing works
Luo, Y., Peter Szolovits, B., Dighe, A. S., & Baron, J. M. (2017) · Journal of the American Medical Informatics Association · doi:10.1093/jamia/ocx133 · Cited by 108 · see citing works
Self-citationMao, C., Yao, L., & Luo, Y. (2022) · IEEE Transactions on Medical Imaging · doi:10.1109/tmi.2022.3153322 · Cited by 101 · see citing works
Self-citationDisease Prediction via Graph Neural Networks
Sun, Z., Yin, H., Chen, H., Chen, T., Cui, L., & Yang, F. (2021) · IEEE Journal of Biomedical and Health Informatics · doi:10.1109/jbhi.2020.3004143 · Cited by 98 · see citing works
Zhang, X. S., He, L., Chen, K., Luo, Y., Zhou, J., & Wang, F. (2018) · Europe PMC (PubMed Central) · doi:10.48550/arxiv.1805.08801 · Cited by 93 · see citing works
Self-citationSyndrome-aware Herb Recommendation with Multi-Graph Convolution Network
Jin, Y., Zhang, W., He, X., Wang, X., & Wang, X. (2020) · IEEE International Conference on Data Engineering (ICDE) · doi:10.1109/icde48307.2020.00020 · Cited by 92 · see citing works
Starren, J. B., Winter, A., & Lloyd‐Jones, D. M. (2015) · Clinical and Translational Science · doi:10.1111/cts.12294 · Cited by 87 · see citing works
Heterogeneous Similarity Graph Neural Network on Electronic Health Records
Liu, Z., Li, X., Peng, H. M., He, L., & Yu, P. S. (2020) · IEEE International Conference on Big Data (BigData) · doi:10.1109/bigdata50022.2020.9377795 · Cited by 76 · see citing works
Karczewski, K. J., Daneshjou, R., & Altman, R. B. (2012) · PLoS Computational Biology · doi:10.1371/journal.pcbi.1002817 · Cited by 65 · see citing works
Li, Y., Jin, R., & Luo, Y. (2018) · Journal of the American Medical Informatics Association · doi:10.1093/jamia/ocy157 · Cited by 64 · see citing works
Self-citationHsieh, K.-L., Wang, Y., Chen, L., Zhao, Z., Savitz, S., Jiang, X., Tang, J., & Kim, Y. (2020) · Research Square · doi:10.21203/rs.3.rs-114758/v1 · Cited by 38 · see citing works
Deep Generative Classifiers for Thoracic Disease Diagnosis with Chest X-ray Images
Mao, C., Yao, L., Pan, Y., Luo, Y., & Zeng, Z. (2018) · IEEE International Conference on Bioinformatics and Biomedicine (BIBM) · doi:10.1109/bibm.2018.8621107 · Cited by 34 · see citing works
Self-citationHarmonized representation learning on dynamic EHR graphs
Lee, D., Jiang, X., & Yu, H. (2020) · Journal of Biomedical Informatics · doi:10.1016/j.jbi.2020.103426 · Cited by 34 · see citing works
A functional multiple imputation approach to incomplete longitudinal data
He, Y., Yucel, R. M., & Raghunathan, T. E. (2011) · Statistics in Medicine · doi:10.1002/sim.4201 · Cited by 32 · see citing works
MGATRx: Discovering Drug Repositioning Candidates Using Multi-View Graph Attention
Yella, J., & Jegga, A. G. (2021) · IEEE Transactions on Computational Biology and Bioinformatics · doi:10.1109/tcbb.2021.3082466 · Cited by 25 · see citing works
Kliethermes, S., & Oleson, J. (2014) · Statistics in Medicine · doi:10.1002/sim.6166 · Cited by 13 · see citing works
Predict or draw blood: An integrated method to reduce lab tests
Yu, L., Zhang, Q., Bernstam, E. V., & Jiang, X. (2020) · Journal of Biomedical Informatics · doi:10.1016/j.jbi.2020.103394 · Cited by 13 · see citing works
Neural Networks for Clinical Order Decision Support.
Wang, J. X., Sullivan, D. K., Wells, A. J., Wells, A. C., & Chen, J. H. (2019) · PubMed Central · Cited by 8 · see citing works
Chen, T., Wang, Z., Li, L., Li, D., Li, Z., Chang, X., Bian, P., Liang, G., Wang, Q., & Xie, T. (2025) · Schloss Dagstuhl – Leibniz-Zentrum für Informatik · doi:10.4230/lipics.ecoop.2025.6
Flagged first, then by citations
Research works printed but not linked: 5
- [6]Zhang Yutao, Chen Robert, Tang Jie, Stewart Walter F, and Sun Jimeng. Leap: learning to prescribe effective and safe treatment combinations for multimorbidity. In proceedings of the 23rd ACM SIGKDD international conference on knowledge Discovery and data Mining, pages 1315– 1324. ACM, 2017.
- [12]Kipf Thomas N and Welling Max. Semi-supervised classification with graph convolutional networks. In International Conference on Learning Representations, 2017.
- [13]Hamilton Will, Ying Zhitao, and Leskovec Jure. Inductive representation learning on large graphs. In Advances in neural information processing systems, pages 1024– 1034, 2017.
- [14]Chen Jie, Ma Tengfei, and Xiao Cao. FastGCN: Fast learning with graph convolutional networks via importance sampling. In International Conference on Learning Representations, 2018.
- [38]Hamilton William L, Ying Rex, and Leskovec Jure. Representation learning on graphs: Methods and applications. IEEE Data Engineering Bulletin, 40(3): 52– 74, 2017.
Cited by
Sorted by citations
Graph representation learning in biomedicine and healthcare
Li, M. M., Huang, K., & Zitnik, M. (2022) · Nature Biomedical Engineering · doi:10.1038/s41551-022-00942-x · Cited by 294 · see citing works
“For example, low-dimensional embed-dings of EHR data can be generated by separately considering medical services, doctors and patients in shallow network embeddings and GNNs 188,189 .”
- Authors
- Mao, C., Yao, L., & Luo, Y.
- Affiliations
- Northwestern University (US)
- Corresponding author
- Luo, Yuan
- Venue
- Journal of Biomedical Informatics
- Publisher
- Elsevier BV
- Published
- 2022-03
- Volume
- 127
- Pages
- 104000
- ISSN
- 1532-0464 (ISSN-L)
- PMC
- PMC8901567
- OpenAlex
- W4210277067
- Cited by
- 70
- References
- 56
- Citations by year
- 2026: 14 · 2025: 18 · 2024: 14 · 2023: 13 · 2022: 11
- FWCI
- 4.61
- Type
- journal article
- Language
- en
- Open access
- bronze
- doi.org
- Licence
- www.elsevier.com/tdm/userlicense/1.0/ (tdm)
- www.elsevier.com/legal/tdmrep-license (tdm)
- www.elsevier.com/open-access/userlicense/1.0/ (vor, 365 d)
- doi.org/10.15223/policy-017 (stm-asf)
- doi.org/10.15223/policy-037 (stm-asf)
- doi.org/10.15223/policy-012 (stm-asf)
- doi.org/10.15223/policy-029 (stm-asf)
- doi.org/10.15223/policy-004 (stm-asf)
- Locations
- Journal of Biomedical Informatics (publishedVersion)
- SSRN Electronic Journal (acceptedVersion)
- PubMed (publishedVersion)
- PubMed Central (acceptedVersion)
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Funders
- National Institutes of Health (5UL1TR001422, N1R21LM012618)
- Topics
- Physical Sciences › Computer Science › Artificial Intelligence › Machine Learning in Healthcare
- Physical Sciences › Computer Science › Artificial Intelligence › Advanced Graph Neural Networks
- Physical Sciences › Computer Science › Information Systems › Recommender Systems and Techniques
- Keywords
- drug recommendation, graph convolutional networks, heterogeneous graphs, multi-task learning, node embeddings, clinical decision support
- MeSH terms
- Artificial Intelligence, Humans, Learning, Neural Networks, Computer
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review