Breast cancer diagnosis model using stacked autoencoder with particle swarm optimization

Abstract

Breast cancer (BrC) stands as the most prevalent cancer affecting women globally, comprising 24.5% of all female cancer diagnoses and contributing to 15.0% of total cancer-related fatalities. The timely detection and precise categorization of breast cancer play pivotal roles in enhancing patient prognosis and treatment outcomes. The main goal is to enhance the precision of classifying mammogram images, thus offering vital support to radiology experts in diagnosing BrCs. The proposed model encompasses several pivotal stages, including pre-processing, feature extraction, segmentation, and classification. To assess the model's efficacy, we employed the INBreast dataset. During pre-processing, mammogram images were enhanced through a customized contrast-limited adaptive histogram equalization (mCLAHE) technique coupled with data augmentation. Segmentation was executed utilizing the Res-SegNet model, and feature extraction employing the VGG-19 model. The classification was conducted via a stacked autoencoder (SAE) with particle swarm optimization (PSO). Our proposed model exhibited notably high performance compared to alternative models such as CNN, Yolo-v4, and Inception-v3. The results unveiled an accuracy of 98.33%, precision of 99.39%, recall of 98.78%, specificity of 93.75%, an F1-score of 99.08%, and an MCC score of 90.04%.

Keywords

• Breast cancer
• CAD
• INBreast
• Res-SegNet
• VGG-19
• Stacked autoencoder