Abstract:In recent years, with the better availability of medical data such as Electronic Health Records (EHR), more and more data mining models have been developed to explore the data-driven insights for better human health. However, there are many challenges for analyzing EHR such as high-dimensionality, temporality, sparsity, etc., which make the data-driven models less reliable. Medical knowledge graph (MKG), which encodes comprehensive knowledge about the medical concepts and relationships extracted from medical literature, holds great promise to regularize the data-driven models as prior knowledge. Nonetheless, the MKGs are typically not complete, which limits its utility in helping with the data mining process. In this paper, we propose a mutual enhancement framework MendMKG for predictive modeling of clinical events with both EHR and MKG. In particular, MendMKG first conducts a self-supervised learning strategy to simultaneously pre-train a graph attention network for embedding nodes and complete the MKG. It iteratively performs (1) an embedding-based knowledge graph completion module to derive missing edges, (2) and a reconstruction module of unlabeled EHR data to select high-quality ones from these edges, which would be further appended to the MKG to update the embedding model. Through the iterations, the two modules mutually benefit each other. Then, MendMKG uses the pre-trained graph attention network and the updated MKG to generate the visit embeddings to represent patient’s historical visits, and predict the diagnosis in future visit, through a fine-tuning approach. Experimental results on real world EHR corpus are provided to demonstrate the superiority of the proposed framework, compared to a series of state-of-the-art baselines.1 1The source code and knowledge graph data have been anonymously uploaded to https://github.com/1317375434/MendMKG. The source code and knowledge graph data have been anonymously uploaded to https://github.com/1317375434/MendMKG.

Graph learning with label attention and hyperbolic embedding for temporal event prediction in healthcare

Collaborative Graph Learning with Auxiliary Text for Temporal Event Prediction in Healthcare

Time-aware Heterogeneous Graph Transformer with Adaptive Attention Merging for Health Event Prediction

Predictive Modeling with Temporal Graphical Representation on Electronic Health Records

HealthGAT: Node Classifications in Electronic Health Records using Graph Attention Networks

Predicting Physiological Response in Heart Failure Management: A Graph Representation Learning Approach using Electronic Health Records

Heterogeneous graph construction and HinSAGE learning from electronic medical records

A hybrid method of recurrent neural network and graph neural network for next-period prescription prediction

GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs

Multi-gate Mixture of Multi-view Graph Contrastive Learning on Electronic Health Record

Time-aware Context-Gated Graph Attention Network for Clinical Risk Prediction

MEGACare: Knowledge-guided multi-view hypergraph predictive framework for healthcare

Graph Transformers on EHRs: Better Representation Improves Downstream Performance

Multi-task Heterogeneous Graph Learning on Electronic Health Records

$\mathtt{MedGraph:}$ Structural and Temporal Representation Learning of Electronic Medical Records

Predictive Modeling of Clinical Events with Mutual Enhancement Between Longitudinal Patient Records and Medical Knowledge Graph

Modeling Long-term Dependencies and Short-term Correlations in Patient Journey Data with Temporal Attention Networks for Health Prediction

Temporal Cross-Attention for Dynamic Embedding and Tokenization of Multimodal Electronic Health Records

Learning the Joint Representation of Heterogeneous Temporal Events for Clinical Endpoint Prediction

Graph Neural Network Based Multi-Label Hierarchical Classification for Disease Predictions in General Practice.

Context-Aware Health Event Prediction via Transition Functions on Dynamic Disease Graphs