Using Multitask Learning to Improve 12-Lead Electrocardiogram Classification

12/03/2018
by   J. Weston Hughes, et al.
0

We develop a multi-task convolutional neural network (CNN) to classify multiple diagnoses from 12-lead electrocardiograms (ECGs) using a dataset comprised of over 40,000 ECGs, with labels derived from cardiologist clinical interpretations. Since many clinically important classes can occur in low frequencies, approaches are needed to improve performance on rare classes. We compare the performance of several single-class classifiers on rare classes to a multi-headed classifier across all available classes. We demonstrate that the addition of common classes can significantly improve CNN performance on rarer classes when compared to a model trained on the rarer class in isolation. Using this method, we develop a model with high performance as measured by F1 score on multiple clinically relevant classes compared against the gold-standard cardiologist interpretation.

READ FULL TEXT

page 1

page 2

page 3

page 4

research
06/30/2023

Dataset balancing can hurt model performance

Machine learning from training data with a skewed distribution of exampl...
research
12/29/2021

Two-phase training mitigates class imbalance for camera trap image classification with CNNs

By leveraging deep learning to automatically classify camera trap images...
research
12/13/2020

Improving the Classification of Rare Chords with Unlabeled Data

In this work, we explore techniques to improve performance for rare clas...
research
03/07/2019

Deep CNN-based Multi-task Learning for Open-Set Recognition

We propose a novel deep convolutional neural network (CNN) based multi-t...
research
11/28/2019

Detection and Mitigation of Rare Subclasses in Neural Network Classifiers

Regions of high-dimensional input spaces that are underrepresented in tr...
research
09/29/2020

EEMC: Embedding Enhanced Multi-tag Classification

The recently occurred representation learning make an attractive perform...
research
07/06/2017

Cardiologist-Level Arrhythmia Detection with Convolutional Neural Networks

We develop an algorithm which exceeds the performance of board certified...

Please sign up or login with your details

Forgot password? Click here to reset