Electrocardiogram analysis for cardiac arrhythmia classification and prediction through self attention based auto encoder

Abstract Sudden cardiac arrest among young people is a recent worldwide risk, and it is noticed that people with cardiac arrhythmia are more susceptible to various heart diseases. Manual classification can be error-prone, and certainly, there is a need for automation to classify ECG signals to predi...

Full description

Saved in:
Bibliographic Details
Main Authors: Ameet Shah, Dhanpratap Singh, Heba G. Mohamed, Salil Bharany, Ateeq Ur Rehman, Seada Hussen
Format: Article
Language:English
Published: Nature Portfolio 2025-03-01
Series:Scientific Reports
Subjects:
Online Access:https://doi.org/10.1038/s41598-025-93906-5
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:Abstract Sudden cardiac arrest among young people is a recent worldwide risk, and it is noticed that people with cardiac arrhythmia are more susceptible to various heart diseases. Manual classification can be error-prone, and certainly, there is a need for automation to classify ECG signals to predict cardiac arrhythmia accurately. The proposed self-attention artificial intelligence auto-encoder algorithm proved an effective cardiac arrhythmia classification strategy with a novel modified Kalman filter pre-processing. We achieved 24.00 SNRimp, 0.055 RMSE, 22.1 PRD% for -5db, 20.4 SNRimp, 0.0245 RMSE, 12 PRD% whereas 14.05 SNRimp, 0.010 RMSE, and 7.25 PRD%, which reduces the ECG signal noise during the pre-processing and improves the visibility of the QRS complex and R-R peaks of ECG waveform. The extracted features were used in network of neurons to execute the classification for MIT-BIH arrhythmia databases using the newly developed self-attention autoencoder (AE) algorithm. The results are compared with existing models, revealing that the proposed system outperforms the classification and prediction of cardiac arrhythmia with a precision of 99.91%, recall of 99.86%, and accuracy of 99.71%. It is confirmed that self-attention-AE training results are promising, and it benefits the diagnosis of ECGs for complex cardiac conditions to solve real-world heart problems.
ISSN:2045-2322