Breast Tumor Segmentation using Deep Learning by U-Net Network


  • Asieh Parhizkar Tarighat Payame Noor University


Breast Cancer, Deep Learning, Segmentation, U-Net


The methods of image analysis are important for the segmentation and detection of breast tumors, where a reliable diagnosis will be supported by portraying crucial morphological.  This paper presents the segmentation of a breast tumor using a deep convolutional neural network by U-net. The structure includes a contractile route to capture the background and an a-symmetric expansion path that provides precise localization. It trains end-to-end from a small number of images. This segmentation algorithm is applied to breast tumours within a Region of Interest (ROI) with a High-Power Field (HPF) in biopsy and its corresponding ground-truth, where it has been labeled by pathologists with benign and malignant. The suggested segmentation model produces binary masks as realistic as possible.  It supplies an Intersection Over Union (IOU) of 68% and attains an overall accuracy as high as 91%, which shows it performs better than the current works.


A. Reddy Vaka, B. Soni and S. Reddy, “Breast cancer detection by leveraging Machine Learning”, ICT Express 2020, pp.1-5.

Sh. Saxena and M. Gyanchandani, “Machine Learning Methods for Computer-Aided Breast Cancer Diagnosis Using Histopathology: A Narrative Review”, Journal of Medical Imaging and Radiation Sciences xx (2019), pp. 1-12.

P. Wang, Q. Song, Y. Li, Sh. Lv, J. Wang, L. Li, and H. Zhang “Cross-task extreme learning machine for breast cancer image classification with deep convolutional features”, Biomedical Signal Processing and Control J, Vol.57, 2020, pp.1-10.

X. Asri, H.Mousannif, H. Moatassime, Th. Noel, “Using Machine Learning Algorithms for Breast Cancer Risk Prediction and Diagnosis”, Procedia Computer Science , vol. 83, 2016, pp: 1064 – 1069.

O. Ronneberger, Ph. Fischer, and T-B. Lu, “U-Net: Convolutional Networks for Biomedical Image Segmentation”, Springer International Publishing Switzerland, May 2015, pp. 234–241.

Y. Zhang, J-H. Chen, K-T. Chang, V- Y.Park, M-J. Kim, S. Chan, P. Chang, D. Chow, A. Luk, T. Kwong, M-Y. Su, “Automatic Breast and Fibroglandular Tissue Segmentation in Breast MRI Using Deep Learning by a Fully Convolutional Residual Neural Network U-Net”, Academic Radiology J, Vol. 26, 2019, pp.1526-1535.

V- K. Singh , H-A. Rashwana, S. Romania , F. Akram , N. Pandey, Md. M- K. Sarker , A. Saleh , M. Arenas , M. Arquez , D. Puig , J. Torrents-Barrena, “Breast tumor segmentation and shape classification in mammograms using generative adversarial and convolutional neural network”, Expert Systems With Applications, vol. 139 , 2020, pp. 1 – 14.

F. A. Spanhol, L. S. Oliveira, C. Petitjean, & L. Heutte, “A dataset for breast cancer histopathological image classification”, IEEE Trans Biomed Eng, vol.63, no. 7, 2016, pp.1455–1462.

J. A. Cruz & D. S. Wishart, “Applications of machine learning in cancer prediction and prognosis. Cancer Inform 2”, 2006, pp. 59–77.

P. J. Sudharshan, C. Petitjean, F. Spanhol, L. E. Oliveira, L. Heutte, & P. Honeine, “Multiple instances learning for histopathological breast cancer image classification”. Expert Syst Appl 117, 2019, pp. 103–111.

T. Araujo, G. Aresta, & E. Castro, “Classification of breast cancer histology images using convolutional neural networks” PLoS One, vol. 12, no. 6, 2017, pp.1–14.

Y. Li, J. Wu, & Q. Wu, (2019). “Classification of breast cancer histology images using multi-size and discriminative patches based on deep Learning”, IEEE Access, 2019, vol. 7, pp. 21400–21408.

A. Osareh, & B. Shadgar, “Machine learning techniques to diagnose breast cancer”, 5th International Symposium on Health Informatics and Bioinformatics, 2010, pp. 114–120.

K. Kourou, T. P. Exarchos, K. P.Exarchos, M. V.Karamouzis, & D. I. Fotiadis, “Machine learning applications in cancer prognosis and prediction”, Comput Struct Biotechnol J, vol. 13, 2015, pp. 8–17.

S. B. Kotsiantis, (2007), “Supervised machine learning: a review of classification techniques”, Informatica, 2007, vol. 31, no. 3, 2007, pp. 249–268.

I. H. Witten, & E. Frank, “Data mining practical machine learning tools and techniques”, (2nd ed.), Morgan Kaufmann Publishers, 2005.

Sh. Saxena, M. Tech, & M. Gyanchandani, “Machine Learning Methods for Computer-Aided Breast Cancer Diagnosis Using Histopathology: A Narrative Review”, Medical Imaging and Radiation Sciences J, 2019, vol. 51, no. 1, 2019, pp. 182–193.

J. Schmidhuber, “Deep learning in neural networks: An overview. Neural Networks, vol. 61, 2015, pp. 85-117.

J. Long, E. Shelhamer, & T. Darrell, “Fully convolutional networks for semantic segmentation”, In Processing of the IEEE conf on computer vision and pattern recognition (cvpr), 2015, pp. 3413-3440.

V. Badrinarayanan, A. Kendall, & R. Cipolla, “Segment: A deep convolutional encoder-decoder architecture for image segmentation”, IEEE Trans on Pattern Analysis and Machine intelligence, vol. 39, no. 12, 2017, pp. 2481-2495.

N. Dhungel, G. Carneiro, & A.P. Bradley, “Deep learning and structured prediction for the segmentation of mass in mammograms”, In Processing of the international conference on medical image computing and computer-assisted intervention (MICCAI), 2015, pp. 605-612.

N. Dhungel, G. Carneiro, & A.P. Bradley, “A deep learning approach for the analysis of masses in mammograms with minimal user intervention”, Medical Image Analysis, vol. 37, 2017, pp. 114-128.




How to Cite

Parhizkar Tarighat, A. (2021). Breast Tumor Segmentation using Deep Learning by U-Net Network. Journal of Telecommunication, Electronic and Computer Engineering (JTEC), 13(2), 49–54. Retrieved from