Semi-supervised learning for ECG classification without patient-specific labeled data
Research output: Journal Publications and Reviews (RGC: 21, 22, 62) › 21_Publication in refereed journal › peer-review
Author(s)
Related Research Unit(s)
Detail(s)
Original language | English |
---|---|
Article number | 113411 |
Journal / Publication | Expert Systems with Applications |
Volume | 158 |
Online published | 7 May 2020 |
Publication status | Published - 15 Nov 2020 |
Link(s)
Abstract
In this paper, we propose a semi-supervised learning-based ECG classification system for detection of supraventricular ectopic beats (SVEB or S beats) and ventricular ectopic beats (VEB or V beats) which does not require manual labeling of the patient-specific ECG data. Owing to inter-subject variability in ECG signal, patient-specific data is usually required to achieve good performance in ECG classification system. However, manual labeling of patient-specific data requires expert intervention, which is costly and time consuming. Our proposed system is based on a 2D convolutional neural network (CNN) with inputs generated from heartbeat triplets. The system also consists of two auxiliary modules: a normal beat estimation module and an iterative beat label update algorithm. The normal beat estimation selects a small amount of patient-specific normal beats accurately from the testing ECG record in an unsupervised manner. These estimated normal beats are used, together with a common pool dataset, to train a preliminary patient-specific CNN classifier which provides initial labels for the testing data. These labels then undergo a semi-supervised iterative update process for improved performance. Our proposed system was evaluated on the MIT-BIH arrhythmia database. The training of our proposed system is fully automatic, and its performance is comparable with several state-of-art supervised methods which require extra manual labeling of patient-specific ECG data. Our proposed system can be a useful tool for batch processing a large amount of ECG data in clinical applications.
Research Area(s)
- Arrhythmia, CNN, ECG classification, Semi-supervised learning, Time series signal
Citation Format(s)
Semi-supervised learning for ECG classification without patient-specific labeled data. / Zhai, Xiaolong; Zhou, Zhanhong; Tin, Chung.
In: Expert Systems with Applications, Vol. 158, 113411, 15.11.2020.Research output: Journal Publications and Reviews (RGC: 21, 22, 62) › 21_Publication in refereed journal › peer-review