Combining Support Vector Machine and Elephant Herding Optimization for Cardiac Arrhythmias

Aboul Ella Hassanien,Moataz Kilany,Essam H. Houssein
DOI: https://doi.org/10.48550/arXiv.1806.08242
2018-06-21
Abstract:Many people are currently suffering from heart diseases that can lead to untimely death. The most common heart abnormality is arrhythmia, which is simply irregular beating of the heart. A prediction system for the early intervention and prevention of heart diseases, including cardiovascular diseases (CDVs) and arrhythmia, is important. This paper introduces the classification of electrocardiogram (ECG) heartbeats into normal or abnormal. The approach is based on the combination of swarm optimization algorithms with a modified PannTompkins algorithm (MPTA) and support vector machines (SVMs). The MPTA was implemented to remove ECG noise, followed by the application of the extended features extraction algorithm (EFEA) for ECG feature extraction. Then, elephant herding optimization (EHO) was used to find a subset of ECG features from a larger feature pool that provided better classification performance than that achieved using the whole set. Finally, SVMs were used for classification. The results show that the EHOSVM approach achieved good classification results in terms of five statistical indices: accuracy, 93.31%; sensitivity, 45.49%; precision, 46.45%; F-measure, 45.48%; and specificity, 45.48%. Furthermore, the results demonstrate a clear improvement in accuracy compared to that of other methods when applied to the MITBIH arrhythmia database.
Signal Processing,Machine Learning
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is to develop an efficient and accurate electrocardiogram (ECG) heartbeat classification system for detecting cardiac rhythm abnormalities (arrhythmias). Specifically, the research aims to pre - process, extract features, select and optimize features, and finally classify electrocardiogram signals by combining the swarm optimization algorithm (especially the Elephant Herd Optimization algorithm, EHO) with the Support Vector Machine (SVM). The goal of the paper is to improve the accuracy of arrhythmia detection, thus contributing to early intervention and prevention of heart diseases. ### Main problem decomposition: 1. **Early detection of arrhythmia**: - Heart diseases are one of the leading causes of death worldwide, among which arrhythmia is one of the most common cardiac abnormalities. - Developing an efficient prediction system that can detect arrhythmia early is of great significance for preventing heart diseases. 2. **Pre - processing of electrocardiogram signals**: - Use the modified Pan - Tompkins algorithm (MPTA) to remove noise in electrocardiogram signals to improve the signal quality. 3. **Feature extraction**: - Extract useful features from electrocardiogram signals, including the morphology, amplitude, and duration of P - wave, QRS complex, and T - wave. - Use the Extended Feature Extraction Algorithm (EFEA) to further extract more features. 4. **Feature selection and optimization**: - Use the Elephant Herd Optimization algorithm (EHO) to select the subset that can best improve the classification performance from a large number of features. - Through multi - stage optimization, perform feature selection and SVM parameter optimization respectively to prevent the opportunity loss caused by simultaneously changing related parameters (features and classification penalties). 5. **Classification**: - Use the Support Vector Machine (SVM) for the final classification. - Evaluate the classification performance, including accuracy, sensitivity, precision, F - value, and specificity. ### Research methods: - **Data set**: Use the MIT - BIH arrhythmia database, which contains records of various types of cardiac abnormalities. - **Pre - processing**: Use MPTA to remove noise. - **Feature extraction**: Combine MPTA and EFEA to extract features. - **Feature selection and optimization**: Use the EHO algorithm to select the best feature subset and optimize SVM parameters. - **Classification**: Use SVM for classification and evaluate the performance through 3 - fold cross - validation. ### Experimental results: - **Classification performance**: The EHO - SVM method performs well on multiple performance indicators, especially reaching an accuracy of 93.31%. - **Comparison with other methods**: Compared with other methods, EHO - SVM has a significant improvement in accuracy on the MIT - BIH arrhythmia database. ### Conclusion: This research has successfully developed an efficient electrocardiogram heartbeat classification system that can detect arrhythmia at an early stage, providing strong support for the prevention and treatment of heart diseases. By combining EHO and SVM, this method has achieved significant improvements in both feature selection and classification performance.