Cross-Database and Cross-Channel ECG Arrhythmia Heartbeat Classification Based on Unsupervised Domain Adaptation

Md Niaz Imtiaz,Naimul Khan
2023-06-07
Abstract:The classification of electrocardiogram (ECG) plays a crucial role in the development of an automatic cardiovascular diagnostic system. However, considerable variances in ECG signals between individuals is a significant challenge. Changes in data distribution limit cross-domain utilization of a model. In this study, we propose a solution to classify ECG in an unlabeled dataset by leveraging knowledge obtained from labeled source domain. We present a domain-adaptive deep network based on cross-domain feature discrepancy optimization. Our method comprises three stages: pre-training, cluster-centroid computing, and adaptation. In pre-training, we employ a Distributionally Robust Optimization (DRO) technique to deal with the vanishing worst-case training loss. To enhance the richness of the features, we concatenate three temporal features with the deep learning features. The cluster computing stage involves computing centroids of distinctly separable clusters for the source using true labels, and for the target using confident predictions. We propose a novel technique to select confident predictions in the target domain. In the adaptation stage, we minimize compacting loss within the same cluster, separating loss across different clusters, inter-domain cluster discrepancy loss, and running combined loss to produce a domain-robust model. Experiments conducted in both cross-domain and cross-channel paradigms show the efficacy of the proposed method. Our method achieves superior performance compared to other state-of-the-art approaches in detecting ventricular ectopic beats (V), supraventricular ectopic beats (S), and fusion beats (F). Our method achieves an average improvement of 11.78% in overall accuracy over the non-domain-adaptive baseline method on the three test datasets.
Signal Processing,Artificial Intelligence
What problem does this paper attempt to address?
### What problems does this paper attempt to solve? This paper aims to solve the problem that electrocardiogram (ECG) signals across databases and channels vary significantly among different individuals, thus affecting the generalization ability of the model. Specifically, the paper focuses on how to achieve efficient arrhythmia heartbeat classification in unlabeled target - domain data. #### Main problems: 1. **Cross - domain adaptation problem**: Due to the significant differences in ECG signals among individuals, the performance of the trained model on new patient data drops significantly. This phenomenon is called "domain shift", that is, the distribution of test data is inconsistent with that of training data. 2. **Lack of labeled data**: Deep - learning models usually require a large amount of labeled data for training. However, in practical applications, obtaining sufficient labeled ECG data is both expensive and time - consuming. Especially when dealing with data from new participants, it is difficult to obtain labels, which limits the application of supervised learning methods. 3. **Improving classification accuracy**: The existing methods still have unsatisfactory performance when dealing with different types of arrhythmia (such as premature ventricular contractions, premature atrial contractions, fusion waves, etc.). Therefore, a method that can effectively meet these challenges is needed to improve classification accuracy. #### Solutions proposed in the paper: To solve the above problems, the authors propose a deep - learning model based on unsupervised domain adaptation (UDA). The model achieves cross - domain ECG heartbeat classification through the following three stages: 1. **Pre - training stage**: Use the labeled data in the source domain for pre - training. Adopt the distributionally robust optimization (DRO) technique to reduce the worst - case training loss and introduce two parallel classifiers to improve the stability of prediction. 2. **Clustering center calculation stage**: Calculate the clustering centers of the source domain and the target domain. By minimizing the intra - cluster compactness and inter - cluster separation losses, the model can better adapt to the target - domain data. 3. **Domain adaptation stage**: By introducing new loss functions (such as cross - domain clustering difference loss and running combination loss), minimize the feature distribution differences between the source domain and the target domain, thereby achieving a more robust model. Finally, this method can significantly improve the accuracy of cross - domain ECG heartbeat classification without additional expert labeling, especially in detecting premature ventricular contractions (V), premature atrial contractions (S) and fusion waves (F). ### Summary: The main contribution of this paper is to propose an effective unsupervised domain adaptation method, which solves the domain shift problem in cross - database and cross - channel ECG heartbeat classification, reduces the dependence on labeled data, and improves the classification accuracy.