Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
skip to main content
research-article

A novel automated CNN arrhythmia classifier with memory-enhanced artificial hummingbird algorithm

Published: 01 March 2023 Publication History

Abstract

Cardiac arrhythmias indicate cardiovascular disease which is the leading cause of mortality worldwide, and can be detected by an electrocardiogram (ECG). Automated deep learning methods have been developed to overcome the disadvantages of manual interpretation by medical experts. The performance of the networks strongly depends on hyperparameter optimization (HPO), and this NP-hard problem is suitable for metaheuristic (MH) methods. In this study, a novel method is proposed for the HPO of a convolutional neural network (CNN) arrhythmia classifier using an MH algorithm. The approach utilizes our variant of an MH method, named the memory-enhanced artificial hummingbird algorithm, which has an additional memory unit that stores the evaluations of the solutions and reduces the computation time significantly. The study also proposes a novel fitness function that considers both the accuracy rate and the total number of parameters of each candidate network. Experiments were conducted on raw ECG samples from the MIT-BIH arrhythmia database. The proposed method was compared with five other MH methods and achieved equal or outperforming results, with classification accuracy reaching 98.87%. The proposed method yielded promising results in finding a high-performing solution with relatively lower complexity.

Highlights

A novel memory-enhanced AHA-based automated CNN arrhythmia classifier is proposed.
A new fitness function is developed to evaluate hyperparameters of candidate CNNs.
The proposed method is utilized to classify cardiac arrhythmia on raw ECG samples.
The obtained results are promising compared to other state-of-the-art methods.

References

[1]
Acharya U.R., Oh S.L., Hagiwara Y., Tan J.H., Adam M., Gertych A., San Tan R., A deep convolutional neural network model to classify heartbeats, Computers in Biology and Medicine 89 (2017) 389–396,.
[2]
Asadi S., Roshan S., Kattan M.W., Random forest swarm optimization-based for heart diseases diagnosis, Journal of Biomedical Informatics 115 (2021),.
[3]
Association for the Advancement of Medical Instrumentation (2012). Testing and reporting performance results of cardiac rhythm and ST segment measurement algorithms: Standard EC57 AAMI.
[4]
Bacanin N., Bezdan T., Tuba E., Strumberger I., Tuba M., Optimizing convolutional neural network hyperparameters by enhanced swarm intelligence metaheuristics, Algorithms 13 (3) (2020) 67,.
[5]
Cheikhrouhou O., Mahmud R., Zouari R., Ibrahim M., Zaguia A., Gia T.N., One-dimensional CNN approach for ECG arrhythmia analysis in fog-cloud environments, IEEE Access 9 (2021) 103513–103523,.
[6]
Chen S.W., Wang S.L., Qi X.Z., Samuri S.M., Yang C., Review of ECG detection and classification based on deep learning: Coherent taxonomy, motivation, open challenges and recommendations, Biomedical Signal Processing Control 74 (2022),.
[7]
Cheraghalipour A., Hajiaghaei-Keshteli M., Paydar M.M., Tree growth algorithm (TGA): A novel approach for solving optimization problems, Engineering Applications of Artificial Intelligence 72 (2018) 393–414,.
[8]
Civicioglu P., Backtracking search optimization algorithm for numerical optimization problems, Applied Mathematics and Computation 219 (15) (2013) 8121–8144,.
[9]
Dey M., Omar N., Ullah M.A., Temporal feature-based classification into myocardial infarction and other CVDs merging CNN and bi-LSTM from ECG signal, IEEE Sensors Journal 21 (19) (2021) 21688–21695,.
[10]
Elmasry W., Akbulut A., Zaim A.H., Evolving deep learning architectures for network intrusion detection using a double PSO metaheuristic, Computer Networks 168 (2020),.
[11]
Erkan U., Toktas A., Ustun D., Hyperparameter optimization of deep CNN classifier for plant species identification using artificial bee colony algorithm, Journal of Ambient Intelligence Humanized Computers (2022),.
[12]
Faramarzi A., Heidarinejad M., Mirjalili S., Gandomi A.H., Marine predators algorithm: A nature-inspired metaheuristic, Expert Systems with Applications 152 (2020),.
[13]
Geem Z.W., Kim J.H., Loganathan G.V., A new heuristic optimization algorithm: Harmony search, Simulation 76 (2) (2001) 60–68,.
[14]
Geweid G.G.N., Chen J.D.Z., Automatic classification of atrial fibrillation from short single-lead ECG recordings using a hybrid approach of dual support vector machine, Expert Systems with Applications 198 (2022),.
[15]
Goh C.-H., Tan L.K., Lovell N.H., Ng S.-C., Tan M.P., Lim E., Robust PPG motion artifact detection using a 1-D convolution neural network, Computational Methods Programs Biomedicine 196 (2020),.
[16]
Goodfellow I., Bengio Y., Courville A., Deep learning, MIT Press, 2016, https://www.deeplearningbook.org.
[17]
Heart Rhythm Society, Heart rhythm disorders, 2022, https://upbeat.org/heart-rhythm-disorders.
[18]
Houssein E.H., Hassaballah M., Ibrahim I.E., AbdElminaam D.S., Wazery Y.M., An automatic arrhythmia classification model based on improved marine predators algorithm and convolutions neural networks, Expert Systems with Applications 187 (2022),.
[19]
Houssein E.H., Ibrahim I.E., Neggaz N., Hassaballah M., Wazery Y.M., An efficient ECG arrhythmia classification method based on manta ray foraging optimization, Expert Systems with Applications 181 (2021),.
[20]
Karaboga D., Basturk B., A powerful and efficient algorithm for numerical function optimization: Artificial bee colony (ABC) algorithm, Journal of Global Optimization 39 (3) (2007) 459–471,.
[21]
Karthiga M., Santhi V., Sountharrajan S., Hybrid optimized convolutional neural network for efficient classification of ECG signals in healthcare monitoring, Biomed. Signal Process. Control 76 (2022),.
[22]
Kaya Y., Detection of bundle branch block using higher order statistics and temporal features, International Arabian Journal of Information Technology 18 (3) (2021) 279–285,.
[23]
Kaya Y., Pehlivan H., Classification of premature ventricular contraction in ECG, International Journal of the Advance Computer Sciences and Applications 6 (7) (2015),.
[24]
Kaya Y., Pehlivan H., Feature selection using genetic algorithms for premature ventricular contraction classification, in: In 2015 9th international conference on electrical and electronics engineering (ELECO), IEEE, 2015, pp. 1229–1232,.
[25]
Kennedy J., Eberhart R., Particle swarm optimization, in: Proceedings of ICNN’95 - international conference on neural networks, Vol. 4, IEEE, 1995, pp. 1942–1948,.
[26]
Kılıç F., Kaya Y., Yildirim S., A novel multi population based particle swarm optimization for feature selection, Knowledge-Based on Systems 219 (2021),.
[27]
Kiranyaz S., Ince T., Gabbouj M., Real-time patient-specific ECG classification by 1-D convolutional neural networks, IEEE Transactions on Biomedical Engineering 63 (3) (2016) 664–675,.
[28]
Kirkpatrick S., Gelatt C.D., Vecchi M.P., Optimization by simulated annealing, Science 220 (4598) (1983) 671–680,.
[29]
Kumar A., Kumar S., Dutt V., Dubey A.K., García-Díaz V., IoT-based ECG monitoring for arrhythmia classification using coyote grey wolf optimization-based deep learning CNN classifier, Biomedical Signal Processing Control 76 (2022),.
[30]
Luo X., Yang L., Cai H., Tang R., Chen Y., Li W., Multi-classification of arrhythmias using a HCRNet on imbalanced ECG datasets, Computer Methods Programs Biomedicine 208 (2021),.
[31]
Martín A., Vargas V.M., Gutiérrez D., Hervás-Martínez C., Optimising convolutional neural networks using a hybrid statistically-driven coral reef optimisation algorithm, Applied Soft Computing 90 (2020),.
[32]
Martis R.J., Acharya U.R., Lim C.M., Suri J.S., Characterization of ECG beats from cardiac arrhythmia using discrete cosine transform in PCA framework, Knowledge-Based on Systems 45 (2013) 76–82,.
[33]
Mirjalili S., SCA: A sine cosine algorithm for solving optimization problems, Knowledge-Based on Systems 96 (2016) 120–133,.
[34]
Mirjalili S., Lewis A., The whale optimization algorithm, Advances in Engineering Software 95 (2016) 51–67,.
[35]
Mirjalili S., Mirjalili S.M., Lewis A., Grey wolf optimizer, Advances in Engineering Software 69 (2014) 46–61,.
[36]
Mohebbanaaz, Kumari L.V.R., Sai Y.P., Classification of ECG beats using optimized decision tree and adaptive boosted optimized decision tree, Signal Image Video Processings 16 (3) (2022) 695–703,.
[37]
Moody G.B., Mark R.G., The impact of the MIT-BIH arrhythmia database, IEEE Engineering Medical and Biological Magazine 20 (3) (2001) 45–50,.
[38]
Murat F., Yildirim O., Talo M., Baloglu U.B., Demir Y., Acharya U.R., Application of deep learning techniques for heartbeats detection using ECG signals-analysis and review, Computers in Biology and Medicine 120 (2020),.
[39]
National Heart, Lung, and Blood Institute, Arrhythmias, 2022, https://www.nhlbi.nih.gov/health/arrhythmias.
[40]
National Heart, Lung, and Blood Institute, Heart tests, 2022, https://www.nhlbi.nih.gov/health/heart-tests.
[41]
Navaneeth B., Suchetha M., PSO optimized 1-D CNN-SVM architecture for real-time detection and classification applications, Computers in Biology and Medicine 108 (2019) 85–92,.
[42]
Nguyen Q.H., Nguyen B.P., Nguyen T.B., Do T.T.T., Mbinta J.F., Simpson C.R., Stacking segment-based CNN with SVM for recognition of atrial fibrillation from single-lead ECG recordings, Biomedical Signal Processing Control 68 (2021),.
[43]
Oh S.L., Ng E.Y.K., Tan R.S., Acharya U.R., Automated diagnosis of arrhythmia using combination of CNN and LSTM techniques with variable length heart beats, Computers in Biology and Medicine 102 (2018) 278–287,.
[44]
Pandey S.K., Janghel R.R., Dev A.V., Mishra P.K., Automated arrhythmia detection from electrocardiogram signal using stacked restricted Boltzmann machine model, SN Applied Science 3 (6) (2021) 624,.
[45]
Passos L.A., Papa J.P., A metaheuristic-driven approach to fine-tune deep Boltzmann machines, Applied Soft Computing 97 (2020),.
[46]
Petmezas G., Haris K., Stefanopoulos L., Kilintzis V., Tzavelis A., Rogers J.A., Katsaggelos A.K., Maglaveras N., Automated atrial fibrillation detection using a hybrid CNN-LSTM network on imbalanced ECG datasets, Biomedical Signal Processing Control 63 (2021),.
[47]
PhysioNet, MIT-BIH arrhythmia database, 2022, https://www.physionet.org/content/mitdb/1.0.0/.
[48]
Ramasamy K., Balakrishnan K., Velusamy D., Detection of cardiac arrhythmias from ECG signals using FBSE and Jaya optimized ensemble random subspace k-nearest neighbor algorithm, Biomedical Signal Processing Control 76 (2022),.
[49]
Salcedo-Sanz S., Del Ser J., Landa-Torres I., Gil-López S., Portilla-Figueras J.A., The coral reefs optimization algorithm: A novel metaheuristic for efficiently solving optimization problems, Science World Journal 2014 (2014),.
[50]
Sinha N., Kumar Tripathy R., Das A., ECG beat classification based on discriminative multilevel feature analysis and deep learning approach, Biomedical Signal Processing Control 78 (2022),.
[51]
Storn R., Price K., Differential evolution – A simple and efficient heuristic for global optimization over continuous spaces, Journal of Global Optimization 11 (4) (1997) 341–359,.
[52]
Tsai C.-W., Hsia C.-H., Yang S.-J., Liu S.-J., Fang Z.-Y., Optimizing hyperparameters of deep learning in predicting bus passengers based on simulated annealing, Applied Soft Computing 88 (2020),.
[53]
World Health Organization, Cardiovascular diseases, 2022, https://www.who.int/health-topics/cardiovascular-diseases.
[54]
Xiao B., Xu Y., Bi X., Zhang J., Ma X., Heart sounds classification using a novel 1-D convolutional neural network with extremely low parameter consumption, Neurocomputing 392 (2020) 153–159,.
[55]
Xu S.S., Mak M.-W., Cheung C.-C., Towards end-to-end ECG classification with raw signal extraction and deep neural networks, IEEE Journal of Biomedical and Health Informatics 23 (4) (2019) 1574–1584,.
[56]
Yang X.-S., Firefly algorithms for multimodal optimization, in: International symposium on stochastic algorithms (SAGA 2009), Springer, 2009, pp. 169–178,.
[57]
Yang X.-S., A new metaheuristic bat-inspired algorithm, in: Nature inspired cooperative strategies for optimization (NICSO 2010), Springer, 2010, pp. 65–74,.
[58]
Yang X.-S., Deb S., Cuckoo search via Lévy flights, in: 2009 World congress on nature and biologically inspired computing (NaBIC), IEEE, 2009, pp. 210–214,.
[59]
Yang H., Meng C., Wang C., Data-driven feature extraction for analog circuit fault diagnosis using 1-D convolutional neural network, IEEE Access 8 (2020) 18305–18315,.
[60]
Yang L., Shami A., On hyperparameter optimization of machine learning algorithms: Theory and practice, Neurocomputing 415 (2020) 295–316,.
[61]
Yang H., Wei Z., A novel approach for heart ventricular and atrial abnormalities detection via an ensemble classification algorithm based on ECG morphological features, IEEE Access 9 (2021) 54757–54774,.
[62]
Zeng W., Yuan J., Yuan C., Wang Q., Liu F., Wang Y., A novel technique for the detection of myocardial dysfunction using ECG signals based on hybrid signal processing and neural networks, Soft Computing 25 (6) (2021) 4571–4595,.
[63]
Zhao W., Wang L., Mirjalili S., Artificial hummingbird algorithm: A new bio-inspired optimizer with its engineering applications, Computer Methods in Applied Mechanics and Engineering 388 (2022),.

Cited By

View all

Index Terms

  1. A novel automated CNN arrhythmia classifier with memory-enhanced artificial hummingbird algorithm
        Index terms have been assigned to the content through auto-classification.

        Recommendations

        Comments

        Information & Contributors

        Information

        Published In

        cover image Expert Systems with Applications: An International Journal
        Expert Systems with Applications: An International Journal  Volume 213, Issue PC
        Mar 2023
        1402 pages

        Publisher

        Pergamon Press, Inc.

        United States

        Publication History

        Published: 01 March 2023

        Author Tags

        1. Arrhythmia classification
        2. ECG
        3. Deep learning
        4. CNN
        5. Hyperparameter optimization
        6. Artificial hummingbird algorithm

        Qualifiers

        • Research-article

        Contributors

        Other Metrics

        Bibliometrics & Citations

        Bibliometrics

        Article Metrics

        • Downloads (Last 12 months)0
        • Downloads (Last 6 weeks)0
        Reflects downloads up to 01 Jan 2025

        Other Metrics

        Citations

        Cited By

        View all
        • (2025)Multi-source deep feature fusion for medical image analysisMultidimensional Systems and Signal Processing10.1007/s11045-024-00897-z36:1Online publication date: 1-Dec-2025
        • (2024)Brain-GCN-NetComputers in Biology and Medicine10.1016/j.compbiomed.2024.108971180:COnline publication date: 18-Nov-2024
        • (2024)Human activity recognition from multiple sensors data using deep CNNsMultimedia Tools and Applications10.1007/s11042-023-15830-y83:4(10815-10838)Online publication date: 1-Jan-2024
        • (2023)RETRACTED ARTICLE: A MobileNet-based CNN model with a novel fine-tuning mechanism for COVID-19 infection detectionSoft Computing - A Fusion of Foundations, Methodologies and Applications10.1007/s00500-022-07798-y27:9(5521-5535)Online publication date: 4-Jan-2023

        View Options

        View options

        Media

        Figures

        Other

        Tables

        Share

        Share

        Share this Publication link

        Share on social media