# Medical Concept Representation Learning from Electronic Health Records and its Application on Heart Failure Prediction

arXiv: https://arxiv.org/abs/1602.03686 · HTML: https://mlsanity.com/paper/1602.03686/

## Key facts

- Cited by 9 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 11 Feb 2016.

## Built on by (most cited first, 9 of 9)

- [Unsupervised Representation Learning for Time Series with Temporal Neighborhood Coding](https://mlsanity.com/paper/2106.00750/) (2021)
- [Time Series Prediction using Deep Learning Methods in Healthcare](https://mlsanity.com/paper/2108.13461/) (2021)
- [AttDMM: An Attentive Deep Markov Model for Risk Scoring in Intensive Care Units](https://mlsanity.com/paper/2102.04702/) (2021)
- [A review of Generative Adversarial Networks for Electronic Health Records: applications, evaluation measures and data sources](https://mlsanity.com/paper/2203.07018/) (2022)
- [Compensating trajectory bias for unsupervised patient stratification using adversarial recurrent neural networks](https://mlsanity.com/paper/2112.07239/) (2021)
- [Assessing mortality prediction through different representation models based on concepts extracted from clinical notes](https://mlsanity.com/paper/2207.10872/) (2022)
- [A Study into patient similarity through representation learning from medical records](https://mlsanity.com/paper/2104.14229/) (2021)
- [Learning to Adapt Clinical Sequences with Residual Mixture of Experts](https://mlsanity.com/paper/2204.02687/) (2022)
- [An Interpretable Deep-Learning Framework for Predicting Hospital Readmissions From Electronic Health Records](https://mlsanity.com/paper/2310.10187/) (2023)

## Abstract (arXiv)

Objective: To transform heterogeneous clinical data from electronic health records into clinically meaningful constructed features using data driven method that rely, in part, on temporal relations among data. Materials and Methods: The clinically meaningful representations of medical concepts and patients are the key for health analytic applications. Most of existing approaches directly construct features mapped to raw data (e.g., ICD or CPT codes), or utilize some ontology mapping such as SNOMED codes. However, none of the existing approaches leverage EHR data directly for learning such concept representation. We propose a new way to represent heterogeneous medical concepts (e.g., diagnoses, medications and procedures) based on co-occurrence patterns in longitudinal electronic health records. The intuition behind the method is to map medical concepts that are co-occuring closely in time to similar concept vectors so that their distance will be small. We also derive a simple method to construct patient vectors from the related medical concept vectors. Results: For qualitative evaluation, we study similar medical concepts across diagnosis, medication and procedure. In quantitative evaluation, our proposed representation significantly improves the predictive modeling performance for onset of heart failure (HF), where classification methods (e.g. logistic regression, neural network, support vector machine and K-nearest neighbors) achieve up to 23% improvement in area under the ROC curve (AUC) using this proposed representation. Conclusion: We proposed an effective method for patient and medical concept representation learning. The resulting representation can map relevant concepts together and also improves predictive modeling performance.
