CNN-based Two Step R Peak Detection Method: Combining Segmentation and Regression.

For semantic segmentation, U-Net provides an end-to-end trainable framework to detect multiple class objects from background. Due to its great achievements in computer vision tasks, U-Net has broadened its application to biomedical signal processing, especially, segmentation of waveforms in ECG signal. Despite its superior performance for QRS complex detection to other traditional signal processing methods, direct application of the U-Net to R peak detection has limitation since the U-Net structures tend to predict high probability around true peak. Such multiple detection results require additional process to determine a unique peak location in each QRS complex. In this study, we use a regression process to detect R peak instead of pixel-wise classification. Such regression process guarantees a unique peak location prediction. We collect data from resting ECG systems and wearable ECG devices as well as public ECG databases and the proposed model is trained on various combinations of the data sources. Especially, we investigate the robustness of the model for input data from the wearable devices when the model is trained by data from heterogeneous devices.