Diagnosis of atrial fibrillation based on unsupervised domain adaptation

被引:10
作者
Du, Mingyu [1 ,2 ,3 ]
Yang, Yuan [1 ,2 ,3 ]
Zhang, Lin [1 ,2 ,3 ]
机构
[1] Beijing Adv Innovat Ctr Big Data Based Precis Med, Sch Med & Engn, 37 Xueyuan Rd, Beijing, Peoples R China
[2] Minist Ind & Informat Technol, Key Lab Big Data Based Precis Med, 37 Xueyuan Rd, Beijing, Peoples R China
[3] Beihang Univ, Sch Automat Sci & Elect Engn, 37 Xueyuan Rd, Beijing, Peoples R China
关键词
Deep learning; Unsupervised domain adaptation; Atrial fibrillation diagnosis; Electrocardiogram; CONVOLUTIONAL NEURAL-NETWORK; DEEP LEARNING APPROACH; HEARTBEAT CLASSIFICATION; ECG; SIGNALS;
D O I
10.1016/j.compbiomed.2023.107275
中图分类号
Q [生物科学];
学科分类号
07 ; 0710 ; 09 ;
摘要
In recent years, the proportion of the elderly in the society is continuously increasing. Cardiovascular disease is a big problem that puzzles the health of the elderly. Among them, atrial fibrillation is one of the most common arrhythmia diseases in recent years, which poses a great threat to human life safety. At the same time, deep learning has become a powerful tool for medical and healthcare applications due to its high accuracy and fast detection speed. The diagnosis of atrial fibrillation is based on electrocardiogram, ECG) timing signals. At pre-sent, the scale of the open ECG data set is limited, and a large amount of labeled ECG data is needed to build a high-precision diagnostic model. In this study, a two-channel network model and a feature queue technique are proposed. A high-quality classification diagnosis model of atrial fibrillation is obtained by unsupervised domain adaptive technique, which uses a small amount of labeled data and a large amount of unlabeled data for training. The research content of this paper includes the following aspects: 1) Build a dual-channel network model, which can analyze ECG signals from different feature dimensions. At the same time, the dual-channel output also improves the reliability of the model's pseudo-label in the adaptive training stage and the accuracy of the output in the testing stage. 2) Innovative feature queue technology including global centroid is proposed to participate in the process of domain discrepancy metric calculation, which can use a small amount of labeled data and a large amount of unlabeled data to achieve a more stable and rapid update of the network. 3) Improved and innovated the domain discrepancy metric function, and introduced an evaluation formula for the credibility of false labels to improve the learning efficiency of unlabeled data. Finally, the experimental results show that the proposed two-channel network model and the feature queue technique with global centroid can achieve a high generalization and high precision depth network model by training with a small amount of labeled data and a large amount of unlabeled data. 4) The proposed model achieved a precision of 95.12%, a recall of 95.36%, an accuracy of 98.05%, and an F1 score of 95.23% in the MIT-BIH Arrhythmia Database. In the MIT-BIH Atrial Fibrillation Database, the model achieved a precision of 98.9%, a recall of 99.03%, an accuracy of 99.13%, and an F1 score of 99.08%.
引用
收藏
页数:14
相关论文
共 35 条
[1]   Deep convolutional neural network for the automated diagnosis of congestive heart failure using ECG signals [J].
Acharya, U. Rajendra ;
Fujita, Hamido ;
Oh, Shu Lih ;
Hagiwara, Yuki ;
Tan, Jen Hong ;
Adam, Muhammad ;
Tan, Ru San .
APPLIED INTELLIGENCE, 2019, 49 (01) :16-27
[2]   Detection of premature ventricular contraction arrhythmias in electrocardiogram signals with kernel methods [J].
Alajlan, Naif ;
Bazi, Yakoub ;
Melgani, Farid ;
Malek, Salim ;
Bencherif, Mohamed A. .
SIGNAL IMAGE AND VIDEO PROCESSING, 2014, 8 (05) :931-942
[3]   Advanced K-means clustering algorithm for large ECG data sets based on a collaboration of compressed sensing theory and K-SVD approach [J].
Balouchestani, Mohammadreza ;
Krishnan, Sridhar .
SIGNAL IMAGE AND VIDEO PROCESSING, 2016, 10 (01) :113-120
[4]   CVAE-GAN: Fine-Grained Image Generation through Asymmetric Training [J].
Bao, Jianmin ;
Chen, Dong ;
Wen, Fang ;
Li, Houqiang ;
Hua, Gang .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :2764-2773
[5]  
Chen M, 2020, IEEE ENG MED BIO, P304, DOI [10.1109/EMBC44109.2020.9175928, 10.1109/embc44109.2020.9175928]
[6]  
Chen Ting, 2019, ICML
[7]   A patient-adapting heartbeat classifier using ECG morphology and heartbeat interval features [J].
de Chazal, Philip ;
Reilly, Richard B. .
IEEE TRANSACTIONS ON BIOMEDICAL ENGINEERING, 2006, 53 (12) :2535-2543
[8]   Automatic Detection of Atrial Fibrillation from Single-Lead ECG Using Deep Learning of the Cardiac Cycle [J].
Dubatovka, Alina ;
Buhmann, Joachim M. .
BME FRONTIERS, 2022, 2022
[9]   Correlation technique and least square support vector machine combine for frequency domain based ECG beat classification [J].
Dutta, Saibal ;
Chatterjee, Amitava ;
Munshi, Sugata .
MEDICAL ENGINEERING & PHYSICS, 2010, 32 (10) :1161-1169
[10]  
Goodfellow IJ, 2014, ADV NEUR IN, V27, P2672