Command Palette
Search for a command to run...
Convolutional Neural Network and Rule-Based Algorithms for Classifying 12-lead ECGs
{Christian Tronstad Bjørn-Jostein Singstad}

Abstract
The objective of this study was to classify 27 cardiac abnormalities based on a data set of 43 101 ECG recordings. A hybrid model combining a rule-based algorithm with different deep learning architectures was developed. We compared two different Convolutional Neural Networks; a Fully Convolutional Neural Network and an Encoder Network, a combination of both, and with the addition of another neural network using age and gender as input. Two of these combinations were finally combined with a rule-based model using derived ECG features. The performance of the models was evaluated on validation data during model development using hold-out validation. Finally, the models were deployed to a Docker image, trainedon the provided development data, and tested on the Challenge validation set. The model that performed best on the Challenge validation set was then deployed and tested on the full Challenge test set. The performance was evaluated based on a particular Challenge score.Our team, TeamUIO, achieved a Challenge validation score of 0.377, and a full test score of 0.206 for our best model. The score on the full test set placed us at 20th out of 41 teams in the official ranking.
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| ecg-classification-on-physionet-challenge | 1D CNN Fully Convolutional Network | Accuracy(stratified10-fold): 0,13±0,02 F1(stratified10-fold): 0,28±0,01 F2(stratified10-fold): 0,36±0,02 G2(stratified10-fold): 0,15±0,01 PhysioNet/CinC Challenge Score(stratified10-fold): 0,36±0,01 |
| ecg-classification-on-physionet-challenge | 1D CNN Encoder | Accuracy(stratified10-fold): 0,20±0,02 F1(stratified10-fold): 0,35±0,01 F2(stratified10-fold): 0,40±0,01 G2(stratified10-fold): 0,19±0,01 PhysioNet Challenge score (test data): 0.206 PhysioNet Challenge score 2020 (validation data): 0.377 PhysioNet/CinC Challenge Score(stratified10-fold): 0,37±0,03 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.