1. Inter-patient ECG arrhythmia heartbeat classification based on unsupervised domain adaptation
- Author
-
Huazhong Yang, Guijin Wang, Jiawei Li, Ming Chen, Zijian Ding, and Ping Zhang
- Subjects
0209 industrial biotechnology ,Domain adaptation ,Heartbeat ,medicine.diagnostic_test ,business.industry ,Computer science ,Generalization ,Cognitive Neuroscience ,Pattern recognition ,02 engineering and technology ,Computer Science Applications ,Statistical classification ,ComputingMethodologies_PATTERNRECOGNITION ,020901 industrial engineering & automation ,Artificial Intelligence ,0202 electrical engineering, electronic engineering, information engineering ,medicine ,Labeled data ,020201 artificial intelligence & image processing ,Artificial intelligence ,business ,Cluster analysis ,Electrocardiography ,Test data - Abstract
Electrocardiography (ECG) arrhythmia heartbeat classification is essential for automatic cardiovascular diagnosis system. However, the enormous differences of ECG signals among individuals and high price of labeled data have brought huge challenges for current classification algorithms based on deep neural networks and prevented these models from achieving satisfactory performance on new data. In order to build a classification system with better adaptability, we propose a novel Domain-Adaptative ECG Arrhythmia Classification (DAEAC) model based on convolutional network and unsupervised domain adaptation (UDA). Based on observation of clustering characteristics of data, we present two original objective functions to enhance the inter-patient performance. A Cluster-Aligning loss is presented to align the distributions of training data and test data. Simultaneously, a Cluster-Maintaining loss is proposed to reinforce the discriminability and structural information of features. The proposed method requires no expert annotations but a short period of unsupervised data in new records to make deep models more adaptive. Extensive experimental results on three public databases demonstrate that our method achieves competitive performance with other state-of-the-arts on the detection of ventricular ectopic beats (V), supraventricular ectopic beats (S) and fusion beats (F). The cross-dataset experimental results also verify the great generalization capability.
- Published
- 2021
- Full Text
- View/download PDF