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.

Knowledge-Enhanced Difference-Aware Clinical Time Series Representation Learning for Diagnosis Prediction.

Knowledge guided diagnosis prediction via graph spatial-temporal network

Stage-Aware Hierarchical Attentive Relational Network for Diagnosis Prediction

Predictive Modeling with Temporal Graphical Representation on Electronic Health Records

Reasoning-Enhanced Healthcare Predictions with Knowledge Graph Community Retrieval

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

Graph Neural Network-Based Diagnosis Prediction

Mdpg: a novel multi-disease diagnosis prediction method based on patient knowledge graphs

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

Deep Knowledge Reasoning Guided Disease Prediction.

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

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

SeqCare: Sequential Training with External Medical Knowledge Graph for Diagnosis Prediction in Healthcare Data

Graph-based structural knowledge-aware network for diagnosis assistant

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

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

An Effective Patient Representation Learning for Time-series Prediction Tasks Based on EHRs

Representation learning for clinical time series prediction tasks in electronic health records

RDKG: A Reinforcement Learning Framework for Disease Diagnosis on Knowledge Graph

K-Diag: Knowledge-enhanced Disease Diagnosis in Radiographic Imaging

Interpretable Representation Learning for Healthcare via Capturing Disease Progression through Time