Heart Block Identification from 12-Lead ECG: Exploring the Generalizability of Self-Supervised AI

Timely diagnosis and treatment of heart blocks are critical for preventing fatal outcomes in patients with cardiac conduction disorders. Expert analysis of clinical 12-lead electro-cardiograms (ECG) remains the standard diagnosis apparatus. In this study, we propose a self-supervised deep learning model that can detect evidence of heart blocks in ECG signals. We build a multichannel residual neural network (KRes) and train this model in a self-supervised contrastive fashion leveraging 3.6 million ECG from patients at the Massachusetts General Hospital (MGH) to learn robust ECG representations. We evaluate the utility of such representations on a large public dataset, PTB-XL, through fine-tuning and linear probing for identifying heart blocks. To compare, we also build a baseline supervised model by adapting KRes. The training data contains a subset of 10.6 thou-sand PTB-XL ECG from patients with and without heart block patterns, and a holdout set of 1319 ECG is used for evaluation. We compare the performances of the finetuning and the linear probing of the self-supervised representations with that of the supervised model using the area under the receiver-operating curve (AUC), sensitivity, specificity, and predictive values, and found those performing equally well. Moreover, we repeat the training cycles of the three pipelines while reducing the number of training samples and demonstrate that the performance of the self-supervised representations remain steady even when labeled data is limited. This research underscores the potential of self-supervised learning in cardiac diagnostics, emphasizing generalizability and performance across diverse datasets and clinical settings, especially in data-scarce paradigms.

Authors

Institutions

Publication Details

Published
2024-10-15
DOI
https://doi.org/10.1109/bsn63547.2024.10780740
Primary Topic
Brain Tumor Detection and Classification
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Heart Block Identification from 12-Lead ECG: Exploring the Generalizability of Self-Supervised AI

Collin M. Stultz, Ridwan Alam, Teya S. Bergamaschi
Brain Tumor Detection and Classification
article

Heart Block Identification from 12-Lead ECG: Exploring the Generalizability of Self-Supervised AI

Collin M. Stultz, Ridwan Alam, Teya S. Bergamaschi
article en

Abstract

Timely diagnosis and treatment of heart blocks are critical for preventing fatal outcomes in patients with cardiac conduction disorders. Expert analysis of clinical 12-lead electro-cardiograms (ECG) remains the standard diagnosis apparatus. In this study, we propose a self-supervised deep learning model that can detect evidence of heart blocks in ECG signals. We build a multichannel residual neural network (KRes) and train this model in a self-supervised contrastive fashion leveraging 3.6 million ECG from patients at the Massachusetts General Hospital (MGH) to learn robust ECG representations. We evaluate the utility of such representations on a large public dataset, PTB-XL, through fine-tuning and linear probing for identifying heart blocks. To compare, we also build a baseline supervised model by adapting KRes. The training data contains a subset of 10.6 thou-sand PTB-XL ECG from patients with and without heart block patterns, and a holdout set of 1319 ECG is used for evaluation. We compare the performances of the finetuning and the linear probing of the self-supervised representations with that of the supervised model using the area under the receiver-operating curve (AUC), sensitivity, specificity, and predictive values, and found those performing equally well. Moreover, we repeat the training cycles of the three pipelines while reducing the number of training samples and demonstrate that the performance of the self-supervised representations remain steady even when labeled data is limited. This research underscores the potential of self-supervised learning in cardiac diagnostics, emphasizing generalizability and performance across diverse datasets and clinical settings, especially in data-scarce paradigms.

Massachusetts General Hospital (US), Massachusetts Institute of Technology (US)
Gender equality
Openalex Percentile: Top 74%
Brain Tumor Detection and Classification
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.