Skip to yearly menu bar Skip to main content

Workshop: Medical Imaging meets NeurIPS

Self-Supervised Contrastive Learning for Electrocardiograms to Detect Left Ventricular Systolic Dysfunction

Mitsuhiko Nakamoto · Hirotoshi Takeuchi


Self-supervised learning has been demonstrated to be a powerful way to use unlabeled data in computer vision tasks. In this study, we propose a self-supervised pretraining approach to improve the performance of deep learning models that detect left ventricular systolic dysfunction from 12-lead electrocardiography data. We first pretrain an encoder that can extract rich features from unlabeled electrocardiography data using self-supervised contrastive learning, and then fine-tune the model on the downstream dataset using the pretrained encoder. In experiments, our proposed approach achieved higher performance than the supervised baseline method, using only 28% of the labels used by the baseline method.

Chat is not available.