CSO-CNN: Cat Swarm Optimization-guided Convolutional Neural Network for Mobile Detection of Breast Cancer

Abstract

Breast cancer has become the most common cancer in the world. Early diagnosis and treatment can greatly improve the survival rate of breast cancer patients. Computer diagnostic technology based on convolutional neural networks (CNNs) can assist in detecting breast cancer based on medical images, effectively improving detection accuracy. Hyperparameters in CNN will affect model performance, so hyperparameter tuning is necessary for model training. However, traditional tuning methods can get stuck in local minimums. Therefore, the weights and biases of artificial neural networks are usually trained using global optimization algorithms. Our research introduces cat swarm optimization (CSO) to construct a cat swarm optimization-guided convolutional neural network (CSO-CNN). The model can quickly obtain the optimal combination of hyperparameters and stably get closer to the global optimal. The statistical results of CSO-CNN obtained a sensitivity of 93.50% ± 2.42%, a specificity of 92.20% ± 3.29%, a precision of 92.35% ± 3.01%, an accuracy of 92.85% ± 2.49%, an F1-score of 92.91% ± 2.44%, Matthews correlation coefficient of 85.74% ± 4.94%, and Fowlkes-Mallows index was 92.92% ± 2.43%. Our CSO-CNN algorithm is superior to five state-of-the-art methods. In addition, we tested the CSO-CNN algorithm on the local computer to simulate the mobile environment and confirmed that the algorithm can be transplanted to the network servers.