TY - JOUR
T1 - A Reliable Deep Learning Model for ECG Interpretation
T2 - Mitigating Overconfidence and Direct Uncertainty Quantification
AU - Li, Xuedong
AU - Zheng, Qingxiao
AU - Zhang, Shibin
AU - Fu, Shipeng
AU - Chen, Yingke
AU - Ye, Ke
PY - 2025/5/20
Y1 - 2025/5/20
N2 - Electrocardiogram (ECG) interpretation using deep learning models holds immense potential for improving cardiac diagnosis. However, existing models often suffer from overconfident predictions and lack the capability to directly quantify uncertainty, leading to unreliable clinical guidance. To address this challenge, we propose a model for uncertainty-aware ECG interpretation. The model employs a deep convolutional architecture with max-pooling residual modules to capture both local and global spatiotemporal features from raw ECG signals. The architectural design respects the symmetry inherent in ECG waveforms—such as periodicity and morphological consistency across cardiac cycles—enabling the network to extract clinically relevant features more effectively. Then, unlike conventional models that rely on softmax-based probability outputs, our approach parameterizes class distributions using the Dirichlet distribution, while Subjective Logic translates these parameters into interpretable belief masses and uncertainty scores. We evaluate the model on the PhysioNet Challenge 2017 dataset, our model achieves an accuracy of 86.12%, an F1 score of 83.14%, a Precision-Recall Area Under the Curve (PR-AUC) of 85.25%, and a Receiver Operating Characteristic Area Under the Curve (ROC-AUC) of 92.87%—outperforming baseline models in three out of four metrics. Critically, the model reduces overconfidence to 0.59% (compared to 12–22% in softmax-based baselines), aligning prediction confidence with true accuracy. By progressively increasing the uncertainty threshold u, the model dynamically filters low-confidence predictions, leading to consistently improved performance—reaching up to 93.59% accuracy, 93.22% F1 score, 89.17% PR-AUC, and 95.10% ROC-AUC at u = 0.1. These results validate the model’s capacity for reliable ECG interpretation while leveraging physiological signal symmetry for enhanced feature extraction.
AB - Electrocardiogram (ECG) interpretation using deep learning models holds immense potential for improving cardiac diagnosis. However, existing models often suffer from overconfident predictions and lack the capability to directly quantify uncertainty, leading to unreliable clinical guidance. To address this challenge, we propose a model for uncertainty-aware ECG interpretation. The model employs a deep convolutional architecture with max-pooling residual modules to capture both local and global spatiotemporal features from raw ECG signals. The architectural design respects the symmetry inherent in ECG waveforms—such as periodicity and morphological consistency across cardiac cycles—enabling the network to extract clinically relevant features more effectively. Then, unlike conventional models that rely on softmax-based probability outputs, our approach parameterizes class distributions using the Dirichlet distribution, while Subjective Logic translates these parameters into interpretable belief masses and uncertainty scores. We evaluate the model on the PhysioNet Challenge 2017 dataset, our model achieves an accuracy of 86.12%, an F1 score of 83.14%, a Precision-Recall Area Under the Curve (PR-AUC) of 85.25%, and a Receiver Operating Characteristic Area Under the Curve (ROC-AUC) of 92.87%—outperforming baseline models in three out of four metrics. Critically, the model reduces overconfidence to 0.59% (compared to 12–22% in softmax-based baselines), aligning prediction confidence with true accuracy. By progressively increasing the uncertainty threshold u, the model dynamically filters low-confidence predictions, leading to consistently improved performance—reaching up to 93.59% accuracy, 93.22% F1 score, 89.17% PR-AUC, and 95.10% ROC-AUC at u = 0.1. These results validate the model’s capacity for reliable ECG interpretation while leveraging physiological signal symmetry for enhanced feature extraction.
KW - electrocardiogram interpretation
KW - Deep Learning
KW - overconfidence
KW - uncertainty quantification
KW - dirichlet distribution
KW - Subjective Logic
UR - https://www.scopus.com/pages/publications/105006805521
U2 - 10.3390/sym17050794
DO - 10.3390/sym17050794
M3 - Article
SN - 2073-8994
VL - 17
JO - Symmetry
JF - Symmetry
IS - 5
M1 - 794
ER -