Wavelet Scattering Transform for ECG Cardiovascular Disease Classification

Abstract

Classifying the ECG dataset is the main technique for diagnosing heart disease. However, the focus of this field is increasingly on prediction, with a growing dependence on machine learning techniques. This study aimed to enhance the accuracy of cardiovascular disease classification using data from the PhysioNet database by employing machine learning (ML). The study proposed several multi-class classification models that accurately identify patterns within three classes: heart failure rhythm (HFR), normal heart rhythm (NHR), and arrhythmia (ARR). This was accomplished by utilizing a database containing 162 ECG signals. The study employed a variety of techniques, including frequency-time domain analysis, spectral features, and wavelet scattering, to extract features and capture unique characteristics from the ECG dataset. The SVM model produced a training accuracy of 97.1% and a testing accuracy of 92%. This work provides a reliable, effective, and human error-free diagnostic tool for identifying heart disease. Furthermore, it could prove to be a valuable resource for future medical research projects aimed at improving the diagnosis and treatment of cardiovascular diseases.