A Cluster-Based Opposition Differential Evolution Algorithm Boosted by a Local Search for ECG Signal Classification

Mehran Pourvahab,Seyed Jalaleddin Mousavirad,Virginie Felizardo,Nuno Pombo,Henriques Zacarias,Hamzeh Mohammadigheymasi,Sebastião Pais,Seyed Nooreddin Jafari,Nuno M.Garcia
2023-10-07
Abstract:Electrocardiogram (ECG) signals, which capture the heart's electrical activity, are used to diagnose and monitor cardiac problems. The accurate classification of ECG signals, particularly for distinguishing among various types of arrhythmias and myocardial infarctions, is crucial for the early detection and treatment of heart-related diseases. This paper proposes a novel approach based on an improved differential evolution (DE) algorithm for ECG signal classification for enhancing the performance. In the initial stages of our approach, the preprocessing step is followed by the extraction of several significant features from the ECG signals. These extracted features are then provided as inputs to an enhanced multi-layer perceptron (MLP). While MLPs are still widely used for ECG signal classification, using gradient-based training methods, the most widely used algorithm for the training process, has significant disadvantages, such as the possibility of being stuck in local optimums. This paper employs an enhanced differential evolution (DE) algorithm for the training process as one of the most effective population-based algorithms. To this end, we improved DE based on a clustering-based strategy, opposition-based learning, and a local search. Clustering-based strategies can act as crossover operators, while the goal of the opposition operator is to improve the exploration of the DE algorithm. The weights and biases found by the improved DE algorithm are then fed into six gradient-based local search algorithms. In other words, the weights found by the DE are employed as an initialization point. Therefore, we introduced six different algorithms for the training process (in terms of different local search algorithms). In an extensive set of experiments, we showed that our proposed training algorithm could provide better results than the conventional training algorithms.
Neural and Evolutionary Computing
What problem does this paper attempt to address?
The paper aims to address the problem of electrophysiological signal classification, particularly the differentiation of various types of arrhythmias and myocardial infarction in electrocardiograms (ECG). Specifically, the study proposes a new method based on an improved Differential Evolution (DE) algorithm to enhance the performance of Multilayer Perceptrons (MLP) in ECG signal classification. Traditionally, although MLPs are widely used for ECG signal classification, the use of gradient-based training methods, such as the backpropagation algorithm, has limitations, such as the potential to get trapped in local optima. To tackle this issue, the paper employs an enhanced DE algorithm for the training process, which is one of the most effective population-based optimization algorithms. This improved DE algorithm is based on clustering strategies, opposition-based learning, and local search, aiming to improve exploration capabilities and convergence speed, thereby finding better weights and bias values. The specific improvements include: 1. **Clustering Strategy**: Used as a multi-parent crossover operator to enhance data utilization and promote better convergence. 2. **Opposition-Based Learning**: By using quasi-oppositional learning to boost exploration capabilities, simplifying the process of finding the optimal solution. 3. **Local Search**: The ideal weights discovered through the improved DE technique are used to initialize six different gradient-based local search algorithms for more precise and effective model fine-tuning. The paper also highlights the application of six different local search algorithms, resulting in six unique ECG classification methods, which provide flexibility and the ability to adapt to different problem scenarios. Through extensive experiments, the paper demonstrates that the proposed training algorithms can provide better results than traditional training methods, proving their effectiveness and superiority in ECG signal classification problems. Furthermore, the paper emphasizes the importance of accurate ECG signal classification for the early detection and treatment of cardiovascular diseases (CVDs), especially in identifying arrhythmias and other cardiac conditions. ECG signal analysis employs computer-assisted methods, and machine learning algorithms have become a promising alternative to traditional methods due to their potential in diagnosing heart diseases from ECG signals. However, despite the many machine learning algorithms available, there is still significant room for improvement in enhancing the effectiveness of prediction results. Therefore, the paper introduces a novel ECG signal classification algorithm that combines MLP with the improved DE algorithm and gradient-based local search algorithms to meet this challenge.