Information, Vol. 16, Pages 227: Application of VGG16 Transfer Learning for Breast Cancer Detection
Information doi: 10.3390/info16030227
Authors: Tanjim Fatima Hamdy Soliman
Breast cancer is among the primary causes of cancer-related deaths globally, highlighting the critical need for effective and early diagnostic methods. Traditional diagnostic approaches, while valuable, often face limitations in accuracy and accessibility. Recent advancements in deep learning, particularly transfer learning, provide promising solutions for enhancing diagnostic precision in breast cancer detection. Due to the limited capability of the BreakHis dataset, transfer learning was utilized to advance the training of our new model with the VGG16 neural network model, well trained on the rich ImageNet dataset. Moreover, the VGG16 architecture was carefully modified, including the fine-tuning of its layers, yielding our new model: M-VGG16. The new M-VGG16 model is designed to carry out the binary cancer/benign classification of breast samples effectively. The experimental results of our M-VGG16 model showed it achieved high validation accuracy (93.68%), precision (93.22%), recall (97.91%), and a high AUC (0.9838), outperforming other peer models in the same field. This study validates the VGG16 model’s suitability for breast cancer detection via transfer learning, providing an efficient, adaptable framework for improving diagnostic accuracy and potentially enhancing breast cancer detection. Key breast cancer detection challenges and potential M-VGG16 model refinements are also discussed.