Abstract
The electrocardiogram (ECG) plays an important role in assisting clinical diagnosis such as arrhythmia detection. However, traditional techniques for ECG analysis are time-consuming and laborious. Recently, deep neural networks have become a popular technique for automatically tracking ECG signals, which has demonstrated that they are more competitive than human experts. However, the minority class of life-threatening arrhythmias causes the model training to skew towards the majority class. To address the problem, we propose a dual-level collaborative neural network (DCNN), which includes data-level and cost-sensitive level modules. In the Data Level module, we utilize the generative adversarial network with Unet as the generator to synthesize ECG signals. Next, the Cost-sensitive Level module employs focal loss to increase the cost of incorrect prediction of the minority class. Empirical results show that the Data Level module generates highly accurate ECG signals with fewer parameters. Furthermore, DCNN has been shown to significantly improve the classification of the ECG.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Tsao, C.W., et al.: Heart disease and stroke statistics-2023 update: a report from the American heart association. Circulation 147(8), e93–e621 (2023)
Essa, E., Xie, X.: Multi-model deep learning ensemble for ECG heartbeat arrhythmia classification. In 2020 28th European Signal Processing Conference (EUSIPCO), pp. 1085–1089. IEEE (2021)
Li, Y., Pang, Y., Wang, J., Li, X.: Patient-specific ECG classification by deeper CNN from generic to dedicated. Neurocomputing 314, 336–346 (2018)
Singh, S., Pandey, S.K., Pawar, U., Janghel, R.R.: Classification of ECG arrhythmia using recurrent neural networks. Procedia Comput. Sci. 132, 1290–1297 (2018)
Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-24574-4_28
Wei, L., Hou, H., Chu, J.: Feature fusion for imbalanced ECG data analysis. Biomed. Signal Process. Control 41, 152–160 (2018)
Japkowicz, N., Stephen, S.: The class imbalance problem: a systematic study. Intell. Data Anal. 6(5), 429–449 (2002)
Kandala NVPS Rajesh and Ravindra Dhuli: Classification of imbalanced ECG beats using re-sampling techniques and AdaBoost ensemble classifier. Biomed. Signal Process. Control 41, 242–254 (2018)
Zhou, Z.-H., Liu, X.-Y.: Training cost-sensitive neural networks with methods addressing the class imbalance problem. IEEE Trans. Knowl. Data Eng. 18(1), 63–77 (2005)
Goodfellow, I., et al.: Generative adversarial networks. Commun. ACM 63(11), 139–144 (2020)
Lin, T.-Y., Goyal, P., Girshick, R., He, K., Dollár, P.: Focal loss for dense object detection. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 2980–2988 (2017)
Petmezas, G., et al.: Automated atrial fibrillation detection using a hybrid CNN-LSTM network on imbalanced ECG datasets. Biomed. Signal Process. Control 63, 102194 (2021)
Saroj Kumar Pandey and Rekh Ram Janghel: Automatic detection of arrhythmia from imbalanced ECG database using CNN model with smote. Australas. Phys. Eng. Sci. Med. 42(4), 1129–1139 (2019)
Gao, J., Zhang, H., Lu, P., Wang, Z., et al.: An effective lstm recurrent network to detect arrhythmia on imbalanced ecg dataset. J. Healthc. Eng. 2019 (2019)
Taissir Fekih Romdhane and Mohamed Atri Pr: Electrocardiogram heartbeat classification based on a deep convolutional neural network and focal loss. Comput. Biol. Med. 123, 103866 (2020)
Lima, J.L.P., Macêdo, D., Zanchettin, C.: Heartbeat anomaly detection using adversarial oversampling. In: 2019 International Joint Conference on Neural Networks (IJCNN), pp. 1–7. IEEE (2019)
Xia, Y., Yangyang, X., Chen, P., Zhang, J., Zhang, Y.: Generative adversarial network with transformer generator for boosting ECG classification. Biomed. Signal Process. Control 80, 104276 (2023)
Golany, T., Lavee, G., Yarden, S.T., Radinsky, K.: Improving ECG classification using generative adversarial networks. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, pp. 13280–13285 (2020)
Golany, T., Radinsky, K., Freedman, D.: SimGANs: simulator-based generative adversarial networks for ECG synthesis to improve deep ECG classification. In: International Conference on Machine Learning, pp. 3597–3606. PMLR (2020)
Sarkar, P., Etemad, A.: CardioGAN: attentive generative adversarial network with dual discriminators for synthesis of ECG from PPG. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 35, pp. 488–496 (2021)
Chen, J., Liao, K., Wei, K., Ying, H., Chen, D.Z., Wu, J.: ME-GAN: Learning panoptic electrocardio representations for multi-view ECG synthesis conditioned on heart diseases. In: International Conference on Machine Learning, pp. 3360–3370. PMLR (2022)
Li, W., Tang, Y.M., Yu, K.M., To, S.: SLC-GAN: an automated myocardial infarction detection model based on generative adversarial networks and convolutional neural networks with single-lead electrocardiogram synthesis. Inf. Sci. 589, 738–750 (2022)
Krizhevsky, A., Sutskever, I., Hinton, G.E.: ImageNet classification with deep convolutional neural networks. In: Advances in Neural Information Processing Systems, 25 (2012)
Simonyan, K., Zisserman, A.: Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556 (2014)
Moody, G.B., Mark, R.G.: The impact of the MIT-BIH arrhythmia database. IEEE Eng. Med. Biol. Mag. 20(3), 45–50 (2001)
Tihonenko, V., Khaustov, A., Ivanov, S., Rivin, A., Yakushenko, E.: St petersburg incart 12-lead arrhythmia database. PhysioBank PhysioToolkit and PhysioNet (2008)
Acknowledgement
This work was supported in part by the National Key Research and Development Program of China (2021YFF1201300); in part by the Changsha Municipal Natural Science Foundation (kq2202106); and in part by the National Natural Science Foundation of China (62102456).
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
An, Y., Xiong, A., Guo, L. (2023). DCNN: Dual-Level Collaborative Neural Network for Imbalanced Heart Anomaly Detection. In: Guo, X., Mangul, S., Patterson, M., Zelikovsky, A. (eds) Bioinformatics Research and Applications. ISBRA 2023. Lecture Notes in Computer Science(), vol 14248. Springer, Singapore. https://doi.org/10.1007/978-981-99-7074-2_31
Download citation
DOI: https://doi.org/10.1007/978-981-99-7074-2_31
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-99-7073-5
Online ISBN: 978-981-99-7074-2
eBook Packages: Computer ScienceComputer Science (R0)