Metaheuristic optimized electrocardiography time-series anomaly classification with recurrent and long-short term neural networks

Article type: Research Article

Authors: Jovanovic, Luka^a | Zivkovic, Miodrag^b | Bacanin, Nebojsa^{c; d; e; *} | Bozovic, Aleksandra^f | Bisevac, Petar^g | Antonijevic, Milos^h

Affiliations: [a] Singidunum University, Danijelova, Belgrade, Serbia | [b] Singidunum University, Danijelova, Belgrade, Serbia | [c] Singidunum University, Danijelova, Belgrade, Serbia | [d] Department of Mathematics, Saveetha School of Engineering, SIMATS, Thandalam, Chennai, Tamilnadu, India | [e] MEU Research Unit, Middle East University, Amman, Jordan | [f] Technica faculty “Mihajlo Pupin”, University of Novi Sad, Dure Dakovica bb, Zrenjanin, Serbia | [g] Singidunum University, Danijelova, Belgrade, Serbia | [h] Singidunum University, Danijelova, Belgrade, Serbia

Correspondence: [*] Corresponding author: Nebojsa Bacanin, Singidunum University, Danijelova 32, 11000 Belgrade, Serbia. E-mail: [email protected].

Abstract: This study explores the realm of time series forecasting, focusing on the utilization of Recurrent Neural Networks (RNN) to detect abnormal cardiovascular rhythms in Electrocardiogram (ECG) signals. The principal objective is to optimize RNN performance by finely tuning hyperparameters, a complex task with known NP-hard complexity. To address this challenge, the study employs metaheuristic algorithms, specialized problem-solving techniques crafted for navigating intricate and non-deterministic optimization landscapes. Additionally, a refined algorithm is introduced to overcome limitations inherent in the original approach. This modified algorithm exhibits significant improvements, surpassing its predecessor in identifying anomalous cardiovascular rhythms within ECG signals. The most successful optimized model achieves an accuracy of 99.26%, outperforming models optimized by other contemporary metaheuristics assessed in the study. Further experimentation extends the initial inquiry by exploring the capabilities of Long Short-Term Memory (LSTM) models augmented by attention layers. In this extension, the best models demonstrate an accuracy of 99.83%, surpassing the original RNN models. These findings underscore the crucial importance of refining machine learning models and emphasize the potential for substantial advancements in healthcare through innovative algorithmic approaches.

Keywords: Electroencephalography, medical data, diagnosis, recurrent neural networks, metaheuristic optimization

DOI: 10.3233/HIS-240005

Journal: International Journal of Hybrid Intelligent Systems, vol. 20, no. 4, pp. 275-300, 2024