Abstract
Early detection of arrhythmia assists clinicians in timely treatment that improves the recovery rate and reduces the mortality rate. The existing research works employed various models to improve Arrhythmia detection performance. The existing models has low classification performance, due to irrelevant feature selection. In this research, K-Means Monarchy Butterfly Optimization (KMMBO) is used as a feature selection method for arrhythmia detection. The Monarchy Butterfly Optimization method escapes from local optima and provides good convergence in the search process. In this research, the role of k-means algorithm is to cluster the current generation population (feature subsets) from the Monarchy Butterfly algorithm into two groups to generate new solutions for the next generation. The relevant features selected by the KMMBO method are stored in a memory cell of Bi-directional Long Short Term Memory (Bi-LSTM) for long term that provides improvement in arrhythmia detection performance. The KMMBO method selection process avoids overfitting problems and imbalance data problems by focusing on unique features of the classes that increases the model sensitivity. The proposed KMMBO method’s efficiency is analyzed on the China Physiological Signal Challenge 2018 (CPSC 2018) and Massachusetts Institute of Technology and Beth Israel Hospital (MIT-BIH) datasets. Next, Dual Tree Complex Wavelet Transform (DTCWT), statistical, and entropy features are utilized for feature extraction and then KMMBO is applied for selecting optimum features. The KMMBO-BiLSTM model has better accuracy and sensitivity of 98.48% and 99.85% on the MIT-BIH dataset. In addition to this, the proposed KMMBO-BiLSTM model attained accuracy and sensitivity of 97.22% and 97.61% on the CPCS 2018 dataset.
Similar content being viewed by others
Data availability
The datasets generated during and/or analyzed during the current study are available in the MIT-BIH/CPSC 2018 repository, https://archive.physionet.org/cgi-bin/atm/ATM, http://2018.icbeb.org/Challenge.html.
References
Abuqaddom I, Mahafzah BA, Faris H (2021) Oriented stochastic loss descent algorithm to train very deep multi-layer neural networks without vanishing gradients. Knowl-Based Syst 230:107391. https://doi.org/10.1016/j.knosys.2021.107391
Acharya UR, Oh SL, Hagiwara Y, Tan JH, Adam M, Gertych A, San TR (2017) A deep convolutional neural network model to classify heartbeats. Comput Biol Med 89:389–396. https://doi.org/10.1016/j.compbiomed.2017.08.022
Alotaiby TN, Alrshoud SR, Alshebeili SA, Aljafar LM (2019) ECG-based subject identification using statistical features and random forest. J Sens 2019:6751932. https://doi.org/10.1155/2019/6751932
Al-Shaikh AA, Mahafzah BA, Alshraideh M (2023) Hybrid harmony search algorithm for social network contact tracing of COVID-19. Soft Comput 27(6):3343–3365. https://doi.org/10.1007/s00500-021-05948-2
Anwar SM, Gul M, Majid M, Alnowami M (2018) Arrhythmia classification of ECG signals using hybrid features. Comput Math Methods Med 2018:1380348. https://doi.org/10.1155/2018/1380348
Bacanin N, Bezdan T, Tuba E, Strumberger I, Tuba M (2020) Monarch butterfly optimization based convolutional neural network design. Mathematics 8(6):936. https://doi.org/10.3390/math8060936
Bassiouni MM, El-Dahshan ESA, Khalefa W, Salem AM (2018) Intelligent hybrid approaches for human ECG signals identification. SIViP 12(5):941–949. https://doi.org/10.1007/s11760-018-1237-5
Bhagyalakshmi V, Pujeri RV, Devanagavi GD (2021) GB-SVNN: genetic BAT assisted support vector neural network for arrhythmia classification using ECG signals. J King Saud Univ-Comput Inform Sci 33(1):54–67. https://doi.org/10.1016/j.jksuci.2018.02.005
Bhurane AA, Sharma M, San-Tan R, Acharya UR (2019) An efficient detection of congestive heart failure using frequency localized filter banks for the diagnosis with ECG signals. Cogn Syst Res 55:82–94. https://doi.org/10.1016/j.cogsys.2018.12.017
Borman R I, Rossi F, Jusman Y, Rahni A A A, Putra S D, & Herdiansah A (2021). Identification of herbal leaf types based on their image using first order feature extraction and multiclass SVM algorithm. In: 2021 1st international conference on electronic and electrical engineering and intelligent system (ICE3IS), 15–16 Oct 2021, Yogyakarta, Indonesia (pp 12–17). IEEE. https://doi.org/10.1109/ICE3IS54102.2021.9649677
Chaudhary A, Kolhe S, Kamal R (2016) An improved random forest classifier for multi-class classification. Inform Process Agric 3(4):215–222. https://doi.org/10.1016/j.inpa.2016.08.002
Chen DW, Miao R, Yang WQ, Liang Y, Chen HH, Huang L, Deng CJ, Han N (2019) A feature extraction method based on differential entropy and linear discriminant analysis for emotion recognition. Sensors 19(7):1631. https://doi.org/10.3390/s19071631
El-Hoseny HM, El-Rahman WA, El-Rabaie ESM, El-Samie FEA, Faragallah OS (2018) An efficient DT-CWT medical image fusion system based on modified central force optimization and histogram matching. Infrared Phys Technol 94:223–231. https://doi.org/10.1016/j.infrared.2018.09.003
Goldberger AL, Amaral LA, Glass L, Hausdorff JM, Ivanov PC, Mark RG, Mietus JE, Moody GB, Peng CK, Stanley HE (2000) PhysioBank, PhysioToolkit, and PhysioNet: components of a new research resource for complex physiologic signals. Circulation 101(23):e215–e220. https://doi.org/10.1161/01.cir.101.23.e215
Gou J, Ma H, Ou W, Zeng S, Rao Y, Yang H (2019) A generalized mean distance-based k-nearest neighbor classifier. Expert Syst Appl 115:356–372. https://doi.org/10.1016/j.eswa.2018.08.021
Gu R (2017) Multiscale Shannon entropy and its application in the stock market. Physica A 484:215–224. https://doi.org/10.1016/j.physa.2017.04.164
Gutiérrez-Gnecchi JA, Morfin-Magana R, Lorias-Espinoza D, del Carmen Tellez-Anguiano A, Reyes-Archundia E, Méndez-Patiño A, Castañeda-Miranda R (2017) DSP-based arrhythmia classification using wavelet transform and probabilistic neural network. Biomed Signal Process Control 32:44–56. https://doi.org/10.1016/j.bspc.2016.10.005
Hammad M, Iliyasu AM, Subasi A, Ho ESL, El-Latif AAA (2020) A multitier deep learning model for arrhythmia detection. IEEE Trans Instrum Meas 70:2502809. https://doi.org/10.1109/TIM.2020.3033072
Huang JS, Chen BQ, Zeng NY, Cao XC, Li Y (2020) Accurate classification of ECG arrhythmia using MOWPT enhanced fast compression deep learning networks. J Ambient Intell Humaniz Comput. https://doi.org/10.1007/s12652-020-02110-y
Jangra M, Dhull SK, Singh KK, Singh A, Cheng X (2021) O-WCNN: an optimized integration of spatial and spectral feature map for arrhythmia classification. Complex Intell Syst. https://doi.org/10.1007/s40747-021-00371-4
Jiang GJA, Fan SZ, Abbod MF, Huang HH, Lan JY, Tsai FF, Chang HC, Yang YW, Chuang FL, Chiu YF, Jen KK, Wu JF, Shieh JS (2015) Sample entropy analysis of EEG signals via artificial neural networks to model patients’ consciousness level based on anesthesiologists experience. BioMed Res Int. https://doi.org/10.1155/2015/343478
Le T, Vo MT, Vo B, Hwang E, Rho S, Baik SW (2019) Improving electric energy consumption prediction using CNN and Bi-LSTM. Appl Sci 9(20):4237. https://doi.org/10.3390/app9204237
Li B, Zhang J, He Y, Wang Y (2017) Short-term load-forecasting method based on wavelet decomposition with second-order gray neural network model combined with ADF test. IEEE Access 5:16324–16331. https://doi.org/10.1109/ACCESS.2017.2738029
Liang Y, Yin S, Tang Q, Zheng Z, Elgendi M, Chen Z (2020) Deep learning algorithm classifies heartbeat events based on electrocardiogram signals. Front Physiol 11:569050. https://doi.org/10.3389/fphys.2020.569050
Liu S, Shao J, Kong T, Malekian R (2020) Ecg arrhythmia classification using high order spectrum and 2d graph fourier transform. Appl Sci 10(14):4741. https://doi.org/10.3390/app10144741
Liu Y, Li Q, Wang K, Liu J, He R, Yuan Y, Zhang H (2021) Automatic multi-label ECG classification with category imbalance and cost-sensitive thresholding. Biosensors 11(11):453. https://doi.org/10.3390/bios11110453
Mahafzah BA, Jabri R, Murad O (2021) Multithreaded scheduling for program segments based on chemical reaction optimizer. Soft Comput 25(4):2741–2766. https://doi.org/10.1007/s00500-020-05334-4
Majumder S, Kar S, Pal T (2018) Mean-entropy model of uncertain portfolio selection problem. In: Mandal J, Mukhopadhyay S, Dutta P (eds) Multi-objective optimization. Springer, Singapore, pp 25–54
Majumder S, Kar S, Pal T (2019) Rough-fuzzy quadratic minimum spanning tree problem. Expert Syst 36(2):e12364. https://doi.org/10.1111/exsy.12364
Mathunjwa BM, Lin YT, Lin CH, Abbod MF, Shieh JS (2021) ECG arrhythmia classification by using a recurrence plot and convolutional neural network. Biomed Signal Process Control 64:102262. https://doi.org/10.1016/j.bspc.2020.102262
Mirjalili S, Lewis A (2016) The whale optimization algorithm. Adv Eng Softw 95:51–67. https://doi.org/10.1016/j.advengsoft.2016.01.008
Mirjalili S, Mirjalili SM, Lewis A (2014) Grey wolf optimizer. Adv Eng Softw 69:46–61. https://doi.org/10.1016/j.advengsoft.2013.12.007
Petmezas G, Haris K, Stefanopoulos L, Kilintzis V, Tzavelis A, Rogers JA, Katsaggelos AK, Maglaveras N (2021) Automated atrial fibrillation detection using a hybrid CNN-LSTM network on imbalanced ECG datasets. Biomed Signal Process Control 63:102194. https://doi.org/10.1016/j.bspc.2020.102194
Qaisar SM, Mihoub A, Krichen M, Nisar H (2021) Multirate processing with selective subbands and machine learning for efficient arrhythmia classification. Sensors 21(4):1511. https://doi.org/10.3390/s21041511
Qiao F, Li B, Zhang Y, Guo H, Li W, Zhou S (2020) A fast and accurate recognition of ECG signals based on ELM-LRF and BLSTM algorithm. IEEE Access 8:71189–71198. https://doi.org/10.1109/ACCESS.2020.2987930
Rahbar M, Yazdani S (2021) Historical knowledge-based MBO for global optimization problems and its application to clustering optimization. Soft Comput 25(5):3485–3501. https://doi.org/10.1007/s00500-020-05381-x
Rahul J, Sora M, Sharma LD, Bohat VK (2021) An improved cardiac arrhythmia classification using an RR interval-based approach. Biocybern Biomed Eng 41(2):656–666. https://doi.org/10.1016/j.bbe.2021.04.004
Sahargahi V, Majidnezhad V, Afshord ST, Jafari Y (2022) An intelligent chaotic clonal optimizer. Appl Soft Comput 115:108126. https://doi.org/10.1016/j.asoc.2021.108126
Sahoo S, Subudhi A, Dash M, Sabut S (2020) Automatic classification of cardiac arrhythmias based on hybrid features and decision tree algorithm. Int J Autom Comput 17(4):551–561. https://doi.org/10.1007/s11633-019-1219-2
Sangaiah AK, Arumugam M, Bian GB (2020) An intelligent learning approach for improving ECG signal classification and arrhythmia analysis. Artif Intell Med 103:101788. https://doi.org/10.1016/j.artmed.2019.101788
Shahid F, Zameer A, Muneeb M (2020) Predictions for COVID-19 with deep learning models of LSTM GRU and Bi-LSTM. Chaos, Solit Fractals 140:110212. https://doi.org/10.1016/j.chaos.2020.110212
Shaker AM, Tantawi M, Shedeed HA, Tolba MF (2020) Generalization of convolutional neural networks for ECG classification using generative adversarial networks. IEEE Access 8:35592–35605. https://doi.org/10.1109/ACCESS.2020.2974712
Shami TM, El-Saleh AA, Alswaitti M, Al-Tashi Q, Summakieh MA, Mirjalili S (2022) Particle swarm optimization: a comprehensive survey. IEEE Access 10:10031–10061. https://doi.org/10.1109/ACCESS.2022.3142859
Sharkawy AN (2020) Principle of neural network and its main types: review. J Adv Appl Comput Mathemat 7:8–19. https://doi.org/10.15377/2409-5761.2020.07.2
Sharma M, Tan RS, Acharya UR (2019) Automated heartbeat classification and detection of arrhythmia using optimal orthogonal wavelet filters. Inform Med Unlocked 16:100221. https://doi.org/10.1016/j.imu.2019.100221
Shen SL, Njock PGA, Zhou A, Lyu HM (2021) Dynamic prediction of jet grouted column diameter in soft soil using Bi-LSTM deep learning. Acta Geotech 16(1):303–315. https://doi.org/10.1007/s11440-020-01005-8
Shrestha A, Li H, Le Kernec J, Fioranelli F (2020) Continuous human activity classification from FMCW radar with Bi-LSTM networks. IEEE Sens J 20(22):13607–13619. https://doi.org/10.1109/JSEN.2020.3006386
Soltani P, Hadavandi E (2019) A monarch butterfly optimization-based neural network simulator for prediction of siro-spun yarn tenacity. Soft Comput 23(20):10521–10535. https://doi.org/10.1007/s00500-018-3624-9
Song YY, Ying LU (2015) Decision tree methods: applications for classification and prediction. Shanghai Arch Psychiatry 27(2):130–135. https://doi.org/10.11919/j.issn.1002-0829.215044
Sun L, Kong X, Xu J, Xue ZA, Zhai R, Zhang S (2019) A hybrid gene selection method based on ReliefF and ant colony optimization algorithm for tumor classification. Sci Rep 9(1):8978. https://doi.org/10.1038/s41598-019-45223-x
Tawhid MA, Ibrahim AM (2023) An efficient hybrid swarm intelligence optimization algorithm for solving nonlinear systems and clustering problems. Soft Comput. https://doi.org/10.1007/s00500-022-07780-8
Ullah A, Anwar SM, Bilal M, Mehmood RM (2020) Classification of arrhythmia by using deep learning with 2-D ECG spectral image representation. Remote Sensing 12(10):1685. https://doi.org/10.3390/rs12101685
Wang J (2021) An intelligent computer-aided approach for atrial fibrillation and atrial flutter signals classification using modified bidirectional LSTM network. Inf Sci 574:320–332. https://doi.org/10.1016/j.ins.2021.06.009
Wang R, Fan J, Li Y (2020) Deep multi-scale fusion neural network for multi-class arrhythmia detection. IEEE J Biomed Health Inform 24(9):2461–2472. https://doi.org/10.1109/JBHI.2020.2981526
Yang H, Wei Z (2020) Arrhythmia recognition and classification using combined parametric and visual pattern features of ECG morphology. IEEE Access 8:47103–47117. https://doi.org/10.1109/ACCESS.2020.2979256
Yang W, Si Y, Wang D, Guo B (2018) Automatic recognition of arrhythmia based on principal component analysis network and linear support vector machine. Comput Biol Med 101:22–32. https://doi.org/10.1016/j.compbiomed.2018.08.003
Yang H, Liu J, Zhang L, Li Y, Zhang H (2021) ProEGAN-MS: a progressive growing generative adversarial networks for electrocardiogram generation. IEEE Access 9:52089–52100. https://doi.org/10.1109/ACCESS.2021.3069827
Yi H, Shiyu S, Xiusheng D, Zhigang C (2017) A study on deep neural networks framework. In: 2016 IEEE advanced information management, communicates, electronic and automation control conference (IMCEC), 03–05 Oct 2016, Xi'an, China (pp 1519–1522). https://doi.org/10.1109/IMCEC.2016.7867471
Zhang D, Jia X, Ding H, Ye D, Thakor NV (2010) Application of Tsallis entropy to EEG: quantifying the presence of burst suppression after asphyxial cardiac arrest in rats. IEEE Trans Biomed Eng 57(4):867–874. https://doi.org/10.1109/TBME.2009.2029082
Zhang J, Liu A, Gao M, Chen X, Zhang X, Chen X (2020) ECG-based multi-class arrhythmia detection using spatio-temporal attention-based convolutional recurrent neural network. Artif Intell Med 106:101856. https://doi.org/10.1016/j.artmed.2020.101856
Zhu C, Idemudia CU, Feng W (2019) Improved logistic regression model for diabetes prediction by integrating PCA and K-means techniques. Inform Med Unlocked 17:100179. https://doi.org/10.1016/j.imu.2019.100179
Funding
This research received no external funding.
Author information
Authors and Affiliations
Contributions
RM contributed to conceptualization, implementation, resource and data acquisition. NG contributed to writing, validation, and visualization.
Corresponding author
Ethics declarations
Conflict of interest
The authors declare that they have no conflict of interest.
Informed consent
Not Applicable.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Mogili, R., Narsimha, G. K-means monarchy butterfly optimization for feature selection and Bi-LSTM for arrhythmia classification. Soft Comput 27, 14935–14951 (2023). https://doi.org/10.1007/s00500-023-08579-x
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s00500-023-08579-x