HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Convolutional Neural Network and Rule-Based Algorithms for Classifying 12-lead ECGs

{Christian Tronstad Bjørn-Jostein Singstad}

Convolutional Neural Network and Rule-Based Algorithms for Classifying 12-lead ECGs

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

BenchmarkMethodologyMetrics
ecg-classification-on-physionet-challenge1D 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-challenge1D 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.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Convolutional Neural Network and Rule-Based Algorithms for Classifying 12-lead ECGs | Papers | HyperAI