Deep Representation Learning With Sample Generation and Augmented Attention Module for Imbalanced ECG Classification

doi:10.1109/JBHI.2023.3325540

Deep Representation Learning With Sample Generation and Augmented Attention Module for Imbalanced ECG Classification

2024-05-01

5

OA

AI

DOI：10.1109/JBHI.2023.3325540

Original

Original request for help

Save

Abstract

En 中文

Developing an efficient heartbeat monitoring system has become a focal point in numerous healthcare applications. Specifically, in the last few years, heartbeat classification for arrhythmia detection has gained considerable interest from researchers. This paper presents a novel deep representation learning method for the efficient detection of arrhythmic beats. To mitigate the issues associated with the imbalanced data distribution, a novel re-sampling strategy is introduced. Unlike the existing oversampling methods, the proposed technique transforms majority-class samples into minority-class samples with a novel translation loss function. This approach assists the model in learning a more generalized representation of crucially important minority class samples. Moreover, by exploiting an auxiliary feature, an augmented attention module is designed that focuses on the most relevant and target-specific information. We adopted an inter-patient classification paradigm to evaluate the proposed method. The experimental results of this study on the MIT-BIH arrhythmia database clearly indicate that the proposed model with augmented attention mechanism and over-sampling strategy significantly learns a balanced deep representation and improves the classification performance of vital heartbeats.

Keywords:

Arrhythmia detection

beat classification

imbalanced learning

remote health monitoring

Deep Representation Learning With Sample Generation and Augmented Attention Module for Imbalanced ECG Classification

Abstract

Journal

Organization