Journal of Sensor Science and Technology (센서학회지)

The Korean Sensors Society (한국센서학회)
권호 표지
DOI QR코드

DOI QR Code

Comparison of Artificial Neural Networks for Low-Power ECG-Classification System

  • Rana, Amrita (Department of Electronic Engineering, Daegu University) ;
  • Kim, Kyung Ki (Department of Electronic Engineering, Daegu University)
  • Received : 2020.01.22
  • Accepted : 2020.01.31
  • Published : 2020.01.31
Download PDF
⟨ Previous Next ⟩

Abstract

Electrocardiogram (ECG) classification has become an essential task of modern day wearable devices, and can be used to detect cardiovascular diseases. State-of-the-art Artificial Intelligence (AI)-based ECG classifiers have been designed using various artificial neural networks (ANNs). Despite their high accuracy, ANNs require significant computational resources and power. Herein, three different ANNs have been compared: multilayer perceptron (MLP), convolutional neural network (CNN), and spiking neural network (SNN) only for the ECG classification. The ANN model has been developed in Python and Theano, trained on a central processing unit (CPU) platform, and deployed on a PYNQ-Z2 FPGA board to validate the model using a Jupyter notebook. Meanwhile, the hardware accelerator is designed with Overlay, which is a hardware library on PYNQ. For classification, the MIT-BIH dataset obtained from the Physionet library is used. The resulting ANN system can accurately classify four ECG types: normal, atrial premature contraction, left bundle branch block, and premature ventricular contraction. The performance of the ECG classifier models is evaluated based on accuracy and power. Among the three AI algorithms, the SNN requires the lowest power consumption of 0.226 W on-chip, followed by MLP (1.677 W), and CNN (2.266 W). However, the highest accuracy is achieved by the CNN (95%), followed by MLP (76%) and SNN (90%).

Keywords

References

  1. R. Lozano, M. Naghavi, K. Foreman, S. Lim, K. Shibuya, V. Aboyans, J. Abraham, T. Adair, R. Aggarwal, S. Y. Ahn, et al., "Global and regional mortality from 235 causes of death for 20 age groups in 1990 and 2010: a systematic analysis for the global burden of disease study 2010", The Lancet, Vol. 380, No. 9859, pp. 2095-2128, 2013. https://doi.org/10.1016/S0140-6736(12)61728-0
  2. M. Mitra and R. K Samanta, "Cardiac Arrhythmia Classification Using Neural Networks with Selected Features: first international conference on Computational Intelligence: Modeling Techniques and Applications (CIMTA)", pp. 76-84, 2013.
  3. T. Habte, T. Saleh, H. Mohammad, B, & Ismail, M., "Ultra Low Power ECG Processing System for IoT Devices: Analog Circuits and Signal Processing", pp. 1-14, 2019.
  4. T. E. Taha, A. E. Sayed, and S. Rafat, "A survey on Classification of ECG Signal Study", Vol. 6, No. 5, pp. 11-16, 2016.
  5. C. V. Maggio, A., Bonomini, M., Laciar, E., and Arini, P., "Quantification of Ventricular Repolarization Dispersion Using Digital Processing of the Surface ECG" , 2012.
  6. M. Aqil,A. Jbari, and A. Bourouhou, "ECG signal denoising by Discrete Wavelet Transform", Vol. 13, No. 09, pp. 51-68, 2017. https://doi.org/10.3991/ijoe.v13i09.7159
  7. https://blog.octo.com/en/time-series-features-extraction-usingfourier-and-wavelet-transforms-on-ecg-data/, (retrieved on Nov. 23, 2019).
  8. Bultheel Bull. Belg. B. Yu, M. Vetterli, "Chapter 4. Data Processing and Analysis, Math Wavelet Transform" IEEE Pub., pp. 35-42, 1995.
  9. M. Sanaullah, "A review of higher order statistics and spectra in communication systems", Vol. 13, No. 4, pp. 31-50, 2013.
  10. https://towardsdatascience.com/a-one-stop-shop-for-principal-component-analysis-5582fb7e0a9c(retrieved on Apr 18, 2017).
  11. E. D. Ubeyli, "Combining recurrent neural networks with eigenvector methods for classification of ECG beats. Digital Signal Processing", Vol. 19, Issue 2, pp. 320-329, 2009. https://doi.org/10.1016/j.dsp.2008.09.002
  12. J. Park, K. Lee, K. Kang, "Arrhythmia detection from heartbeat using the k-nearest neighbor classifier." IEEE International Conference on Bioinformatics and Biomedicine, pp 15-22 (2013).
  13. S. Kiranyaz, T. Ince, M. Gabbouj, "Real-time patient-specific ECG classification by 1-D convolutional neural networks", IEEE Transactions on Biomedical Engineering, Vol. 63, No. 3, pp.664-675, 2016. https://doi.org/10.1109/TBME.2015.2468589
  14. T. J. Jun, H. M. Nguyen, D. Kang, D. Kim, D. Kim, and Y.H. Kim, "ECG arrhythmia classification using a 2-D convolutional neural network", 18 Apr 2018.
  15. S. Savalia and V. Emamian, "Cardiac Classification by Multi-Layer Perceptron and Convolutional Neural Networks", Vol. 5, No. 35, pp. 1-12, 2018. https://doi.org/10.3934/bioeng.2018.1.1
  16. A. Das, P. Pradhapan, W. Groenendaal, P. Adiraju, R. T. Rajan, F. Catthoor, S. Schaafsma, J.L, Krichmar, N.Dutt, and C. Van Hoot, "Unsupervised heart-rate estimation in wearable with liquid states and a probabilistic readout", Neural Networks. Vol.99, pp.134-147, 2018. https://doi.org/10.1016/j.neunet.2017.12.015
  17. A. Amirshahi and M. Hashemi, "Ultra Low-Power and Real-time ECG classification Based on STDP and R-STDP Neural networks for Wearable Devices", IEEE Transactions on Biomedical Circuits and Systems (TBioCAS), Vol. 13, Issue. 6, pp. 1483-1493, 2019. https://doi.org/10.1109/tbcas.2019.2948920
  18. K. Guo, J. Yu, Y. Wang, and H. Yang, "A survey of FPGA Based Neural Network Accelerator", ACM Transactions on Reconfigurable Technology and Systems, Vol. 9, No. 4, Article 11, Dec 2017.
  19. C. V. Nguyen, T. Le Quang, T. N. Vu, H. Le Thi, K. N. Van, T. H. Trong, K. Ishibashi, "A non-contact infection screening system using medical radar and Linux-embedded FPGA: Implementation and preliminary validation.", Informatics in Medicine Unlocked, Vol.16, pp. 1-9, 2019.