Assignment 2
Nurtay Raiymbek
Paper title: ECG-based machine-learning algorithms for heartbeat classifcation.
Authors: SairaAziz, SajidAhmed & Mohamed-Slim Alouini
Journal: Nature
Main idea of the work: The proposed work is mainly aimed at developing a novel algorithm for
analyzing electrocardiogram (ECG) signals to enhance the detection of heart diseases. The
algorithm brings out the combined effect of two-event related moving-averages (TERMA) and
fractional Fourier-transform (FrFT) to improve the detection of critical peaks (P, QRS, and T
waves) within ECG signals. Besides, the paper presents a new machine learning model that is
trained on a large database from the Shaoxing People's Hospital database and implemented
with the aim of classifying the cardiovascular diseases through features derived from the ECG.
Background: Cardiovascular diseases, rank as the most common death in the world, thus, early
and proper diagnosis is a necessity for optimal treatment and management. The electrical
activity of the heart, which is diagnosed by an electrocardiogram (ECG), can be used for
differentiation of myocardial conditions. However, at the same time, ECG signals are generally
noisy, and artifacts can obscure important features. Accurate P, QRS, and T wave detection is
vital for the diagnosis of heart diseases and requires sophisticated techniques for signal
processing.
Several processing techniques for signals was used over the years to improve quality in ECG
signals and detection of waveforms. These include filtering, wavelet transforms, and empirical
mode decomposition. Authors state that there is a call for a move towards more advanced
algorithms that can result in greater reliability and accuracy in ECG signal analysis, especially
where noise and artifacts are present. In addition, machine learning models have also been
developed towards classification of heart diseases on the basis of ECG signals. These models
aim to perform the diagnosis process automatically in learning from a large dataset of ECG
recordings. However, the performance of such models is so much dependent on the quality of
the input signals and the features extracted from them. Therefore, improving the accuracy of
ECG signal analysis is a prerequisite step towards developing more reliable and efficient
diagnostic tools for cardiovascular diseases.
The goals and objectives: The primary objective of the work is to develop an enhanced
algorithm that combines both two-event-related moving averages (TERMA) and a fractional
Fourier transform to produce improved detection of critical peaks such as P, QRS, and T waves in
ECG signals and thus enable heart diseases to be better diagnosed. By improving the detection
of peaks in ECG signals, the algorithm provides higher accuracy and trustworthy parameters in
diagnosing cardiovascular diseases. The work intends to train the presented machine learning
model using the Shaoxing People’s Hospital dataset, which involves ECG data from over 10,000
patients, to ensure the accuracy and relevance of this model when diagnosing heart diseases.
Assessment of performance in respect of peak detection accuracy, and also comparing with the
existing methods to demonstrate the superiority of the proposed algorithm. The research aims
to develop a machine learning model capable of categorizing the different forms of heart
diseases and accurately discriminating them. Another objective of the research is to ensure the
applicability of the machine learning model when dealing with cross-database training and
testing scenarios to ensure its generalizability and its ability to perform well in different clinical
environments.
Methodology: The ECG signals are first preprocessed in order to eliminate noise and artifacts.
These are accomplished by means of conventional wavelet-transform-based filtering methods,
which are important in order to isolate the relevant ECG signal components while reducing the
effect of unwanted noise.
The next step was to apply the peak detection method. The proposed methodology makes use
of a proposed algorithm that is used to combine two-event related moving averages (TERMA)
and fractional Fourier-transform in order to improve the peak detection. TERMA is used to
determine places of interest in the ECG signal for peak detection, while FrFT is used in rotating
the ECG signals in the time-frequency plane so that the locations of each of the different peaks
of interest are made more prominent. These two techniques are employed to accurately detect
peaks that denote the P, QRS, and T waves in the ECG signal that are known to be indicative of
heart diseases.
Feature extraction is carried out on the detected peaks of the ECG signals, which includes the
locations of the P, R, and T peaks as well as between peak durations and other relevant signal
characteristics. Such features are used to characterize the representation of an ECG signal in a
form that would allow the machine learning algorithms being put into use for disease
classification.
A machine learning model is trained on the extracted features for disease classification. The
dataset used for training the model is large, and among them is a widely-used ECG database
named the Shaoxing People’s Hospital database, which contains more than 10,000 patients’ ECG
data. Various machine learning algorithms, like support vector machines and multi-layer
perceptron, are tried out to decide the most suitable method for classifying different types of
heart diseases depending on the ECG features. In addition to this, the model is tested on
different databases, including the MIT-BIH arrhythmia database and the INCART database, in
order to check on how generalizable and effective it would be in varied clinical contexts.
The proposed algorithm and the machine learning model performance was evaluated using
measures of sensitivity, positive predictivity, and error rate for peak detection and disease
classification, respectively.
Results:
The proposed algorithm using fractional Fourier transform was able to distinguish the P, R, and T
peaks quite accurately across the whole 48 records of the MIT-BIH database, using Lead II (MLII)
data. Performing sensitivity (SE), positive predictive value (+Pr), and error rate (Err) as metrics,
with the following results for the R-peak detection: SE = 99.83%, +Pr = 99.9%, Err = 0.00259. For
P and T peak detection, the proposed algorithm outperformed the TERMA algorithm, with
significant differences in detection performance for the remaining 38 records of the MIT-BIH
database. For example, the SE for the P-peak detection was 75.8% with an Err of 0.40,
compared to TERMA's SE of 67.5% and Err of 0.51. Simulation for the classification simulations
allocated 70% of the feature data for training the machine learning model and 30% for testing.
From the extracted features, we classified the ECG signals into categories such as normal, PVC,
APC, LBBB, RBBB, and PACE heartbeats using SVM and MLP classifiers. The four proposed
classifiers were tested on the SPH database, which had 12 lead ECG signals from 10,646
patients. The overall accuracy of the proposed classifier with four features (PR, RT, age, and sex)
was 82.2% for the MIT-BIH database and 84.2% for the SPH database. The MLP classifier showed
better performance than the SVM classifier on the SPH database, with an accuracy of 90.7%
using the four features. Since the MLP classifier was trained using the MIT-BIH arrhythmia
database, it was then tested on the St. Petersburg INCART and SPH databases to classify
Normal, RBBB, and PVC heartbeats.
Overall, the accuracy of the trained model on the INCART database was 99.85%, while it was
68% for the SPH database. The lower accuracy on the SPH database was attributed to the
classifier's inability to correctly classify RBBB and PVC heartbeats due to the lack of detection of
certain T-peak variations.
Conclusion:
This work presents a significant improvement to the state-of-the-art ECG signal analysis
methods by presenting a novel algorithm, combining two-event-related moving averages
(TERMA) with the fractional Fourier transform, for improved peak detection in ECG signals. This
proposed algorithm has better peak detection results than other work, hence its superior
accuracy as the base to diagnose heart diseases accurately. Moreover, a machine learning
model trained on a large dataset from the Shaoxing People's Hospital database further improves
disease classification through adding a new classifier to the model. This model demonstrates
the best prediction accuracy and reliability on the new SPH database. Cross-database testing
confirms the generality and robustness of the model, showing the possible practical
applications in various clinical settings. In summary, the work adds value to the enhancement of
the accuracy and efficiency of the diagnosis of cardiovascular diseases, essential for effective
treatment and management.
Individual Point of View:
The use of a machine learning model for classification of heart disease, where data from all the
various databases would be trained, is a positive feature that can be highlighted from this work.
It not only improves the accuracy of the disease classification but also claims the generalization
of the model across different databases, which is key for its deployment in the clinical setup.
However, future studies may further integrate deep learning techniques, which have proved
successful in different medical imaging and signal processing fields. Besides, further validation
of the model in clinical trials and his integration with portable ECG monitoring devices may be
explored to assess its practical value in remote patient monitoring and telemedicine.
In summary, this work lays a solid foundation for the future improvements of ECG signal analysis
and detection of heart disease, with the potential of having good impacts on patient care and
outcomes in cardiology.
Presentation:
[Link]
DAF90NsMzdk&utm_campaign=designshare&utm_medium=link2&utm_source=sharebutton