Building and training a deep spiking neural network for ECG classification

Abstract

The electrocardiogram (ECG) reflects the electrical activity of the heart, and is one the most widely used biophysical signals that evaluate heart-related conditions. With years of experiences, medical professionals are able to identify and classify various ECG patterns. However, manually classifying ECG signals is prone to errors and takes considerable amount of time and effort, and thus people start to explore computational models for ECG classification. In recent years, deep artificial neural networks (ANNs) have gained increasing popularity in many fields for their outstanding performances. Traditional ANNs consist of computational units which are inspired from biological neurons but ignore the neural signal transmission details. Spiking neural networks (SNNs), on the other hand, are based on impulse neurons that more closely mimic biological neurons, and thus have a great potential to achieve similar performance with much less power. Nevertheless, SNNs have not become prevalent, and one of the primary reasons is that training SNNs especially the ones with deep structures remains a challenge. In this paper, we aim to propose an efficient way to build and train a deep SNN for ECG classification by constructing a counterpart structure of a deep ANN, transferring the trained parameters, and replacing the activation functions with leaky integrate-and-fire (LIF) neurons. The results show that the accuracy of the deep SNN even exceeds the original ANN. In addition, we compare and discuss the effects of different ANN activation functions on the SNN performance.