QRS detection of ECG signal using U-Net and DBSCAN

QRS detection is a crucial task for ECG signal analysis, which is the preliminary and essential step to further recognition and diagnosis. This paper proposes a U-Net based method for QRS detection. The method consists of three steps including preprocessing, U-Net model, and density-based spatial clustering of applications with noise(DBSCAN). The normalization is carried out using the Z-score method in preprocessing. In this study, location prediction is conducted by the U-Net model. Subsequently, the U-Net outputs are thresholded and clustered by DBSCAN. Finally, the middle points of the cluster are regards as the R-peak of the QRS complex. We demonstrate that the proposed method achieving high accuracy on ECG signals from the MIT-BIH Arrhythmia Database(MITDB). The experimental results show an average sensitivity of 99.98 %, positive predictivity of 99.95 %, accuracy of 99.93 %, and F1-score of 99.97 %. Compared with other existing methods, the overall performance is comparable and even better in terms of accuracy and F1-score.

[1]  D.S. Benitez,et al.  A new QRS detection algorithm based on the Hilbert transform , 2000, Computers in Cardiology 2000. Vol.27 (Cat. 00CH37163).

[2]  G.B. Moody,et al.  The impact of the MIT-BIH Arrhythmia Database , 2001, IEEE Engineering in Medicine and Biology Magazine.

[3]  Yu Pu,et al.  Robust QRS Detection Using High-Resolution Wavelet Packet Decomposition and Time-Attention Convolutional Neural Network , 2020, IEEE Access.

[4]  Salah Hamdi,et al.  Real time QRS complex detection using DFA and regular grammar , 2017, BioMedical Engineering OnLine.

[5]  Sajjad Farashi,et al.  A multiresolution time-dependent entropy method for QRS complex detection , 2016, Biomed. Signal Process. Control..

[6]  Ghulam Muhammad,et al.  Development of the Arabic Voice Pathology Database and Its Evaluation by Using Speech Features and Machine Learning Algorithms , 2017, Journal of healthcare engineering.

[7]  Jeffrey M. Hausdorff,et al.  Physionet: Components of a New Research Resource for Complex Physiologic Signals". Circu-lation Vol , 2000 .

[8]  Manas Rakshit,et al.  An efficient wavelet-based automated R-peaks detection method using Hilbert transform , 2017 .

[9]  Marko Sarlija,et al.  A convolutional neural network based approach to QRS detection , 2017, Proceedings of the 10th International Symposium on Image and Signal Processing and Analysis.

[10]  Willis J. Tompkins,et al.  A Real-Time QRS Detection Algorithm , 1985, IEEE Transactions on Biomedical Engineering.

[11]  Kamalesh Kumar Sharma,et al.  QRS complex detection in ECG signals using locally adaptive weighted total variation denoising , 2017, Comput. Biol. Medicine.

[12]  U. Rajendra Acharya,et al.  Automated diagnosis of arrhythmia using combination of CNN and LSTM techniques with variable length heart beats , 2018, Comput. Biol. Medicine.

[13]  Ying Liu,et al.  ECG QRS Complex Detection Using Slope Vector Waveform (SVW) Algorithm , 2004, The 26th Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[14]  Moncef Gabbouj,et al.  Real-Time Patient-Specific ECG Classification by 1-D Convolutional Neural Networks , 2016, IEEE Transactions on Biomedical Engineering.

[15]  Hans-Peter Kriegel,et al.  A Density-Based Algorithm for Discovering Clusters in Large Spatial Databases with Noise , 1996, KDD.

[16]  Shintaro Izumi,et al.  Noise tolerant QRS detection using template matching with short-term autocorrelation , 2014, 2014 36th Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[17]  Jianqing Li,et al.  An Adaptive and Time-Efficient ECG R-Peak Detection Algorithm , 2017, Journal of healthcare engineering.

[18]  Xingjun Wang,et al.  QRS Detection in ECG Signal Based on Residual Network , 2019, 2019 IEEE 11th International Conference on Communication Software and Networks (ICCSN).

[19]  Minho Choi,et al.  QRS detection method based on fully convolutional networks for capacitive electrocardiogram , 2019, Expert Syst. Appl..

[20]  Zhenmin Tang,et al.  Sparse representation-based ECG signal enhancement and QRS detection , 2016, Physiological measurement.

[21]  Chengyu Liu,et al.  A Faster R CNN-Based Real-Time QRS Detector , 2019, 2019 Computing in Cardiology (CinC).

[22]  Pornchai Phukpattaranont,et al.  QRS detection algorithm based on the quadratic filter , 2015, Expert Syst. Appl..

[23]  Chi-Sang Poon,et al.  Analysis of First-Derivative Based QRS Detection Algorithms , 2008, IEEE Transactions on Biomedical Engineering.

[24]  Zhitao Lin,et al.  Automatic QRS complex detection using two-level convolutional neural network , 2018, BioMedical Engineering OnLine.

[25]  Vincenzo Piuri,et al.  Deep-ECG: Convolutional Neural Networks for ECG biometric recognition , 2019, Pattern Recognit. Lett..

[26]  Durbadal Mandal,et al.  An optimally designed digital differentiator based preprocessor for R-peak detection in electrocardiogram signal , 2019, Biomed. Signal Process. Control..

[27]  M. Sabarimalai Manikandan,et al.  A novel method for detecting R-peaks in electrocardiogram (ECG) signal , 2012, Biomed. Signal Process. Control..

[28]  Wenjie Cai,et al.  QRS Complex Detection Using Novel Deep Learning Neural Networks , 2020, IEEE Access.

[29]  Masoumeh Haghpanahi,et al.  Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network , 2019, Nature Medicine.

[30]  Sadasivan Puthusserypady,et al.  DENS-ECG: A Deep Learning Approach for ECG Signal Delineation , 2020, Expert Syst. Appl..

[31]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[32]  Abdelmalik Taleb-Ahmed,et al.  R-peaks detection based on stationary wavelet transform , 2015, Comput. Methods Programs Biomed..