Abstract
Cardiovascular disease diagnosis and prompt medical care depend critically on the classification of heart sounds. In recent years, deep learning-based approaches have shown promising results in automating the process of heart sound categorization. This paper proposes a model HeartBeatNet (an attention UNet-based system) for heart sound classification that demonstrates comparatively better performance. The proposed system combines the strengths of attention mechanisms and the UNet architecture to effectively capture relevant features and to make accurate predictions. The system is trained on the PhysioNet/Cinc 2016 dataset consisting of annotated heart sound recordings, which are first converted into Mel Spectrograms before feeding into the UNet based network. The results indicate that the proposed system achieves high accuracy of 95.14%, sensitivity of 90.00%, and specificity of around 96.72% to classify various heart sound abnormalities.
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
Khairy, P., Poirier, N., Mercier, L.-A.: Univentricular heart. Circulation 115(6), 800–812 (2007)
Nabel, E.G.: Cardiovascular disease. N. Engl. J. Med. 349(1), 60–72 (2003)
Wang, J., et al.: Detecting cardiovascular disease from mammograms with deep learning. IEEE Trans. Med. Imaging 36(5), 1172–1181 (2017)
Guo, Y., Liu, Y., Georgiou, T., Lew, M.S.: A review of semantic segmentation using deep neural networks. Int. J. Multimedia Inf. Retrieval 7, 87–93 (2018)
Nguyen, M.T., Lin, W.W., Huang, J.H.: Heart sound classification using deep learning techniques based on log-mel spectrogram. Circ. Syst. Sig. Process. 42(1), 344–360 (2023)
Zhou, Z., Rahman Siddiquee, M.M., Tajbakhsh, N., Liang, J.: UNet++: a nested U-Net architecture for medical image segmentation. In: Stoyanov, D., et al. (eds.) DLMIA/ML-CDS -2018. LNCS, vol. 11045, pp. 3–11. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-00889-5_1
Nannan, Yu., He, Yu., Li, H., Ma, N., Chunai, H., Wang, J.: A robust deep learning segmentation method for hematoma volumetric detection in intracerebral hemorrhage. Stroke 53(1), 167–176 (2022)
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
Gharehbaghi, A., Partovi, E., Babic, A.: Parralel recurrent convolutional neural network for abnormal heart sound classification. CARING IS SHARING-EXPLOITING THE VALUE IN DATA FOR HEALTH AND INNOVATION, pp. 526 (2023)
Taneja, K., Arora, V., Verma, K.: Classifying the heart sound signals using textural-based features for an efficient decision support system. Expert Syst. 40(6), e13246 (2023)
Malik, A.E.F., Barin, S., Emin Yüksel, M.: Accurate classification of heart sound signals for cardiovascular disease diagnosis by wavelet analysis and convolutional neural network: preliminary results. In: 2020 28th Signal Processing and Communications Applications Conference (SIU), pp. 1–4. IEEE (2020)
Ren, Z., Cummins, N., Pandit, V., Han, J., Qian, K., Schuller, B.: Learning image-based representations for heart sound classification. In: Proceedings of the 2018 International Conference on Digital Health, pp. 143–147 (2018)
Meintjes, A., Lowe, A., Legget, M.: Fundamental heart sound classification using the continuous wavelet transform and convolutional neural networks. In: 2018 40th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), pp. 409–412. IEEE (2018)
Demir, F., Şengür, A., Bajaj, V., Polat, K.: Towards the classification of heart sounds based on convolutional deep neural network. Health Inf. Sci. Syst. 7, 1–9 (2019)
Zabihi, M., Rad, A.B., Kiranyaz, S., Gabbouj, M., Katsaggelos, A.K.: Heart sound anomaly and quality detection using ensemble of neural networks without segmentation. In: 2016 Computing in Cardiology Conference (CinC), pp. 613–616. IEEE (2016)
Kay, E., Agarwal, A.: DropConnected neural networks trained on time-frequency and inter-beat features for classifying heart sounds. Physiol. Meas. 38(8), 1645 (2017)
Mashhoor, R.Y., Ayatollahi, A.: HeartSiam: A domain invariant model for heart sound classification. arXiv preprint arXiv:2210.16394 (2022)
Potes, C., Parvaneh, S., Rahman, A., Conroy, B.: Ensemble of feature-based and deep learning-based classifiers for detection of abnormal heart sounds. In: 2016 Computing in Cardiology Conference (CinC), pp. 621–624. IEEE (2016)
Huai, X., Kitada, S., Choi, D., Siriaraya, P., Kuwahara, N., Ashihara, T.: Heart sound recognition technology based on convolutional neural network. Inform. Health Soc. Care 46(3), 320–332 (2021)
Rubin, J., Abreu, R., Ganguli, A., Nelaturi, S., Matei, I., Sricharan, K.: Recognizing abnormal heart sounds using deep learning. arXiv preprint arXiv:1707.04642 (2017)
Reyna, M.A., et al.: Heart murmur detection from phonocardiogram recordings: the George B. moody physionet challenge 2022. In: 2022 Computing in Cardiology (CinC), vol. 498, pp. 1–4. IEEE (2022)
Van Dyk, D.A., Meng, X.-L.: The art of data augmentation. J. Comput. Graph. Stat. 10(1), 1–50 (2001)
Moreland, K., Angel, E.: The FFT on a GPU. In: Proceedings of the ACM SIGGRAPH/EUROGRAPHICS Conference on Graphics Hardware, pp. 112–119 (2003)
Sivagami, S., Chitra, P., Kailash, G.S.R., Muralidharan, S.R.: UNet architecture based dental panoramic image segmentation. In: 2020 International Conference on Wireless Communications Signal Processing and Networking (WiSPNET), pp. 187–191. IEEE (2020)
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2024 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Singh, G., Mehta, A., Arora, V. (2024). HeartBeatNet: Unleashing the Power of Attention in Cardiology. In: Muthalagu, R., P S, T., Pawar, P.M., R, E., Prasad, N.R., Fiorentino, M. (eds) Computational Intelligence and Network Systems. CINS 2023. Communications in Computer and Information Science, vol 1978. Springer, Cham. https://doi.org/10.1007/978-3-031-48984-6_2
Download citation
DOI: https://doi.org/10.1007/978-3-031-48984-6_2
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-48983-9
Online ISBN: 978-3-031-48984-6
eBook Packages: Computer ScienceComputer Science (R0)