Abstract
Arrhythmia detection is an important research task in healthcare, which can be accomplished by classifying ECG heartbeats. Recently, there is a growing trend of applying deep learning models to solving this problem. Most existing deep learning-based methods consist of three steps: preprocessing, heartbeat segmentation and beat-wise classification. This methodology suffers from two drawbacks. First, explicit heartbeat segmentation can undermine the simplicity of the entire model. Second, performing classification on individual heartbeats fails to take into account inter-heartbeat contextual information that may be vital to accurate classification. In view of these drawbacks, we propose a novel deep learning model that simultaneously conducts heartbeat segmentation and classification. Without explicit heartbeat segmentation, the overall workflow of our method is streamlined to be simpler than existing methods. We achieve simultaneous segmentation and classification with a Faster R-CNN-based deep network that has been customized to handle ECG data. To capture inter-heartbeat contextual information, we utilize inverted residual blocks and a novel feature fusion and normalization subroutine which incorporates average pooling and max-pooling. We conduct extensive experiments on the well-known MIT-BIH arrhythmia database to validate the effectiveness of our method in both intra- and inter-patient arrhythmia detection tasks.
Similar content being viewed by others
References
Acharya, U.R., Oh, S.L., Hagiwara, Y., Tan, J.H., Adam, M., Gertych, A., San Tan, R.: A deep convolutional neural network model to classify heartbeats. Comput. Biol. Med. 89, 389–396 (2017)
Alickovic, E., Subasi, A.: Medical decision support system for diagnosis of heart arrhythmia using dwt and random forests classifier. J. Med. Syst. 40(4), 108 (2016)
Atkielsk, A.: Atom. File:SinusRhythm withoutLabels.svg (2019)
Chawla, N.V., Bowyer, K.W., Hall, L.O., Kegelmeyer, W.P.: Smote: synthetic minority over-sampling technique. J. Artif. Intell. Res. 16, 321–357 (2002)
Chen, S., Hua, W., Li, Z., Li, J., Gao, X.: Heartbeat classification using projected and dynamic features of ecg signal. Biomed. Signal Process. Control 31, 165–173 (2017)
Clevert, DA., Unterthiner, T., Hochreiter, S.: Fast and accurate deep network learning by exponential linear units (elus). arXiv preprint arXiv:1511.07289 (2015)
De Chazal, P., O’Dwyer, M., Reilly, R.B.: Automatic classification of heartbeats using ECG morphology and heartbeat interval features. IEEE Trans. Biomed. Eng. 51(7), 1196–1206 (2004)
EC57 AA, for the Advancement of Medical Instrumentation A, et al.: Testing and reporting performance results of cardiac rhythm and st segment measurement algorithms. Association for the Advancement of Medical Instrumentation, Arlington, VA (1998)
Fawaz, H.I., Forestier, G., Weber, J., Idoumghar, L., Muller, P.A.: Deep learning for time series classification: a review. Data Min. Knowl. Disc. 33(4), 917–963 (2019)
Goldberger, A.L., Amaral, L.A., Glass, L., Hausdorff, J.M., Ivanov, P.C., Mark, R.G., Mietus, J.E., Moody, G.B., Peng, C.K., Stanley, H.E.: Physiobank, physiotoolkit, and physionet: components of a new research resource for complex physiologic signals. Circulation 101(23), e215–e220 (2000)
Hamilton, P.S., Tompkins, W.J.: Quantitative investigation of qrs detection rules using the mit/bih arrhythmia database. IEEE Trans. Biomed. Eng. 12, 1157–1165 (1986)
Hannun, A.Y., Rajpurkar, P., Haghpanahi, M., Tison, G.H., Bourn, C., Turakhia, M.P., Ng, A.Y.: Cardiologist-level arrhythmia detection and classification in ambulatory electrocardiograms using a deep neural network. Nat. Med. 25(1), 65 (2019)
Hao, P., Gao, X., Li, Z., Zhang, J., Wu, F., Bai, C.: Multi-branch fusion network for myocardial infarction screening from 12-lead ecg images. Comput. Methods Prog. Biomed. 184, 105286 (2020)
He, J., Rong, J., Sun, L., Wang, H., Zhang, Y.: An advanced two-step dnn-based framework for arrhythmia detection. In: Pacific-Asia Conference on Knowledge Discovery and Data Mining, pp. 422–434. Springer, New York (2020)
He, Z., Niu, J., Ren, J., Shi, Y., Zhang, W.: A deep learning method for heartbeat detection in ECG image. In: Chinese Intelligent Automation Conference, pp. 356–363. Springer, New York (2019)
Ji, Y., Zhang, S., Xiao, W.: Electrocardiogram classification based on faster regions with convolutional neural network. Sensors 19(11), 2558 (2019)
Kiranyaz, S., Ince, T., Gabbouj, M.: Real-time patient-specific ECG classification by 1-d convolutional neural networks. IEEE Trans. Biomed. Eng. 63(3), 664–675 (2015)
Lin, T.Y., Dollár, P., Girshick, R., He, K., Hariharan, B., Belongie, S.: Feature pyramid networks for object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2117–2125 (2017)
Liu, F., Zhou, X., Cao, J., Wang, Z., Wang, H., Zhang, Y.: Arrhythmias classification by integrating stacked bidirectional lstm and two-dimensional cnn. In: Pacific-Asia Conference on Knowledge Discovery and Data Mining, pp. 136–149. Springer, New York (2019a)
Liu, F., Zhou, X., Cao, J., Wang, Z., Wang, H., Zhang, Y.: A lstm and cnn based assemble neural network framework for arrhythmias classification. ICASSP 2019–2019 IEEE International Conference on Acoustics, pp. 1303–1307. Speech and Signal Processing (ICASSP), IEEE (2019b)
Luz, E.J.S., Schwartz, W.R., Cámara-Chávez, G., Menotti, D.: Ecg-based heartbeat classification for arrhythmia detection: a survey. Comput. Methods Programs Biomed. 127, 144–164 (2016)
Ma, J., Sun, L., Wang, H., Zhang, Y., Aickelin, U.: Supervised anomaly detection in uncertain pseudoperiodic data streams. ACM Trans. Internet Technol. (TOIT) 16(1), 1–20 (2016)
Martínez, J.P., Almeida, R., Olmos, S., Rocha, A.P., Laguna, P.: A wavelet-based ECG delineator: evaluation on standard databases. IEEE Trans. Biomed. Eng. 51(4), 570–581 (2004)
Martis, R.J., Acharya, U.R., Min, L.C.: ECG beat classification using pca, lda, ica and discrete wavelet transform. Biomed. Signal Process. Control 8(5), 437–448 (2013)
Melgani, F., Bazi, Y.: Classification of electrocardiogram signals with support vector machines and particle swarm optimization. IEEE Trans. Inf Technol. Biomed. 12(5), 667–677 (2008)
Moody, G.B., Mark, R.G.: The impact of the mit-bih arrhythmia database. IEEE Eng. Med. Biol. Mag. 20(3), 45–50 (2001)
Mousavi, S., Afghah, F.: Inter-and intra-patient ECG heartbeat classification for arrhythmia detection: a sequence to sequence deep learning approach. ICASSP 2019–2019 IEEE International Conference on Acoustics, pp. 1308–1312. Speech and Signal Processing (ICASSP), IEEE (2019)
Oh, S.L., Ng, E.Y., San Tan, R., Acharya, U.R.: Automated beat-wise arrhythmia diagnosis using modified u-net on extended electrocardiographic recordings with heterogeneous arrhythmia types. Comput. Biol. Med. 105, 92–101 (2019)
Pan, J., Tompkins, W.J.: A real-time qrs detection algorithm. IEEE Trans. Biomed. Eng. 32(3), 230–236 (1985)
Podrid, P.J., Kowey, P.R.: Cardiac Arrhythmia: Mechanisms, Diagnosis, and Management. Lippincott Williams & Wilkins, Philadelphia (2001)
Ren, S., He, K., Girshick, R., Sun, J.: Faster r-cnn: Towards real-time object detection with region proposal networks. In: Advances in Neural Information Processing Systems, pp. 91–99 (2015)
Sandler, M., Howard, A., Zhu, M., Zhmoginov, A., Chen, LC.: Mobilenetv2: inverted residuals and linear bottlenecks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 4510–4520 (2018)
Shi, H., Wang, H., Jin, Y., Zhao, L., Liu, C.: Automated heartbeat classification based on convolutional neural network with multiple kernel sizes. In: 2019 IEEE Fifth International Conference on Big Data Computing Service and Applications (BigDataService). IEEE, pp. 311–315 (2019)
Sun, L., Wang, Y., He, J., Li, H., Peng, D., Wang, Y.: A stacked lstm for atrial fibrillation prediction based on multivariate ecgs. Health Inf. Sci. Syst. 8, 1–7 (2020)
Tereshchenko, L.G., Josephson, M.E.: Frequency content and characteristics of ventricular conduction. J. Electrocardiol. 48(6), 933–937 (2015)
Warrick, P., Homsi, MN.: Cardiac arrhythmia detection from ECG combining convolutional and long short-term memory networks. In: 2017 Computing in Cardiology (CinC). IEEE, pp. 1–4 (2017)
Yan, G., Liang, S., Zhang, Y., Liu, F.: Fusing transformer model with temporal features for ecg heartbeat classification. In: 2019 IEEE International Conference on Bioinformatics and Biomedicine (BIBM). IEEE, pp. 898–905 (2019)
Ye, C., Coimbra, MT., Kumar, BV.: Arrhythmia detection and classification using morphological and dynamic features of ecg signals. In: 2010 Annual International Conference of the IEEE Engineering in Medicine and Biology. IEEE, pp. 1918–1921 (2010)
Yildirim, Ö.: A novel wavelet sequence based on deep bidirectional lstm network model for ECG signal classification. Comput. Biol. Med. 96, 189–202 (2018)
Yu, R., Gao, Y., Duan, X., Zhu, T., Wang, Z., Jiao, B.: Qrs detection and measurement method of ECG paper based on convolutional neural networks. In: 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC). IEEE, pp. 4636–4639 (2018)
Zhang, Z., Dong, J., Luo, X., Choi, K.S., Wu, X.: Heartbeat classification using disease-specific feature selection. Comput. Biol. Med. 46, 79–89 (2014)
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
On behalf of all authors, the corresponding author states that there is no conflict of interest.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
This is an extended version of our PAKDD 2020 paper Simultaneous ECG Heartbeat Segmentation and Classification with Feature Fusion and Long Term Context Dependencies. This work is funded by NSFC grant 61672161 and Dongguan Innovative Research Team Program (No. 2018607201008). We sincerely thank Prof Chun Liang and Dr Zhiqing He from Department of Cardiology, Shanghai Changzheng Hospital for their valuable advice.
Rights and permissions
About this article
Cite this article
Qiu, X., Liang, S., Meng, L. et al. Exploiting feature fusion and long-term context dependencies for simultaneous ECG heartbeat segmentation and classification. Int J Data Sci Anal 11, 181–193 (2021). https://doi.org/10.1007/s41060-020-00239-9
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s41060-020-00239-9