Deep learning-based bacterial foraging optimization algorithm to improve digital mammography-based breast cancer detection.
Authors
Affiliations (4)
Affiliations (4)
- Paavai Engineering College, Pachal, Namakkal, 637018, India. [email protected].
- School of Computer Science Engineering and Information Systems, Vellore Institute of Technology, Vellore, 632014, India. [email protected].
- School of Computer Science Engineering and Information Systems, Vellore Institute of Technology, Vellore, 632014, India. [email protected].
- School of Computer Science Engineering and Information Systems, Victorian Institute of Technology, Melbourne, VIC, 3000, Australia. [email protected].
Abstract
This study focuses on improving the detection of breast cancer at an early stage. The common approach for diagnosing breast cancer is mammography, but it is quite tedious as it is subject to subjective analysis. To address these challenges, the research will explore how the mammogram analysis employs deep learning-based techniques to enhance the screening process. Various computer vision models, including Visual Geometry Group (VGG) 19, Inception V3, and custom 20 Convolutional Neural Network (CNN) architecture, are investigated using the Digital Database for Screening Mammography (DDSM) mammogram dataset. The research community widely uses the DDSM for mammographic image analysis. In the domain of CNNs, the models have demonstrated considerable promise due to their efficacy in various tasks, such as image recognition and classification. It is also seen that the CNN model performance is enhanced through hyperparameter optimization. However, manually tuning hyperparameters is laborious and time-consuming. To overcome this challenge, automatic hyperparameter optimization of CNNs uses population-based metaheuristic approaches. This automation mitigates the time required for finding optimal hyperparameters and boosts the CNN model's efficacy. The proposed approach involves using the Bacterial Foraging Optimization (BFO) algorithm to optimize CNN to enhance breast cancer detection. BFO optimizes hyperparameters such as filter size, number of filters, and hidden layers in the CNN model. The experiments show that the proposed BFO-CNN method outperforms other state-of-the-art methods in terms of accuracy, showing improvements of 7.62% for VGG 19, 9.16% for InceptionV3, and 1.78% for the custom CNN-20 layer model. In conclusion, this work aims to leverage deep learning techniques and automatic hyperparameter optimization to enhance breast cancer detection through mammogram analysis. The BFO-CNN model has much potential to improve breast cancer diagnosis accuracy compared to conventional CNN architecture.