Optimized wavelet neural network for ECG arrhythmia detection

Authors

https://doi.org/10.22105/thi.v3i1.48

Abstract

Accurate detection of cardiac arrhythmias from electrocardiogram (ECG) signals plays a vital role in computer-aided diagnosis systems. This study presents an enhanced arrhythmia classification framework based on Discrete Wavelet Transform (DWT) and Multilayer Perceptron (MLP) Neural Networks (NN). ECG recordings are decomposed into multiple frequency bands using DWT, and a set of statistical descriptors is extracted from the resulting wavelet coefficients to characterize signal behavior in both time and frequency domains. These descriptors are integrated with ECG morphological information and heartbeat interval features to construct a comprehensive feature representation. To eliminate redundant information and improve classification efficiency, a feature selection stage is employed before the classification process. The selected features are then used to train an optimized MLP classifier for distinguishing normal heartbeats from different arrhythmia types. Experiments conducted on recordings from the MIT-BIH Arrhythmia Database demonstrate that the proposed combination of wavelet-based statistical analysis, feature optimization, and neural-network classification provides reliable and accurate arrhythmia recognition performance.

Keywords:

Electrocardiogram, Arrhythmia, Discrete wavelet transform, Neural network, Principal component analysis

References

  1. [1] Moody, G. B., & Mark, R. G. (2001). The impact of the MIT-BIH arrhythmia database. IEEE Engineering in Medicine and Biology Magazine, 20(3), 45–50. https://doi.org/10.1109/51.932724
  2. [2] Xiao, Q., Lee, K., Mokhtar, S. A., Ismail, I., Pauzi, A. L. bin M., Zhang, Q., & Lim, P. Y. (2023). Deep learning-based ECG arrhythmia classification: A systematic review. Applied Sciences, 13(8), 4964. https://doi.org/10.3390/app13084964
  3. [3] De Chazal, P., O’Dwyer, M., & Reilly, R. B. (2004). Automatic classification of heartbeats using ECG morphology and heartbeat interval features. IEEE Transactions on Biomedical Engineering, 51(7), 1196–1206. https://doi.org/10.1109/TBME.2004.827359
  4. [4] Christov, I. I. (2004). Real time electrocardiogram QRS detection using combined adaptive threshold. Biomedical Engineering Online, 3(1), 28. https://doi.org/10.1186/1475-925X-3-28
  5. [5] Osowski, S., & Linh, T. H. (2001). ECG beat recognition using fuzzy hybrid neural network. IEEE Transactions on Biomedical Engineering, 48(11), 1265–1271. https://doi.org/10.1109/10.959322
  6. [6] Mallat, S. (2008). A Wavelet Tour of Signal Processing. Academic Press. https://www.di.ens.fr/~mallat/papiers/WaveletTourChap1-2-3.pdf
  7. [7] Das, A., Catthoor, F., & Schaafsma, S. (2018). Heartbeat classification in wearables using multi-layer perceptron and time-frequency joint distribution of ecg [presentation]. Proceedings of the 2018 IEEE/ACM International Conference on Connected Health: Applications, Systems and Engineering Technologies (pp. 69–74). https://doi.org/10.1145/3278576.3278598
  8. [8] Khan, A. H., Hussain, M., & Malik, M. K. (2021). Arrhythmia classification techniques using deep neural network. Complexity, 2021(1), 9919588. https://doi.org/10.1155/2021/9919588
  9. [9] Kachuee, M., Fazeli, S., & Sarrafzadeh, M. (2018). Ecg heartbeat classification: A deep transferable representation. 2018 IEEE International Conference on Healthcare Informatics (ICHI) (pp. 443–444). IEEE. https://doi.org/10.1109/ICHI.2018.00092
  10. [10] Singh, R., Rajpal, N., & Mehta, R. (2021). An empiric analysis of wavelet-based feature extraction on deep learning and machine learning algorithms for arrhythmia classification. International Journal of Interactive Multimedia and Artificial Intelligence, 6(6), 25–34. https://doi.org/10.9781/ijimai.2020.11.005
  11. [11] Zhang, D., Wang, S., Li, F., Wang, J., Sangaiah, A. K., Sheng, V. S., & Ding, X. (2019). An ECG signal de-noising approach based on wavelet energy and sub-band smoothing filter. Applied Sciences, 9(22), 4968. https://doi.org/10.3390/app9224968
  12. [12] Gualsaquí, M., Vizcaíno, I., Rosero, V. G. P., & Calero, M. J. F. (2018). ECG signal denoising using discrete wavelet transform: A comparative analysis of threshold values and functions. Maskana, 9(1), 105–114. https://doi.org/10.18537/mskn.09.01.10
  13. [13] Robnik-Šikonja, M., & Kononenko, I. (2003). Theoretical and empirical analysis of ReliefF and RReliefF. Machine Learning, 53(1), 23–69. https://doi.org/10.1023/A:1025667309714
  14. [14] Vidal, R., Ma, Y., & Sastry, S. S. (2016). Principal component analysis. In Generalized Principal Component Analysis (pp. 25–62). Springer. https://doi.org/10.1007/978-0-387-87811-9_2
  15. [15] Savalia, S., & Emamian, V. (2018). Cardiac arrhythmia classification by multi-layer perceptron and convolution neural networks. Bioengineering, 5(2), 35. https://doi.org/10.3390/bioengineering5020035
  16. [16] Kancharla, N. K., & Brindha, V. (2024). Single and multi-lead ECG heartbeat classification using XGBoost on pre-trained CNN feature layer. https://doi.org/10.21203/rs.3.rs-4028614/v1
  17. [17] Pham, B. T., Le, P. T., Tai, T. C., Hsu, Y. C., Li, Y. H., & Wang, J. C. (2023). Electrocardiogram heartbeat classification for arrhythmias and myocardial infarction. Sensors, 23(6), 2993. https://doi.org/10.3390/s23062993
  18. [18] Cretu, I., Tindale, A., Abbod, M., Khir, A., Balachandran, W., & Meng, H. (2023). Multimodal arrhythmia classification using deep neural networks. Proceedings of the 9th World Congress on Electrical Engineering and Computer Systems and Sciences (EECSS’23) London, United Kingdom. Avestia Publishing. https://doi.org/10.11159/icbes23.152

Downloads

Published

2026-03-10

Issue

Vol. 3 No. 1 (2026): Trends in Health Informatics

Section

Articles

How to Cite

Kiani Sarkale, M., Poorahangaryan, F., & Kiani, A. . (2026). Optimized wavelet neural network for ECG arrhythmia detection. Trends in Health Informatics, 3(1), 22-35. https://doi.org/10.22105/thi.v3i1.48

Similar Articles

1-10 of 13

You may also start an advanced similarity search for this article.