mlsanityData to 14 Oct 2026

Next Visit Diagnosis Prediction via Medical Code-Centric Multimodal Contrastive EHR Modelling with Hierarchical Regularisation

arXiv 2401.11648 ↗First posted 22 Jan 2024cs.LG cs.AI cs.IRcited by 5

Not on the map. Why?

Abstract (arXiv)

Predicting next visit diagnosis using Electronic Health Records (EHR) is an essential task in healthcare, critical for devising proactive future plans for both healthcare providers and patients. Nonetheless, many preceding studies have not sufficiently addressed the heterogeneous and hierarchical characteristics inherent in EHR data, inevitably leading to sub-optimal performance. To this end, we propose NECHO, a novel medical code-centric multimodal contrastive EHR learning framework with hierarchical regularisation. First, we integrate multifaceted information encompassing medical codes, demographics, and clinical notes using a tailored network design and a pair of bimodal contrastive losses, all of which pivot around a medical codes representation. We also regularise modality-specific encoders using a parental level information in medical ontology to learn hierarchical structure of EHR data. A series of experiments on MIMIC-III data demonstrates effectiveness of our approach.

Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 28 of its 40 references that are in the map. First posted on arXiv 22 Jan 2024.

Builds on

28 of 40 references.

References · 28

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  4. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  5. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  6. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  7. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  8. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  9. 2016Layer Normalizationcited by 5,669
  10. 2014Convolutional Neural Networks for Sentence Classificationcited by 1,127
  11. 2019Publicly Available Clinical BERT Embeddingscited by 1,002
  12. 2019Multimodal Transformer for Unaligned Multimodal Language Sequencescited by 645
16 more
  1. 2020Contrastive Learning of Medical Visual Representations from Paired Images and Textcited by 641
  2. 2017Tensor Fusion Network for Multimodal Sentiment Analysiscited by 433
  3. 2016RETAIN: An Interpretable Predictive Model for Healthcare using Reverse Time Attention Mechanismcited by 349
  4. 2018xDeepFM: Combining Explicit and Implicit Feature Interactions for Recommender Systemscited by 278
  5. 2018Efficient Low-rank Multimodal Fusion with Modality-Specific Factorscited by 270
  6. 2015Doctor AI: Predicting Clinical Events via Recurrent Neural Networkscited by 208
  7. 2021Learning Modality-Specific Representations with Self-Supervised Multi-Task Learning for Multimodal Sentiment Analysiscited by 180
  8. 2016GRAM: Graph-based Attention Model for Healthcare Representation Learningcited by 156
  9. 2017Dipole: Diagnosis Prediction in Healthcare via Attention-based Bidirectional Recurrent Neural Networkscited by 116
  10. 2019Using Clinical Notes with Time Series Data for ICU Managementcited by 38
  11. 2021Collaborative Graph Learning with Auxiliary Text for Temporal Event Prediction in Healthcarecited by 30
  12. 2021How to Leverage Multimodal EHR Data for Better Medical Predictions?cited by 26
  13. 2020Characterizing the Value of Information in Medical Notescited by 12
  14. 2022On the Importance of Clinical Notes in Multi-modal Learning for EHR Datacited by 10
  15. 2021MIPO: Mutual Integration of Patient Journey and Medical Ontology for Healthcare Representation Learningcited by 4
  16. 2022InterMulti:Multi-view Multimodal Interactions with Text-dominated Hierarchical High-order Fusion for Emotion Analysiscited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 2; 2025: 1; 2026: 2.

5 papers.