Abstract:Background Deep learning has demonstrated diagnostic performance in electrocardiogram (ECG) analysis comparable to that of clinicians. However, existing models primarily excel in diagnosing common conditions such as arrhythmias and heart blocks, which are easily identifiable using standard predictor variables. Conversely, models aiming to uncover novel insights, such as detecting structural changes or the presence of myocardial scar, often lack generalizability due to insufficient data, model architecture limitations, and other factors. Moreover, these models frequently encounter challenges, such as a decline in performance when applied to external datasets. To address these issues comprehensively, we propose the ECGWiz — a unified model capable of tackling diverse ECG-related problems with consistently high performance. Objective To develop a generalizable Foundation AI model for ECG diagnosis capable of detecting diverse cardiac abnormalities and diseases using few training samples. Methods We utilized publicly available datasets like PTB-XL, CPSC, etc. To build our model, totaling approximately 80,000 ECGs. These 12 lead ECGs are scaled and filtered before using them for training. The AI model is pretrained using Self Supervised Learning and Contrastive Learning strategies. Further, the model undergoes supervised training using a hierarchy of labels that increase in granularity, encompassing arrhythmias, heart blocks, other cardiac abnormalities, heart failure, sudden cardiac death and more. Lastly, the model is prepared for Few-Shot Learning — where the model learns from a minimal set of samples (5 to 10 ECGs) of the target disease. For testing the model, we are using unique datasets like PROSE which consists of non-ischemic cardiomyopathy patients with and without myocardial scaring detected using cardiac MRI imaging at the Johns Hopkins University. Results Our model, trained on a dataset of only 5 ECG recordings, demonstrated a 74% accuracy in identifying myocardial scarring as detected by LGE MRI, utilizing a standard 12-lead ECG. This performance is comparable to that of conventional deep learning methods, which necessitate datasets containing upwards of 120 ECG recordings. Conclusion Our model represents a promising stride in creating a unified and adaptable foundation AI model for comprehensive ECG-based diagnosis. Its capacity to learn from a minimal set of examples signifies a robust framework that can evolve and adapt to new challenges in ECG diagnosis. Ongoing improvements, particularly through dataset expansion, underscore its potential to enhance cardiovascular diagnostics, offering precise and efficient patient care.

Generalization challenges in electrocardiogram deep learning: insights from dataset characteristics and attention mechanism

Improving generalization performance of electrocardiogram classification models

DeepECG: Image-based Electrocardiogram Interpretation with Deep Convolutional Neural Networks

Deep Discriminative Domain Generalization with Adversarial Feature Learning for Classifying ECG Signals

Deep neural networks generalization and fine-tuning for 12-lead ECG classification

ECGWiz: a unified and generalizable AI model for comprehensive ECG diagnosis

Opportunities and Challenges of Deep Learning Methods for Electrocardiogram Data: A Systematic Review

Multi-label Classification of Abnormalities in 12-Lead ECG Using Deep Learning.

Deep Learning for Personalized Electrocardiogram Diagnosis: A Review

Application of an end-to-end model with self-attention mechanism in cardiac disease prediction

Improving Deep Learning-based Cardiac Abnormality Detection in 12-Lead ECG with Data Augmentation

Optimal ECG-lead selection increases generalizability of deep learning on ECG abnormality classification

Image based deep learning in 12-lead ECG diagnosis

Generalising electrocardiogram detection and delineation: training convolutional neural networks with synthetic data augmentation

ECG classification based on guided attention mechanism

A Generalizable Deep Learning System for Cardiac MRI

Cardiologist-level interpretable knowledge-fused deep neural network for automatic arrhythmia diagnosis

Accurate detection of arrhythmias on raw electrocardiogram images: An aggregation attention multi-label model for diagnostic assistance

Self-supervised Anomaly Detection Pretraining Enhances Long-tail ECG Diagnosis

Prospects for artificial intelligence-enhanced electrocardiogram as a unified screening tool for cardiac and non-cardiac conditions: an explorative study in emergency care