Age Prediction From 12-lead Electrocardiograms Using Deep Learning: A Comparison of Four Models on a Contemporary, Freely Available Dataset

Andrew Barros,Ian German-Mesner,N Rich Nguyen,J Randall Moorman
DOI: https://doi.org/10.1101/2024.02.02.24302201
2024-07-22
Abstract:Objective The 12-lead electrocardiogram (ECG) is routine in clinical use and deep learning approaches have been shown to have the identify features not immediately apparent to human interpreters including age and sex. Several models have been published but no direct comparisons exist. Approach We implemented three previously published models and one unpublished model to predict age and sex from a 12-lead ECG and then compared their performance on an open-access data set. Main results All models converged and were evaluated on the holdout set. The best preforming age prediction model had a hold-out set mean absolute error of 8.06 years. The best preforming sex prediction model had a hold-out set area under the receiver operating curve of 0.92. Significance We compared performance of four models on an open-access dataset.
What problem does this paper attempt to address?
The main objective of this paper is to predict age and gender using 12-lead electrocardiograms (ECG) and to compare the performance of four different models on this task. Specifically, the researchers implemented three published models and one unpublished model, and evaluated their performance on a public dataset. The study found that in the age prediction task, the ResNet model had the lowest mean absolute error (MAE) on the test set, reaching 8.06 years; while in the gender prediction task, the ResNet+MHA model performed the best, with its accuracy, AUC-ROC, and other metrics surpassing those of the other models. Additionally, the study discussed the limitations of the models in practical clinical applications and the directions for future research.