A token selection-based multi-scale dual-branch CNN-transformer network for 12-lead ECG signal classification

被引:15
作者
Zhang, Siyuan [1 ]
Lian, Cheng [1 ]
Xu, Bingrong [1 ]
Zang, Junbin [2 ]
Zeng, Zhigang [3 ]
机构
[1] Wuhan Univ Technol, Sch Automat, Wuhan 430074, Peoples R China
[2] North Univ China, Sch Instrument & Elect, Taiyuan 038507, Peoples R China
[3] Huazhong Univ Sci & Technol, Sch Artificial Intelligence & Automat, Wuhan 430074, Peoples R China
关键词
12-lead ECG signal classification; Convolutional neural network; Transformer; Multi-scale learning; Token selection; Information redundancy; ARRHYTHMIA DETECTION; MODEL;
D O I
10.1016/j.knosys.2023.111006
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
The timely identification of cardiovascular diseases is critical for effective intervention, with the electrocardiogram (ECG) serving as a pivotal diagnostic tool. Recent advancements in deep learning -based methods have significantly enhanced the accuracy of ECG signal classification. In clinical settings, cardiologists rely on diagnoses derived from standardized 12-lead ECG recordings. It must be acknowledged that there is considerable redundancy in the 12-lead ECG recordings used for ECG signal classification, thereby hindering their generalization capabilities. Meanwhile, considering multi-scale features in 12-lead ECG recordings is a crucial aspect that is often overlooked by existing methods. Based on the above observations, we develop a multi-scale Convolutional Transformer network for ECG signal classification. By utilizing learnable Convolutional neural network (CNN) blocks and novel dual-branch Transformer encoders, the proposed network automatically extracts features at different scales, resulting in superior feature representation. Additionally, by discarding low-importance patches and focusing on high-importance patches, we effectively alleviate information redundancy in the 12 -lead ECG recordings. We conduct comprehensive experiments on three commonly used ECG datasets. The Research results show that our proposed network outperforms existing state-of-the-art networks in multiple tasks.(c) 2023 Elsevier B.V. All rights reserved.
引用
收藏
页数:12
相关论文
共 35 条
[1]   DSP-based arrhythmia classification using wavelet transform and probabilistic neural network [J].
Antonio Gutierrez-Gnecchi, Jose ;
Morfin-Magana, Rodrigo ;
Lorias-Espinoza, Daniel ;
del Carmen Tellez-Anguiano, Adriana ;
Reyes-Archundia, Enrique ;
Mendez-Patino, Arturo ;
Castaneda-Miranda, Rodrigo .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2017, 32 :44-56
[2]  
Chen MZ, 2022, Arxiv, DOI arXiv:2203.03821
[3]   Single-Chain Lanthanide Luminescence Biosensors for Cell-Based Imaging and Screening of Protein-Protein Interactions [J].
Chen, Ting ;
Ha Pham ;
Mohamadi, Ali ;
Miller, Lawrence W. .
ISCIENCE, 2020, 23 (09)
[4]  
Dosovitskiy A, 2021, Arxiv, DOI arXiv:2010.11929
[5]   InceptionTime: Finding AlexNet for time series classification [J].
Fawaz, Hassan Ismail ;
Lucas, Benjamin ;
Forestier, Germain ;
Pelletier, Charlotte ;
Schmidt, Daniel F. ;
Weber, Jonathan ;
Webb, Geoffrey, I ;
Idoumghar, Lhassane ;
Muller, Pierre-Alain ;
Petitjean, Francois .
DATA MINING AND KNOWLEDGE DISCOVERY, 2020, 34 (06) :1936-1962
[6]   A Machine-Learning Approach for Detection and Quantification of QRS Fragmentation [J].
Goovaerts, Griet ;
Padhy, Sibasankar ;
Vandenberk, Bert ;
Varon, Carolina ;
Willems, Rik ;
Van Huffel, Sabine .
IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2019, 23 (05) :1980-1989
[7]   Multimodal multi-instance learning for long-term ECG classification [J].
Han, Haozhan ;
Lian, Cheng ;
Zeng, Zhigang ;
Xu, Bingrong ;
Zang, Junbin ;
Xue, Chenyang .
KNOWLEDGE-BASED SYSTEMS, 2023, 270
[8]   Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network [J].
Hannun, Awni Y. ;
Rajpurkar, Pranav ;
Haghpanahi, Masoumeh ;
Tison, Geoffrey H. ;
Bourn, Codie ;
Turakhia, Mintu P. ;
Ng, Andrew Y. .
NATURE MEDICINE, 2019, 25 (01) :65-+
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]   Bag of Tricks for Image Classification with Convolutional Neural Networks [J].
He, Tong ;
Zhang, Zhi ;
Zhang, Hang ;
Zhang, Zhongyue ;
Xie, Junyuan ;
Li, Mu .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :558-567