ECG Classification With a Convolutional Recurrent Neural Network
No Thumbnail Available
Author
Sigurthorsdottir, Halla
Van Zaen, Jérôme
Delgado-Gonzalo, Ricard
Lemay, Mathieu
Abstract
We developed a convolutional recurrent neural network to classify 12-lead ECG signals for the challenge of PhysioNet/Computing in Cardiology 2020 as team Pink Irish Hat. The model combines convolutional and recurrent layers, takes sliding windows of ECG signals as input and yields the probability of each class as output. The convolutional part extracts features from each sliding window. The bi-directional gated recurrent unit (GRU) layer and an attention layer aggregate these features from all windows into a single feature vector. Finally, a dense layer outputs class probabilities. The ?nal decision is made using test time augmentation (TTA) and an optimized decision threshold. Several hyperparameters of our architecture were optimized, the most important of which turned out to be the choice of optimizer and the number of ?lters per convolutional layer. Our network achieved a challenge score of 0.511 on the hidden validation set and 0.167 on the full hidden test set, ranking us 24th out of 41 in the of?cial ranking.
Publication Reference
2020 Computing in Cardiology Conference, Rimini (Italy)
Year
2020-12-30