Deep Semi-Supervised Embedded Clustering (DSEC) for Stratification of Heart Failure Patients

Abstract

Determining phenotypes of diseases can have considerable benefits for in-hospital patient care and to drug development. The structure of high dimensional data sets such as electronic health records are often represented through an embedding of the data, with clustering methods used to group data of similar structure. If subgroups are known to exist within data, supervised methods may be used to influence the clusters discovered. We propose to extend deep embedded clustering to a semi-supervised deep embedded clustering algorithm to stratify subgroups through known labels in the data. In this work we apply deep semi-supervised embedded clustering to determine data-driven patient subgroups of heart failure from the electronic health records of 4,487 heart failure and control patients. We find clinically relevant clusters from an embedded space derived from heterogeneous data. The proposed algorithm can potentially find new undiagnosed subgroups of patients that have different outcomes, and, therefore, lead to improved treatments.

6 pages, 2 figures, HSYS workshop at ICML conference

References in corpus (6)

• Deep EHR: A Survey of Recent Advances in Deep Learning Techniques for Electronic Health Record (EHR) Analysis
• Patient2Vec: A Personalized Interpretable Deep Representation of the Longitudinal Electronic Health Record
• Medical Concept Representation Learning from Electronic Health Records and its Application on Heart Failure Prediction
• BEHRT: Transformer for Electronic Health Records
• Embedding Electronic Health Records for Clinical Information Retrieval
• Learning Hierarchical Representations of Electronic Health Records for Clinical Outcome Prediction