Alert button
Picture for Yuesong Zou

Yuesong Zou

Alert button

Modeling electronic health record data using a knowledge-graph-embedded topic model

Jun 03, 2022
Yuesong Zou, Ahmad Pesaranghader, Aman Verma, David Buckeridge, Yue Li

Figure 1 for Modeling electronic health record data using a knowledge-graph-embedded topic model
Figure 2 for Modeling electronic health record data using a knowledge-graph-embedded topic model
Figure 3 for Modeling electronic health record data using a knowledge-graph-embedded topic model
Figure 4 for Modeling electronic health record data using a knowledge-graph-embedded topic model

The rapid growth of electronic health record (EHR) datasets opens up promising opportunities to understand human diseases in a systematic way. However, effective extraction of clinical knowledge from the EHR data has been hindered by its sparsity and noisy information. We present KG-ETM, an end-to-end knowledge graph-based multimodal embedded topic model. KG-ETM distills latent disease topics from EHR data by learning the embedding from the medical knowledge graphs. We applied KG-ETM to a large-scale EHR dataset consisting of over 1 million patients. We evaluated its performance based on EHR reconstruction and drug imputation. KG-ETM demonstrated superior performance over the alternative methods on both tasks. Moreover, our model learned clinically meaningful graph-informed embedding of the EHR codes. In additional, our model is also able to discover interpretable and accurate patient representations for patient stratification and drug recommendations.

Viaarxiv icon

Hyper-SAGNN: a self-attention based graph neural network for hypergraphs

Nov 06, 2019
Ruochi Zhang, Yuesong Zou, Jian Ma

Figure 1 for Hyper-SAGNN: a self-attention based graph neural network for hypergraphs
Figure 2 for Hyper-SAGNN: a self-attention based graph neural network for hypergraphs
Figure 3 for Hyper-SAGNN: a self-attention based graph neural network for hypergraphs
Figure 4 for Hyper-SAGNN: a self-attention based graph neural network for hypergraphs

Graph representation learning for hypergraphs can be used to extract patterns among higher-order interactions that are critically important in many real world problems. Current approaches designed for hypergraphs, however, are unable to handle different types of hypergraphs and are typically not generic for various learning tasks. Indeed, models that can predict variable-sized heterogeneous hyperedges have not been available. Here we develop a new self-attention based graph neural network called Hyper-SAGNN applicable to homogeneous and heterogeneous hypergraphs with variable hyperedge sizes. We perform extensive evaluations on multiple datasets, including four benchmark network datasets and two single-cell Hi-C datasets in genomics. We demonstrate that Hyper-SAGNN significantly outperforms the state-of-the-art methods on traditional tasks while also achieving great performance on a new task called outsider identification. Hyper-SAGNN will be useful for graph representation learning to uncover complex higher-order interactions in different applications.

Viaarxiv icon