Abstract
Although the outcomes of the DL techniques obtained are promising, performance is always limited to some extent due to a need for more sufficient data. We observe the viability of using DL techniques on a test data set in this study. We introduce an easy-to-use method for performing 2D slice-by-slice segmentation based on Region of Interest (ROI). This method employs an improved training system to enhance segmentation. Here, a novel attention-based U-Net architecture (\(a-{\text{UNet}}\)) is used. Additionally, by examining our methodology on two separate sets of data, the imATFIB set of data and the openly accessible ACDC challenge belonging to our internal medical research, the combination of both the attention and U-Net model shows utterly different architectures and yields consistent advantages. ATFIB dataset outcomes demonstrate that the suggested method works well with the practice extents offered, averaging 89.89% on the validation set, Dice Similarity Coefficient of the entire heart. Additionally, with a mean Dice score of 91.87% on the ACDC validation, our algorithm ranked as the second-best strategy for the challenge. Our method offers the chance to act as a component of a software-aided diagnosis in medicine.
Similar content being viewed by others
Data availability
Enquiries about data availability should be directed to the authors.
References
Akkus Z, Galimzianova A, Hoogi A, Rubin DL, Erickson BJ (2017) Deep learning for brain MRI segmentation: state of the art and future directions. J Digit Imaging 30(4):449–459
Badrinarayanan V, Kendall A, Cipolla R (2017) SegNet: a deep convolutional encoder-decoder architecture for image segmentation. IEEE Trans Pattern Anal Mach Intell 39(12):2481–2495
Bochkovskiy A, Wang CY, Liao HYM (2020) Yolov4: optimal speed and accuracy of object detection. arXiv:200410934
Boutillon A, Borotikar B, Burdin V, Conze PH (2020) Combining shape priors with conditional adversarial networks for improved scapula segmentation in mr images. In: Proceedings of the IEEE international symposium on biomedical imaging (ISBI), pp 1164–1167
Cherukuri V, Ssenyonga P, Warf BC, Kulkarni AV, Monga V, Schiff SJ (2017) Learning based segmentation of ct brain images: application to post-operative hydrocephalic scans. IEEE Trans Bio-Med Eng 65(8):1871–1884
Çiçek O (2016) 3D U-net: learning dense, volumetric segmentation from sparse annotation. In: Proceedings of international conference on medical image computing and computer assisted intervention, pp 424–432
Dhungel N (2015) Deep learning and structured prediction for mass segmentation in mammograms. In: Proceedings of international conference medical image computing computer-assisted intervention (MICCIA), 2015, pp 605–612
Feng X, Wang C, Cheng S, Guo L (2019) Automatic liver and tumour segmentation of CT based on cascaded U-net. In: Proceedings of the Chinese intelligent systems conference, pp. 155–164
Fu H, Cheng J, Xu Y, Wong DWK, Liu J, Cao X (2018) Joint optic disc and cup segmentation based on multi-label deep network and polar transformation. IEEE Trans Med Image 37(7):1597–1605
Gao et al (2020) Joint disc and cup segmentation based on recurrent fully convolutional network. PLoS ONE 15(9):1–23
Gu et al (2019) Segmentation and suppression of pulmonary vessels in low-dose chest CT scans. Med Phys 46(8):13648
He K, Cao X, Shi Y, Nie D, Gao Y, Shen D (2019) Pelvic organ segmentation using distinctive curve guided fully convolutional networks. IEEE Trans Med Imaging 38(2):585–595
Hou Z, Siddiquee MMR, Tajbakhsh N, Liang J (2019) Unet++: redesigning skip connections to exploit multiscale features in image segmentation. IEEE Trans Med Imag 39(6):1856–1867
Kaluva KC, Khened M, Kori A, Krishnamurthi G (2018) 2D-densely connected convolution neural networks for automatic liver and tumour segmentation. arXiv:180202182
Li X, Chen H, Qi X, Dou Q, Fu CW, Heng PA (2018) H-denseunet: hybrid densely connected UNet for liver and tumour segmentation from ct volumes. IEEE Trans Med Imaging 37(12):2663–2674
Men K, Dai J, Li Y (2017) Automatic segmentation of the clinical target volume and organs at risk in the planning CT for rectal cancer using deep dilated convolutional neural networks. Med Phys 44(12):6377–6389
Ren S, He K, Girshick R, Sun J (2016) Faster R-CNN: towards real-time object detection with region proposal networks. IEEE Trans Pattern Anal Mach Intell 39(6):1137–1149
Ronneberger O, Fischer P, Brox T (2015) U-net: convolutional networks for biomedical image segmentation. In: Proceedings of medical image computing computed-assisted interventions (MICCAI), 2015, pp 234–241
Schreibmann DM, Marcus D, Fox T (2014) Multiatlas segmentation of thoracic and abdominal anatomy with level set-based local search. J Appl Clin Med Phys 15(4):22–38
Setio AAA, Ciompi F, Litjens G, Gerke P, Jacobs C, van Riel SJ, Wille MMW, Naqibullah M, Sanchez CI, van Ginneken B (2016) Pulmonary nodule detection in CT images: false positive reduction using multi-view convolutional networks. IEEE Trans Med Imaging 35(5):1160–1169
Setio AAA et al (2017) Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: the LUNA16 challenge. Med Image Anal 42:1–13
Shen D, Wu G, Suk HI (2017) Deep learning in medical image analysis. Ann Rev Biomed Eng 19:221–248
Sherubha (2019) An efficient network threat detection and classification method using ANP-MVPS algorithm in wireless sensor networks. Int J Innov Technol Explor Eng 8(11) (ISSN: 2278-3075)
Sherubha (2023) Graph-based event measurement for analyzing distributed anomalies in sensor networks. Sådhanå 45:212. https://doi.org/10.1007/s12046-020-01451-w
Sherubha (2023) An efficient intrusion detection and authentication mechanism for detecting clone attack in wireless sensor networks. J Adv Res Dyn Control Syst 11(5):55–68
Singh VK, Rashwan HA, Romani S, Akram F, Pandey N, Sarker MMK et al (2020) Breast tumour segmentation and shape classification in mammograms using generative adversarial and convolutional neural network. Expert Syst Appl 139:112855
Wang S, Zhou M, Liu Z, Liu Z, Gu D, Zang Y et al (2017) Central-focused convolutional neural networks: developing a data-driven model for lung nodule segmentation. Med Image Anal 40:172–183
Xing F, Xie Y, Yang L (2016) An automatic learning-based framework for robust nucleus segmentation. IEEE Trans Med Imaging 35(2):550–566
Xu X, Lu Q, Yang L, Hu S, Chen D, Hu Y et al (2018) Quantization of fully convolutional networks for accurate biomedical image segmentation. In: Proceedings of the IEEE conference on computer vision and pattern recognition (CVPR), pp 8300–8308. IEEE, Piscataway, NJ (2018)
Zhou MMR, Siddiquee, Tajbakhsh N, Liang J (2018) Unet++: a nested u-net architecture for medical image segmentation. In: Deep learning in medical image analysis and multimodal learning for clinical decision support, vol 11045. Springer, Cham, pp 3–11
Funding
The authors have not disclosed any funding.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
The authors declare no conflicts of interest.
Research involving human participants and/or animals
This article does not contain any studies with human participants or animals performed by any of the authors.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Subaramani, N., Sasikala, E. An attention-based dense network model for cardiac image segmentation using learning approaches. Soft Comput 28, 765–775 (2024). https://doi.org/10.1007/s00500-023-09482-1
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s00500-023-09482-1