ECG-LVEF model: deep learning detection of left ventricular systolic dysfunction from the 12-lead ECG
This record contains the trained model weights and the code for a deep learning model that detects left ventricular systolic dysfunction (LVEF ≤40%) from a standard 10-second, 12-lead ECG.
Model. The model is a ConvNeXt V2 base network (pretrained on ImageNet), fine-tuned on ECG images. Each ECG is filtered and cut into five 2-second strips of eight leads (I, II, V1–V6), and each strip is converted to an image. The model scores each strip, and the five probabilities are averaged into one probability of LVEF ≤40% per ECG.
Training data. The model was developed with ECGs from Amsterdam UMC, paired with LVEF from transthoracic echocardiography or cardiac magnetic resonance (CMR). Soft labels reflected the measurement uncertainty of each imaging method. The model was externally validated in an independent CMR cohort from Erasmus MC. The patient data are not included.
Contents
weights/model.pt: trained model weights
src/: code for preprocessing, training, validation and inference
config/: configuration files
README.md: installation and usage
Use. Place your ECGs (DICOM) in a zip file, set the paths in config/inference.json, and run uv run ecg-inference. The output is one probability per ECG. The model overestimates absolute risk. Recalibrate it and check thresholds locally before use.
Intended use and limitations. This model is for research, independent validation and non-commercial evaluation. It is not a CE-marked medical device and must not be used for clinical decisions. It was developed in patients referred for imaging at Dutch academic centres. Performance with other ECG vendors, healthcare systems and populations is unknown.
License. CC BY-NC 4.0. This follows the license of the pretrained ConvNeXt V2 weights.
Citation. Please cite: Selder JL, van der Meer O, van der Zwaard S, et al. A publicly available, externally validated ECG model for left ventricular systolic dysfunction, trained on echocardiography- and CMR-derived ejection fraction. [Journal, year, DOI – to be added]