UGGNet: Bridging U-Net and VGG for Advanced Breast Cancer Diagnosis
DOI:
https://doi.org/10.4108/eetcasa.4681Keywords:
Breast Cancer, Classification, Deep Learning, Segmentation, Ultrasonic ImageAbstract
In the field of medical imaging, breast ultrasound has emerged as a crucial diagnostic tool for early detection of breast cancer. However, the accuracy of diagnosing the location of the affected area and the extent of the disease depends on the experience of the physician. In this paper, we propose a novel model called UGGNet, combining the power of the U-Net and VGG architectures to enhance the performance of breast ultrasound image analysis. The U-Net component of the model helps accurately segment the lesions, while the VGG component utilizes deep convolutional layers to extract features. The fusion of these two architectures in UGGNet aims to optimize both segmentation and feature representation, providing a comprehensive solution for accurate diagnosis in breast ultrasound images. Experimental results have demonstrated that the UGGNet model achieves a notable accuracy of 78.2\% on the "Breast Ultrasound Images Dataset."
References
Abien Fred M Agarap. On breast cancer detection: an application of machine learning algorithms on the wisconsin diagnostic dataset. In Proceedings of the 2nd international conference on machine learning and soft computing, pages 5–9, 2018. DOI: https://doi.org/10.1145/3184066.3184080
Walid Al-Dhabyani, Mohammed Gomaa, Hussien Khaled, and Fahmy Aly. Deep learning approaches for data augmentation and classification of breast masses using ultrasound images. Int. J. Adv. Comput. Sci. Appl, 10(5):1–11, 2019.
Walid Al-Dhabyani, Mohammed Gomaa, Hussien Khaled, and Aly Fahmy. Dataset of breast ultrasound images. Data in brief, 28:104863, 2020.
Michal Byra, Piotr Jarosik, Aleksandra Szubert, Michael Galperin, Haydee Ojeda-Fournier, Linda Olson, Mary O’Boyle, Christopher Comstock, and Michael Andre. Breast mass segmentation in ultrasound with selective kernel u-net convolutional neural network. Biomedical Signal Processing and Control, 61:102027, 2020.
Arkapravo Chattopadhyay and Mausumi Maitra. Mribased brain tumour image detection using cnn based deep learning method. Neuroscience informatics, 2(4):100060, 2022.
Erqiang Deng, Zhiguang Qin, Dajiang Chen, Zhen Qin, Yi Ding, Ji Geng, and Ning Zhang. Engan: Enhancement generative adversarial network in medical image segmentation. 2022.
Ashutosh Kumar Dubey, Umesh Gupta, and Sonal Jain. Analysis of k-means clustering approach on the breast cancer wisconsin dataset. International journal of computer assisted radiology and surgery, 11:2033–2047, 2016. DOI: https://doi.org/10.1007/s11548-016-1437-9
Oliver Faust, U Rajendra Acharya, Kristen M Meiburger, Filippo Molinari, Joel EW Koh, Chai Hong Yeong, Pailin Kongmebhol, and Kwan Hoong Ng. Comparative assessment of texture features for the identification of cancer in ultrasound images: a review. Biocybernetics and Biomedical Engineering, 38(2):275–296, 2018. DOI: https://doi.org/10.1016/j.bbe.2018.01.001
Behnaz Gheflati and Hassan Rivaz. Vision transformers for classification of breast ultrasound images. In 2022 44th Annual International Conference of the IEEE Engineering in Medicine & Biology Society (EMBC), pages 480–483. IEEE, 2022.
MJ Ghrabat, Zaid Alaa Hussien, Mustafa S Khalefa, Zaid Ameen Abduljabba, Vincent Omollo Nyangaresi, Mustafa A Al Sibahee, and Enas Wahab Abood. Fully automated model on breast cancer classification using deep learning classifiers. Indonesian Journal of Electrical Engineering and Computer Science, 28(1):183–191, 2022.
Angela N Giaquinto, Hyuna Sung, Kimberly D Miller, Joan L Kramer, Lisa A Newman, Adair Minihan, Ahmedin Jemal, and Rebecca L Siegel. Breast cancer statistics, 2022. CA: a cancer journal for clinicians, 72(6):524–541, 2022.
Yujuan Guo, Jingjuan Liao, and Guozhuang Shen. A deep learning model with capsules embedded for highresolution image classification. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 14:214–223, 2020.
Mohamed Hosni, Ibtissam Abnane, Ali Idri, Juan M Carrillo de Gea, and José Luis Fernández Alemán. Reviewing ensemble classification methods in breast cancer. Computer methods and programs in biomedicine, 177:89–112, 2019.
Rizwana Irfan, Abdulwahab Ali Almazroi, Hafiz Tayyab Rauf, Robertas Damaševičius, Emad Abouel Nasr, and Abdelatty E Abdelgawad. Dilated semantic segmentation for breast ultrasonic lesion detection using parallel feature fusion. Diagnostics, 11(7):1212, 2021.
Kiran Jabeen, Muhammad Attique Khan, Majed Alhaisoni, Usman Tariq, Yu-Dong Zhang, Ameer Hamza, Art¯uras Mickus, and Robertas Damaševičius. Breast cancer classification from ultrasound images using probability-based optimal deep learning feature fusion. Sensors, 22(3):807, 2022.
Aishik Konwer, Xuan Xu, Joseph Bae, Chao Chen, and Prateek Prasanna. Temporal context matters: Enhancing single image prediction with disease progression representations. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 18824– 18835, 2022.
Alex Krizhevsky, Ilya Sutskever, and Geoffrey E Hinton. Imagenet classification with deep convolutional neural networks. Advances in neural information processing systems, 25, 2012.
Neeraj Kumar, Ruchika Verma, Ashish Arora, Abhay Kumar, Sanchit Gupta, Amit Sethi, and Peter H Gann. Convolutional neural networks for prostate cancer recurrence prediction. In Medical Imaging 2017: Digital Pathology, volume 10140, pages 106–117. SPIE, 2017. DOI: https://doi.org/10.1117/12.2255774
Jorge F Lazo, Sara Moccia, Emanuele Frontoni, and Elena De Momi. Comparison of different cnns for breast tumor classification from ultrasound images. arXiv preprint arXiv:2012.14517, 2020.
Vivian Man, Wing-Pan Luk, Ling-Hiu Fung, and Ava Kwong. The role of pre-operative axillary ultrasound in assessment of axillary tumor burden in breast cancer patients: a systematic review and meta-analysis. Breast Cancer Research and Treatment, 196(2):245–254, 2022.
O Ibrahim Obaid, Mazin Abed Mohammed, Mohd Kanapi Abd Ghani, A Mostafa, Fahad Taha, et al. Evaluating the performance of machine learning techniques in the classification of wisconsin breast cancer. International Journal of Engineering & Technology, 7(4.36):160–166, 2018.
Olaf Ronneberger, Philipp Fischer, and Thomas Brox. Unet: Convolutional networks for biomedical image segmentation. In Medical Image Computing and Computer- Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany, October 5-9, 2015, Proceedings, Part III 18, pages 234–241. Springer, 2015. DOI: https://doi.org/10.1007/978-3-319-24574-4_28
David E Rumelhart, Geoffrey E Hinton, and Ronald J Williams. Learning representations by back-propagating errors. nature, 323(6088):533–536, 1986. DOI: https://doi.org/10.1038/323533a0
Karen Simonyan and Andrew Zisserman. Very deep convolutional networks for large-scale image recognition. arXiv preprint arXiv:1409.1556, 2014.
Ahmet Solak, Rahime Ceylan, Mustafa Alper Bozkurt, Hakan Cebeci, and Mustafa Koplay. Adrenal tumor segmentation on u-net: A study about effect of different parameters in deep learning. Vietnam Journal of Computer Science, pages 1–25, 2023.
Manu Subramoniam, TR Aparna, PR Anurenjan, and KG Sreeni. Deep learning-based prediction of alzheimer’s disease from magnetic resonance images. In Intelligent vision in healthcare, pages 145–151. Springer, 2022.
Christian Szegedy, Vincent Vanhoucke, Sergey Ioffe, Jon Shlens, and Zbigniew Wojna. Rethinking the inception architecture for computer vision. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 2818–2826, 2016. DOI: https://doi.org/10.1109/CVPR.2016.308
Mesut TOĞAÇAR and Burhan ERGEN. Deep learning approach for classification of breast cancer. In 2018 International Conference on Artificial Intelligence and Data Processing (IDAP), pages 1–5. IEEE, 2018.
Saliha Zahoor, Umar Shoaib, and Ikram Ullah Lali. Breast cancer mammograms classification using deep neural network and entropy-controlled whale optimization algorithm. Diagnostics, 12(2):557, 2022.
Downloads
Published
How to Cite
Issue
Section
License
Copyright (c) 2024 EAI Endorsed Transactions on Context-aware Systems and Applications
This work is licensed under a Creative Commons Attribution 3.0 Unported License.
This is an open-access article distributed under the terms of the Creative Commons Attribution CC BY 3.0 license, which permits unlimited use, distribution, and reproduction in any medium so long as the original work is properly cited.
