An ECG Beat Classification Method using Multi-kernel ResNet with Transformer

In this work, we investigated the electrocardiogram (ECG) for cardiovascular disease diagnosis using the deep learning method. The diagnosis of a more complicated cardiovascular disease can be performed by analyzing beat-unit abnormalities. Beat classification plays the basic role of classifying arr...

Full description

Saved in:
Bibliographic Details
Published in2023 IEEE International Conference on Big Data and Smart Computing (BigComp) pp. 140 - 144
Main Authors Chon, Sangil, Ha, Kwon-Woo, Park, Seongjae, Jung, Sunghoon
Format Conference Proceeding
LanguageEnglish
Published IEEE 01.02.2023
Subjects
Arrhythmia
Beat classification
Data models
Deep learning
ECG
Electrocardiography
Feature extraction
Heart rate
Multikernel ResNet
Position embedding
Transfer learning
Transformer
Transformers
Online AccessGet full text

Cover

More Information
Summary:In this work, we investigated the electrocardiogram (ECG) for cardiovascular disease diagnosis using the deep learning method. The diagnosis of a more complicated cardiovascular disease can be performed by analyzing beat-unit abnormalities. Beat classification plays the basic role of classifying arrhythmia in ECG. To classify the four beat classes of the MIT-BIH arrhythmia data base, we extracted features using the multi-kernel ResNet, trained the system to successfully learn the mutual information of continuous beats using the transformer encoder, and enabled the transformer to learn stably by applying position embedding based on the heart rate. We used the icentiallk data set, which is a big public data set, for the transfer-learning and could make a more general model with a variety of patient data. We obtained an average F1 score of 0.839, which is better than any existing score, for the four-class beat classification of the MIT-BIH arrhythmia database.
ISSN:2375-9356
DOI:10.1109/BigComp57234.2023.00031