Early diagnosis and administration of suitable treatment can substantially enhance the probability of human survival from breast cancer. This study utilized a large dataset comprising thousands of labeled breast images representing various types of breast cancer to train and validate the ResNet-50 model. Important features were extracted from the images using the residual blocks of the network and then fine-tuned for optimal performance. The experiments demonstrated that the ResNet-50 model achieved a fair level of accuracy in differentiating various forms of breast cancer, such as benign, malignant, and others. The ResNet-50 performed reasonably well, identifying benign, malignant, and normal cases with 98% accuracy when using accuracy as the metric, 97% when using precision, and 100% when using recall. Consequently, the trained ResNet-50 model was combined with the Flask framework to generate a simple user interface. These results suggest that employing residual networks in detecting breast cancer can significantly aid in early diagnosis and treatment. This study has important implications for public health and medical practice, providing physicians with a valuable resource in their fight against breast tumors.
Breast cancer prediction using a pretrained CNN Model ResNet-50
Authors
- Akinbowale Nathaniel Babatunde Department of Computer Science, Faculty of Information and Communication Technology, Kwara State University, Malete, Nigeria. https://orcid.org/0000-0001-5168-6071
- Bukola Fatimah Balogun School of Computer Science and Informatics, De Montfort University, Leicester, United Kingdom. https://orcid.org/0009-0009-7069-4208
- Omosola Jacob Olabode Faculty of Computing and Applied Sciences, Department of Mathematical and Computing Science, Thomas Adewumi University, Oko-Irese, Nigeria. https://orcid.org/0000-0002-8622-056X
- Joseph Bamidele Awotunde Department of Computer Science, Faculty of Communication and Information Sciences, University of Ilorin, Ilorin 240003, Nigeria. https://orcid.org/0000-0002-1020-4432
- Agbotiname Lucky Imoize Department of Electrical and Electronics Engineering, Faculty of Engineering, University of Lagos, Akoka, Lagos 100213, Nigeria. https://orcid.org/0000-0001-8921-8353