U-Net Based Deep Learning Approach for 2D Cardiovascular Image Segmentation

International Journal of Electronics and Communication Engineering

© 2024 by SSRG - IJECE Journal

Volume 11 Issue 9

Year of Publication : 2024

Authors : P. Sudheer, K. Indumathy, Pallapati Ravi Kumar, J. Manoranjini, Vijayalaxmi Bindla, Birjis.Fathima

10.14445/23488549/IJECE-V11I9P118

How to Cite?

P. Sudheer, K. Indumathy, Pallapati Ravi Kumar, J. Manoranjini, Vijayalaxmi Bindla, Birjis.Fathima, "U-Net Based Deep Learning Approach for 2D Cardiovascular Image Segmentation," SSRG International Journal of Electronics and Communication Engineering, vol. 11,  no. 9, pp. 207-214, 2024. Crossref, https://doi.org/10.14445/23488549/IJECE-V11I9P118

Abstract:

The goal of medical image segmentation is to organize pixels into several areas according to the several characteristics of the images. Due to several factors, such as variations in data signal-to-noise ratios, signal intensities, and individual variations in heart morphologies, segmenting 2D echo cardiovascular images remains a difficult process. This research introduces 3D U-Net-SparseVoxNet, a unique and effective 3D sparse convolutional network based on U-Net. Any two layers in this network that have the same feature map size may have direct connections with each other, and there are fewer connections overall. Consequently, by drastically reducing the network depth and ultimately utilizing a spatial selfattention mechanism to improve feature representation, 3D U-Net-SparseVoxNet can successfully handle the optimization issue of gradients disappearing when using a limited sample of data to train a 3D deep neural network architecture. This research presents a detailed evaluation of the suggested technique using the HVSMR 2016 dataset. The strategy performs better when compared to other approaches. The proposed method proved to provide good and efficient results in classifying the data with an accuracy of 90% compared to 3D U-Net and VoxResNet, with 74% and 80% accuracy, respectively.

Keywords:

2-D Echocardiography, U-Net, Segmentation, Images, Convolution neural network.

References:

[1] Michaela Schmidt et al., “Novel Highly Accelerated Real-Time CINE-MRI Featuring Compressed Sensing with K-T Regularization in Comparison to TSENSE Segmented and Real-Time Cine Imaging,” Journal of Cardiovascular Magnetic Resonance, vol. 15, pp. 1-2, 2013.
[CrossRef] [Google Scholar] [Publisher Link]
[2] Michael S. Hansen et al., “Retrospective Reconstruction of High Temporal Resolution Cine Images from Real-Time MRI Using Iterative Motion Correction,” Magnetic Resonance in Medicine, vol. 68, no. 3, pp. 741-750, 2012.
[CrossRef] [Google Scholar] [Publisher Link]
[3] Li Feng et al., “5D Whole-Heart Sparse MRI,” Magnetic Resonance in Medicine, vol. 79, no. 2, pp. 826-838, 2018.
[CrossRef] [Google Scholar] [Publisher Link]
[4] Simone Coppo et al., “Free-Running 4D Whole-Heart Self-Navigated Golden Angle MRI: Initial Results,” Magnetic Resonance in Medicine, vol. 74, no. 5, pp. 1306-1316, 2015.
[CrossRef] [Google Scholar] [Publisher Link]
[5] M. Usman et al., “Free Breathing Whole-Heart 3D CINE MRI with Self-Gated Cartesian Trajectory,” Magnetic Resonance in Medicine, vol. 38, pp. 129-137, 2017.
[CrossRef] [Google Scholar] [Publisher Link]
[6] Cameron Hassani et al., “Myocardial Radiomics in Cardiac CMR,” American Journal of Roentgenology, vol. 214, no. 3, pp. 536- 545, 2019.
[CrossRef] [Google Scholar] [Publisher Link]
[7] Yucheng Song et al., “Deep Learning-Based Automatic Segmentation of Images in Cardiac Radiography: A Promising Challenge,” Computer Methods and Programs in Biomedicine, vol. 220, 2022.
[CrossRef] [Google Scholar] [Publisher Link]
[8] Daniel Rueckert, Gabor Fichtinger, and S. Kevin Zhou, Handbook of Medical Image Computing and Computer Assisted Intervention, Elsevier, 2020.
[Google Scholar] [Publisher Link]
[9] Ian Goodfellow, Yoshua Bengio, and Aaron Courville, Deep Learning, MIT Press, pp. 1-800, 2016.
[Google Scholar] [Publisher Link]
[10] Bosung Seo et al., “Cardiac MRI Image Segmentation for Left Ventricle and Right Ventricle Using Deep Learning,” arXiv, pp. 1-27, 2017.
[CrossRef] [Google Scholar] [Publisher Link]
[11] David Chen et al., “Deep Neural Network for Cardiac Magnetic Resonance Image Segmentation,” Journal of Imaging, vol. 8, no. 5, pp. 1-11, 2022.
[CrossRef] [Google Scholar] [Publisher Link]
[12] Kamalika Some, The History, Evolution and Growth of Deep Learning, Analytics Insight, 2018. [Online]. Available: https://www.analyticsinsight.net/the-history-evolution-and-growth-of-deep-learning/
[13] Xiaofeng Ding et al., “Cab U-Net: An End-to-End Category Attention Boosting Algorithm For Segmentation,” Computerized Medical Imaging and Graphics, vol. 84, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[14] S. Jeevakala et al., “Artificial Intelligence in Detection and Segmentation of Internal Auditory Canal and Its Nerves Using Deep Learning Techniques,” International Journal of Computer Assisted Radiology and Surgery, vol. 15, no. 11, pp. 1859-1867, 2020.
[CrossRef] [Google Scholar] [Publisher Link]
[15] Geert Litjens et al., “A Survey on Deep Learning in Medical Image Analysis,” Medical Image Analysis, vol. 42, pp. 60-88, 2017.
[CrossRef] [Google Scholar] [Publisher Link]
[16] Fisher Yu, and Vladlen Koltun, “Multi-Scale Context Aggregation by Dilated Convolutions,” arXiv, pp. 1-13, 2016.
[CrossRef] [Google Scholar] [Publisher Link]
[17] Jelmer M. Wolterink et al., “Dilated Convolutional Neural Networks for Cardiovascular MR Segmentation in Congenital Heart Disease,” Reconstruction, Segmentation, and Analysis of Medical Images: First International Workshops, Athens, Greece, pp. 95- 102, 2017.
[CrossRef] [Google Scholar] [Publisher Link]
[18] Fisher Yu, Vladlen Koltun, and Thomas Funkhouser, “Dilated Residual Network,” Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 472-480, 2017.
[Google Scholar] [Publisher Link]
[19] Hao Chen et al., “VoxResNet: Deep Voxelwise Residual Networks for Brain Segmentation from 3D MR Images,” NeuroImage, vol. 170, pp. 446-455, 2017.
[CrossRef] [Google Scholar] [Publisher Link]
[20] Lequan Yu et al., “Volumetric ConvNets with Mixed Residual Connections for Automated Prostate Segmentation from 3D MR Images,” Proceedings of the AAAI Conference on Artificial Intelligence, vol. 31, no. 1, pp. 66-72, 2017.
[CrossRef] [Google Scholar] [Publisher Link]
[21] Lee R. Dice, “Measures of the Amount of Ecologic Association between Species,” Ecology, vol. 26, no. 3, pp. 297-302, 1945.
[CrossRef] [Google Scholar] [Publisher Link]
[22] Kolawole O. Babalola et al., “Comparison and Evaluation of Segmentation Techniques for Subcortical Structures in Brain MRI,” Medical Image Computing and Computer-Assisted Intervention: 11th International Conference, New York, NY, USA, pp. 409-416, 2008.
[CrossRef] [Google Scholar] [Publisher Link]
[23] Vasily Zyuzin et al., “Identification of the Left Ventricle Endocardial Border on Two-Dimensional Ultrasound Images Using the Convolutional Neural Network Unet,” 2018 Ural Symposium on Biomedical Engineering, Radioelectronics and Information Technology, Yekaterinburg, Russia, pp. 76-78, 2018.
[CrossRef] [Google Scholar] [Publisher Link]
[24] Ozan Oktay et al., “Anatomically Constrained Neural Networks (Acnns): Application to Cardiac Image Enhancement and Segmentation,” IEEE Transactions on Medical Imaging, vol. 37, no. 2, pp. 384-395, 2018.
[CrossRef] [Google Scholar] [Publisher Link]
[25] Özgün Çiçek et al., “3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation,” Medical Image Computing and Computer-Assisted Intervention: 19th International Conference, Athens, Greece, pp. 424–432, 2016.
[CrossRef] [Google Scholar] [Publisher Link]
[26] Olivier Bernard et al., “Standardized Evaluation System for Left Ventricular Segmentation Algorithms in 3D Echocardiography,” IEEE Transactions on Medical Imaging, vol. 35, no. 4, pp. 967-777, 2016.
[CrossRef] [Google Scholar] [Publisher Link]
[27] Jianxu Chen et al., “Combining Fully Convolutional And Recurrent Neural Networks For 3d Biomedical Image Segmentation,” Advances in Neural Information Processing Systems, vol. 29, 2016.
[Google Scholar] [Publisher Link]
[28] M. Ghafoorian et al., “Non-Uniform Patch Sampling with Deep Convolutional Neural Networks for White Matter Hyperintensity Segmentation,” IEEE 13th International Symposium on Biomedical Imaging, Prague, Czech Republic, pp. 1414-1417, 2016.
[CrossRef] [Google Scholar] [Publisher Link]
[29] Rudra P.K. Poudel, Pablo Lamata, and Giovanni Montana, “Recurrent Fully Convolutional Neural Networks for Multi-Slice MRI Cardiac Segmentation,” Reconstruction, Segmentation, and Analysis of Medical Images: First International Workshops, Athens, Greece, pp. 83-94, 2017.
[CrossRef] [Google Scholar] [Publisher Link]
[30] Jonathan Long, Evan Shelhamer, and Trevor Darrell, “Fully Convolutional Networks for Semantic Segmentation,” Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3431-3440, 2015.
[Google Scholar] [Publisher Link]
[31] Tsung-Yi Lin et al., “Feature Pyramid Networks for Object Detection,” Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2117–25, 2017.
[Google Scholar] [Publisher Link]
[32] Sarah Leclerc et al., “Deep Learning for Segmentation using an Open Large-Scale Dataset in 2D Echocardiography,” IEEE Transactions on Medical Imaging, vol. 38, no. 9, pp. 2198-2210, 2019.
[CrossRef] [Google Scholar] [Publisher Link]