Skip to main content

Advertisement

Log in

An attention-based dense network model for cardiac image segmentation using learning approaches

  • Application of soft computing
  • Published:
Soft Computing Aims and scope Submit manuscript

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.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7

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

    Article  Google Scholar 

  • 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

    Google Scholar 

  • Ç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

    Article  Google Scholar 

  • Gao et al (2020) Joint disc and cup segmentation based on recurrent fully convolutional network. PLoS ONE 15(9):1–23

    Article  Google Scholar 

  • Gu et al (2019) Segmentation and suppression of pulmonary vessels in low-dose chest CT scans. Med Phys 46(8):13648

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

    Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • Shen D, Wu G, Suk HI (2017) Deep learning in medical image analysis. Ann Rev Biomed Eng 19:221–248

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

    Article  Google Scholar 

  • 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

Download references

Funding

The authors have not disclosed any funding.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Nandhagopal Subaramani.

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.

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

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

Download citation

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s00500-023-09482-1

Keywords

Navigation