MHGRL: An Effective Representation Learning Model for Electronic Health Records

所属单位：

School of Informatics, Xiamen University,

发表刊物：

LREC-COLING 2024

关键字：

"representation learning";"multimodal heterogeneous graph";"contrastive learning"

摘要：

Electronic health records (EHRs) serve as a digital repository storing comprehensive medical information about patients. Representation learning for EHRs plays a crucial role in healthcare applications. In this paper, we propose a Multimodal Heterogeneous Graph-enhanced Representation Learning, denoted as MHGRL, aimed at learning effective EHR representations. To address the challenge posed by data insufficiency of EHRs, MHGRL utilizes a multimodal heterogeneous graph to model an EHR. Specifically, we construct a heterogeneous graph for each EHR and enrich it by incorporating multimodal information with medical ontology and textual notes. With the integration of pre-trained model, graph neural network, and attention mechanism, MHGRL effectively incorporates both node attributes and structural information across a multimodal heterogeneous graph. Moreover, we employ contrastive learning to ensure the consistency of representations for similar EHRs and improve the model robustness. The experimental results show that MHGRL outperforms all baselines on two real clinical datasets in downstream tasks, including EHR clustering and disease prediction. The code is available at https://github.com/emmali808/MHGRL.

第一作者：

Feiyan Liu

通讯作者：

Xiaoli Wang

合写作者：

LiangZhi Li,Chang Liu,Feng Luo,Jinsong Su,Yiming Qian

论文类型：

Original Research

学科门类：

医疗人工智能 / 计算机科学 / 多模态 / 视觉语言模型

文献类型：

Conference Paper

是否译文：

否

发表时间：

2024-05-01