A Domain Generalization Approach for Out-Of-Distribution 12-lead ECG Classification with Convolutional Neural Networks

08/20/2022
by   Aristotelis Ballas, et al.
14

Deep Learning systems have achieved great success in the past few years, even surpassing human intelligence in several cases. As of late, they have also established themselves in the biomedical and healthcare domains, where they have shown a lot of promise, but have not yet achieved widespread adoption. This is in part due to the fact that most methods fail to maintain their performance when they are called to make decisions on data that originate from a different distribution than the one they were trained on, namely Out-Of-Distribution (OOD) data. For example, in the case of biosignal classification, models often fail to generalize well on datasets from different hospitals, due to the distribution discrepancy amongst different sources of data. Our goal is to demonstrate the Domain Generalization problem present between distinct hospital databases and propose a method that classifies abnormalities on 12-lead Electrocardiograms (ECGs), by leveraging information extracted across the architecture of a Deep Neural Network, and capturing the underlying structure of the signal. To this end, we adopt a ResNet-18 as the backbone model and extract features from several intermediate convolutional layers of the network. To evaluate our method, we adopt publicly available ECG datasets from four sources and handle them as separate domains. To simulate the distributional shift present in real-world settings, we train our model on a subset of the domains and leave-out the remaining ones. We then evaluate our model both on the data present at training time (intra-distribution) and the held-out data (out-of-distribution), achieving promising results and surpassing the baseline of a vanilla Residual Network in most of the cases.

READ FULL TEXT
research
03/20/2023

Towards Domain Generalization for ECG and EEG Classification: Algorithms and Benchmarks

Despite their immense success in numerous fields, machine and deep learn...
research
08/22/2021

A Transformer Architecture for Stress Detection from ECG

Electrocardiogram (ECG) has been widely used for emotion recognition. Th...
research
05/26/2023

CNN Feature Map Augmentation for Single-Source Domain Generalization

In search of robust and generalizable machine learning models, Domain Ge...
research
04/02/2023

CNNs with Multi-Level Attention for Domain Generalization

In the past decade, deep convolutional neural networks have achieved sig...
research
12/11/2021

Optimization of Residual Convolutional Neural Network for Electrocardiogram Classification

The interpretation of the electrocardiogram (ECG) gives clinical informa...
research
08/07/2022

Learning Modular Structures That Generalize Out-of-Distribution

Out-of-distribution (O.O.D.) generalization remains to be a key challeng...
research
10/15/2020

Combining Scatter Transform and Deep Neural Networks for Multilabel Electrocardiogram Signal Classification

An essential part for the accurate classification of electrocardiogram (...

Please sign up or login with your details

Forgot password? Click here to reset