Leveraging graph-based hierarchical medical entity embedding for healthcare applications

被引:13
|
作者
Wu, Tong [1 ]
Wang, Yunlong [1 ]
Wang, Yue [1 ]
Zhao, Emily [1 ]
Yuan, Yilian [1 ]
机构
[1] IQVIA Inc, Adv Analyt, Plymouth Meeting, PA 19462 USA
关键词
PREDICTION; RECORDS; DISEASE;
D O I
10.1038/s41598-021-85255-w
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Automatic representation learning of key entities in electronic health record (EHR) data is a critical step for healthcare data mining that turns heterogeneous medical records into structured and actionable information. Here we propose ME2Vec, an algorithmic framework for learning continuous low-dimensional embedding vectors of the most common entities in EHR: medical services, doctors, and patients. ME2Vec features a hierarchical structure that encapsulates different node embedding schemes to cater for the unique characteristic of each medical entity. To embed medical services, we employ a biased-random-walk-based node embedding that leverages the irregular time intervals of medical services in EHR to embody their relative importance. To embed doctors and patients, we adhere to the principle "it's what you do that defines you" and derive their embeddings based on their interactions with other types of entities through graph neural network and proximity-preserving network embedding, respectively. Using real-world clinical data, we demonstrate the efficacy of ME2Vec over competitive baselines on diagnosis prediction, readmission prediction, as well as recommending doctors to patients based on their medical conditions. In addition, medical service embeddings pretrained using ME2Vec can substantially improve the performance of sequential models in predicting patients clinical outcomes. Overall, ME2Vec can serve as a general-purpose representation learning algorithm for EHR data and benefit various downstream tasks in terms of both performance and interpretability.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] Leveraging graph-based hierarchical medical entity embedding for healthcare applications
    Tong Wu
    Yunlong Wang
    Yue Wang
    Emily Zhao
    Yilian Yuan
    Scientific Reports, 11
  • [2] Hierarchical stochastic graphlet embedding for graph-based pattern recognition
    Anjan Dutta
    Pau Riba
    Josep Lladós
    Alicia Fornés
    Neural Computing and Applications, 2020, 32 : 11579 - 11596
  • [3] Hierarchical stochastic graphlet embedding for graph-based pattern recognition
    Dutta, Anjan
    Riba, Pau
    Llados, Josep
    Fornes, Alicia
    NEURAL COMPUTING & APPLICATIONS, 2020, 32 (15): : 11579 - 11596
  • [4] Efficient Medical Knowledge Graph Embedding: Leveraging Adaptive Hierarchical Transformers and Model Compression
    Li, Xuexiang
    Yang, Hansheng
    Yang, Cong
    Zhang, Weixing
    ELECTRONICS, 2023, 12 (10)
  • [5] Enhancement of Medical Named Entity Recognition Using Graph-based Features
    Keretna, Sara
    Lim, Chee Peng
    Creighton, Doug
    2015 IEEE INTERNATIONAL CONFERENCE ON SYSTEMS, MAN, AND CYBERNETICS (SMC 2015): BIG DATA ANALYTICS FOR HUMAN-CENTRIC SYSTEMS, 2015, : 1895 - 1900
  • [6] Leveraging deep graph-based text representation for sentiment polarity applications
    Bijari, Kayvan
    Zare, Hadi
    Kebriaei, Emad
    Veisi, Hadi
    EXPERT SYSTEMS WITH APPLICATIONS, 2020, 144
  • [7] A Knowledge Graph Entity Disambiguation Method Based on Entity-Relationship Embedding and Graph Structure Embedding
    Ma, Jiangtao
    Li, Duanyang
    Chen, Yonggang
    Qiao, Yaqiong
    Zhu, Haodong
    Zhang, Xuncai
    COMPUTATIONAL INTELLIGENCE AND NEUROSCIENCE, 2021, 2021 (2021)
  • [8] Graph-based hierarchical conceptual clustering
    Jonyer, I
    Cook, DJ
    Holder, LB
    JOURNAL OF MACHINE LEARNING RESEARCH, 2002, 2 (01) : 19 - 43
  • [9] Evaluating hierarchical graph-based segmentation
    Haxhimusa, Yll
    Ion, Adrian
    Kropatsch, Walter G.
    18TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION, VOL 2, PROCEEDINGS, 2006, : 195 - +
  • [10] Graph-Based Hierarchical Video Cosegmentation
    Rodrigues, Franciele
    Leal, Pedro
    Kenmochi, Yukiko
    Cousty, Jean
    Najman, Laurent
    Guimaraes, Silvio
    Patrocinio, Zenilton, Jr.
    IMAGE ANALYSIS AND PROCESSING,(ICIAP 2017), PT I, 2017, 10484 : 15 - 26