Generic hybrid model for breast cancer mammography image classification using EfficientNetB2

  • Abstract
  • Literature Map
  • Similar Papers
Abstract
Translate article icon Translate Article Star icon

Breast cancer is a global health issue that necessitates precise classification for early detection and effective treatment. In recent years, pre-trained models have shown great potential in the field of medical image classification, including breast cancer classification. These models have been trained on extensive datasets, and they possess the ability to capture intricate features and patterns within medical images, facilitating accurate classification. However, some of the models are non-generic. They can be sensitive to dataset biases, leading to over fitting on specific patterns present in the training data, and they equally struggle to handle data from different distributions. In this work, we proposed a generic hybrid model for image classification. The features were extracted from two datasets: the mammographic image analysis society (MIAS) and the INbreast dataset, respectively, through the pre trained EfficientNetB2 architecture. However, three classifiers were used in the image classification of the extracted features: MGSVM, CUBIC SVM, and XGBOOST. Eight evaluation metrics were selected to assess the performance of the proposed models. These metrics include accuracy, precision, F1-score, AUC, sensitivity, false negative rate (FNR), Kappa score, and time complexity. Experimental results show that the hybrid of EfficientNetB2 and the MGSVM classifier is more generic and efficient for breast cancer diagnosis and classification. It exhibits a strong performance when classifying mammography breast images from both datasets, achieving impressive metrics such as an overall accuracy of 99.47%, a sensitivity rate of 99.31%, precision of 99.44%, F1-score of 99.44%, AUC of 99.44%, a low FNR (False Negative Rate) of 0.007, a kappa score of 0.98, and a manageable time complexity of 231.44 seconds on the MIAS Dataset.

Similar Papers
  • Research Article
  • Cite Count Icon 57
  • 10.1371/journal.pone.0304868
An optimized model based on adaptive convolutional neural network and grey wolf algorithm for breast cancer diagnosis.
  • Aug 19, 2024
  • PloS one
  • Khaled Alnowaiser + 3 more

Medical image classification (IC) is a method for categorizing images according to the appropriate pathological stage. It is a crucial stage in computer-aided diagnosis (CAD) systems, which were created to help radiologists with reading and analyzing medical images as well as with the early detection of tumors and other disorders. The use of convolutional neural network (CNN) models in the medical industry has recently increased, and they achieve great results at IC, particularly in terms of high performance and robustness. The proposed method uses pre-trained models such as Dense Convolutional Network (DenseNet)-121 and Visual Geometry Group (VGG)-16 as feature extractor networks, bidirectional long short-term memory (BiLSTM) layers for temporal feature extraction, and the Support Vector Machine (SVM) and Random Forest (RF) algorithms to perform classification. For improved performance, the selected pre-trained CNN hyperparameters have been optimized using a modified grey wolf optimization method. The experimental analysis for the presented model on the Mammographic Image Analysis Society (MIAS) dataset shows that the VGG16 model is powerful for BC classification with overall accuracy, sensitivity, specificity, precision, and area under the ROC curve (AUC) of 99.86%, 99.9%, 99.7%, 97.1%, and 1.0, respectively, on the MIAS dataset and 99.4%, 99.03%, 99.2%, 97.4%, and 1.0, respectively, on the INbreast dataset.

  • PDF Download Icon
  • Research Article
  • Cite Count Icon 156
  • 10.1007/s00521-022-07445-5
An optimized deep learning architecture for breast cancer diagnosis based on improved marine predators algorithm
  • Jun 8, 2022
  • Neural Computing & Applications
  • Essam H Houssein + 2 more

Breast cancer is the second leading cause of death in women; therefore, effective early detection of this cancer can reduce its mortality rate. Breast cancer detection and classification in the early phases of development may allow for optimal therapy. Convolutional neural networks (CNNs) have enhanced tumor detection and classification efficiency in medical imaging compared to traditional approaches. This paper proposes a novel classification model for breast cancer diagnosis based on a hybridized CNN and an improved optimization algorithm, along with transfer learning, to help radiologists detect abnormalities efficiently. The marine predators algorithm (MPA) is the optimization algorithm we used, and we improve it using the opposition-based learning strategy to cope with the implied weaknesses of the original MPA. The improved marine predators algorithm (IMPA) is used to find the best values for the hyperparameters of the CNN architecture. The proposed method uses a pretrained CNN model called ResNet50 (residual network). This model is hybridized with the IMPA algorithm, resulting in an architecture called IMPA-ResNet50. Our evaluation is performed on two mammographic datasets, the mammographic image analysis society (MIAS) and curated breast imaging subset of DDSM (CBIS-DDSM) datasets. The proposed model was compared with other state-of-the-art approaches. The obtained results showed that the proposed model outperforms the compared state-of-the-art approaches, which are beneficial to classification performance, achieving 98.32% accuracy, 98.56% sensitivity, and 98.68% specificity on the CBIS-DDSM dataset and 98.88% accuracy, 97.61% sensitivity, and 98.40% specificity on the MIAS dataset. To evaluate the performance of IMPA in finding the optimal values for the hyperparameters of ResNet50 architecture, it compared to four other optimization algorithms including gravitational search algorithm (GSA), Harris hawks optimization (HHO), whale optimization algorithm (WOA), and the original MPA algorithm. The counterparts algorithms are also hybrid with the ResNet50 architecture produce models named GSA-ResNet50, HHO-ResNet50, WOA-ResNet50, and MPA-ResNet50, respectively. The results indicated that the proposed IMPA-ResNet50 is achieved a better performance than other counterparts.

  • Research Article
  • Cite Count Icon 121
  • 10.1016/j.irbm.2020.12.004
Automatic Detection and Classification of Mammograms Using Improved Extreme Learning Machine with Deep Learning
  • Jan 8, 2021
  • IRBM
  • S.R Sannasi Chakravarthy + 1 more

Automatic Detection and Classification of Mammograms Using Improved Extreme Learning Machine with Deep Learning

  • Research Article
  • Cite Count Icon 3
  • 10.1002/ima.22886
Classifications of benign and malignant mammogram images using Gabor‐modified CNN architecture
  • Apr 6, 2023
  • International Journal of Imaging Systems and Technology
  • V Swetha + 1 more

Breast cancer is one of the leading life killing cancers in women patients around the world. Digital mammogram is used to detect and segment the abnormal mass portions in breast region. In this article, the breast cancer regions are detected and segmented using the Gabor transform based Convolutional Neural Networks (CNN) classification approach. This proposed breast cancer detection method consist of the following modules Kirsch's edge detector, Gabor transform, CNN classification and Segmentation. The cancer pixels and healthy pixels are differentiated in edge pixels. Hence, the Kirsch's edge detector is used to detect the fine edge boundary pixelsthe in source mammogram image. Then, the spatial edge detected mammogram image is converted into multi resolution mammogram image using Gabor transform. This transformed multi resolution mammogram image is classified into Normal, Benign or Malignant in this article using the proposed CNN architecture. The developed CNN structure extracts more complex feature maps from their internal layers to improve the classification rate. The conventional and proposed CNN structure is differed with respect to the internal layers. The proposed CNN structure is optimized by reducing the usage of the internals and increases Convolutional Filters (CF) in each Convolutional Layer during utilization design process. After classification process over, the cancer regions in the abnormal mammogram images are segmented using morphological functions. The proposed breast cancer detection is evaluated on Mammographic Image Analysis Society (MIAS) dataset and their cancer region segmentation results are 98.4% of sensitivity, 98.9% of specificity and 99.2% of cancer region segmentation accuracy. The average Detection Rate (DR) of the proposed breast cancer detection method is about 98.3% on the set of mammogramthe images from MIAS dataset. These simulation results are compared with other state‐of‐the art methods and cross verified by the k‐fold verification algorithm.

  • Research Article
  • Cite Count Icon 185
  • 10.1016/j.compag.2021.106081
Performance of deep learning models for classifying and detecting common weeds in corn and soybean production systems
  • Mar 13, 2021
  • Computers and Electronics in Agriculture
  • Aanis Ahmad + 4 more

Performance of deep learning models for classifying and detecting common weeds in corn and soybean production systems

  • Research Article
  • Cite Count Icon 3
  • 10.21037/qims-2024-2911
Analyzing explainability of YOLO-based breast cancer detection using heat map visualizations.
  • Jul 1, 2025
  • Quantitative imaging in medicine and surgery
  • Awika Ariyametkul + 1 more

Breast cancer is the most frequently diagnosed and leading cause of cancer-related mortality among women worldwide. The danger of this disease is due to its asymptomatic nature in the early stages, thereby underscoring the importance of early detection. Mammography, a specialized X-ray imaging technique for breast examination, has been pivotal in facilitating early detection and reducing mortality rates. In recent years, artificial intelligence (AI) has gained substantial popularity across various fields, including medicine. Numerous studies have leveraged AI techniques, particularly convolutional neural networks (CNNs) and You Only Look Once (YOLO)-based models, for medical image detection and classification. However, the predictions of such AI models often lack transparency and explainability, resulting in low trustworthiness. This study aims to address this gap by investigating three state-of-the-art versions of the YOLO algorithm-YOLO version 9 (YOLOv9), YOLO version 10 (YOLOv10), and YOLO version 11 (YOLO11)-trained on breast cancer imaging datasets, specifically the INbreast and Mammographic Image Analysis Society (MIAS) databases. Additionally, to address the challenges posed by the lack of explainability and transparency, we integrate seven explainable artificial intelligence (XAI) methods: Grad-CAM, Grad-CAM++, Eigen-CAM, EigenGrad-CAM, XGrad-CAM, LayerCAM, and HiResCAM. This study utilized two publicly available breast cancer image databases: INbreast: toward a Full-field Digital Mammographic Database and the MIAS dataset. Preprocessing steps were applied to standardize all images in accordance with the input requirements of the YOLO architecture, as these datasets were used to train the three most recent versions of YOLO. The YOLO model demonstrating the highest performance-measured by mean average precision (mAP), precision, and recall-was selected for integration with seven different XAI methods. The performance of each XAI technique was evaluated both qualitatively through visual inspection and quantitatively using several metrics, including matching ground truth (mGT), Pearson correlation coefficient (PCC), precision, recall, and root mean square error (RMSE). These methodologies were employed to interpret and visualize the "black box" decision-making processes of the top-performing YOLO model. Based on our experimental findings, YOLO11 outperformed YOLOv9 (mAP 0.868) and YOLOv10 (mAP 0.926), achieving the highest mAP of 0.935, with classification accuracies of 95% for benign and 80% for malignant cases. Among the evaluated XAI techniques, HiResCAM provided the most effective visual explanations, attaining the highest mGT score of 0.49, surpassing EigenGrad-CAM (0.45) and LayerCAM (0.42) in both visual and quantitative evaluations. The integration of YOLO11 with HiResCAM offers a robust solution that combines high detection accuracy with improved model interpretability. This approach not only enhances user trustworthiness by revealing decision-making patterns and limitations but also provide insights into the weaknesses of the model, enabling developers to refine and improve AI performance further.

  • Book Chapter
  • Cite Count Icon 1
  • 10.1007/978-981-19-7184-6_25
Research on Image Processing Method and Image Classification Model Based on Artificial Intelligence
  • Jan 1, 2023
  • Jingxin Hu

In the development of social economy and scientific and technological innovation, the image processing mode and classification model chosen by network technology platform is becoming more and more changeable, but in essence, it is necessary to obtain characteristic information in effective image recognition and choose high-quality network algorithm and processing technology to complete image processing and image classification. Therefore, on the basis of understanding the current research trend of computer image processing and image classification model methods, this paper conducts in-depth discussion on the image processing methods and image classification model training design with artificial intelligence as the core and takes the image classification model of transfer learning as an example for practical exploration. The final results show that the image processing method and image classification model based on artificial intelligence have strong performance advantages in practical application.KeywordsArtificial intelligenceImage processingImage classificationThe migration study

  • Research Article
  • Cite Count Icon 5
  • 10.14445/23488379/ijeee-v10i5p110
English
  • May 30, 2023
  • International Journal of Electrical and Electronics Engineering
  • Varsha Nemade + 2 more

Breast cancer is among the top causes of fatalities related to cancer in females. Radiologists commonly use mammogram images to detect breast tumors in their early stages. However, mammography can produce low-contrast images, making it difficult and time-consuming to segment abnormal regions. Deep convolutional neural networks (CNNs) are commonly used for image evaluations. This study used deep CNN models to develop a computer-aided diagnostic (CAD) system for feature extraction and classification. The proposed approach consists of three phases. In the first phase, a shallow, deep CNN model comprising five convolutional layers, five max-pooling layers, one batch normalization layer, and one dropout layer was developed and used to extract recombined images and novel features. In the second phase, the Inception-v3 model was used for label smoothing and classification due to its multiple filters with different sizes. In the third phase, features were extracted using shallow, deep CNN and Inception-v3 models. The Infallible Euclidean distance-based nonlinear dimensionality reduction approach was used to minimize dimensionality. Finally, the Gini-index-based C4.5 decision tree was used for the binary classification of mammogram images from the Digital Database for Screening Mammography (DDSM) + Curated Breast Imaging Subset of DDSM (CBIS-DDSM) and Mammographic Image Analysis Society (MIAS) datasets. The proposed hybrid shallow, deep CNN and Inception-v3 model achieved 99.52% accuracy, a 96% AUC on the DDSM + CBISDDSM dataset, and an accuracy of 97.53% and an AUC value of 97% on the MIAS dataset. Compared with other cutting-edge CAD systems, the proposed hybrid approach achieved higher accuracy by combining in-depth features across both datasets.

  • Research Article
  • Cite Count Icon 8
  • 10.22219/kinetik.v7i3.1462
Mass Classification of Breast Cancer Using CNN and Faster R-CNN Model Comparison
  • Aug 30, 2022
  • Kinetik: Game Technology, Information System, Computer Network, Computing, Electronics, and Control
  • Sunardi Sunardi + 2 more

Threat of breast cancer is a frightening type and threatens the female population worldwide. Early detection is preventive solution to determine cancer diagnosis or tumors in the female breast area. Today, machine learning technology in managing medical images has become an innovative trend in the health sector. This technology can accelerate diagnosing disease based on the acquisition of accuracy values. The primary purpose of this research is to innovate by comparing two deep learning models to build a prediction system for early-stage breast cancer. This research utilizes Convolutional Neural Network (CNN) sequential models and Faster Region-based Convolutional Neural Network (R-CNN) models that can determine the classification of normal or abnormal breast image data, which can determine the normal or abnormal classification of breast image. The dataset's source in this study came from the Mammographic Image Analysis Society (MIAS). This dataset consists of 322 mammogram data with 123 abnormal and 199 normal classes. The experimental results of this study show that the accuracy of the CNN and R-CNN models in image classification are 91.26% and 63.89%, respectively. Based on these results, the CNN sequential model has better accuracy than the Faster R-CNN model, because it does not require unique characteristics to detect breast cancer.

  • Research Article
  • Cite Count Icon 15
  • 10.3322/caac.21643
Multidisciplinary considerations in the treatment of triple-negative breast cancer.
  • Sep 28, 2020
  • CA: A Cancer Journal for Clinicians
  • Jennifer R Bellon + 4 more

Multidisciplinary considerations in the treatment of triple-negative breast cancer.

  • PDF Download Icon
  • Research Article
  • Cite Count Icon 391
  • 10.1109/access.2021.3079204
A Novel Deep-Learning Model for Automatic Detection and Classification of Breast Cancer Using the Transfer-Learning Technique
  • Jan 1, 2021
  • IEEE Access
  • Abeer Saber + 4 more

Breast cancer (BC) is one of the primary causes of cancer death among women. Early detection of BC allows patients to receive appropriate treatment, thus increasing the possibility of survival. In this work, a new deep-learning (DL) model based on the transfer-learning (TL) technique is developed to efficiently assist in the automatic detection and diagnosis of the BC suspected area based on two techniques namely 80-20 and cross-validation. DL architectures are modeled to be problem-specific. TL uses the knowledge gained during solving one problem in another relevant problem. In the proposed model, the features are extracted from the mammographic image analysis- society (MIAS) dataset using a pre-trained convolutional neural network (CNN) architecture such as Inception V3, ResNet50, Visual Geometry Group networks (VGG)-19, VGG-16, and Inception-V2 ResNet. Six evaluation metrics for evaluating the performance of the proposed model in terms of accuracy, sensitivity, specificity, precision, F-score, and area under the ROC curve (AUC) has been chosen. Experimental results show that the TL of the VGG16 model is powerful for BC diagnosis by classifying the mammogram breast images with overall accuracy, sensitivity, specificity, precision, F-score, and AUC of 98.96%, 97.83%, 99.13%, 97.35%, 97.66%, and 0.995, respectively for 80-20 method and 98.87%, 97.27%, 98.2%, 98.84%, 98.04%, and 0.993 for 10-fold cross-validation method.

  • Research Article
  • Cite Count Icon 236
  • 10.1016/j.aej.2021.03.048
Deep learning in mammography images segmentation and classification: Automated CNN approach
  • Apr 5, 2021
  • Alexandria Engineering Journal
  • Wessam M Salama + 1 more

Deep learning in mammography images segmentation and classification: Automated CNN approach

  • Conference Article
  • Cite Count Icon 2
  • 10.1109/aidas47888.2019.8970757
Deleterious Effects of Uncertainty in Color Imagery Streams on Classification Models
  • Sep 1, 2019
  • Syed Muslim Jameel + 4 more

Image Classification (IC) is most prominent among other Artificial Intelligence (AI) domains. Mainly, IC participates rigorously for the development of society in a variety of application areas such as finance, marketing, health, industrial automation, education, and safety and security. Typically, an IC model takes image input data and tunes itself as per the required application task and classify accordingly. Among the various categories of images, color image category is better due to the capability of capturing more details, which are essential for classification purpose. However, the modern world demands Realtime or online image classification, which involves Imagery Streams. The highly likely uncertainty in Imagery Streams is due to non-stationary environment, for example, certain features or class boundaries which are valid at one-time step are not adequate for another time step. These uncertainties in Imagery Streams have deleterious effects on IC models, which causes performance degradation in terms of accuracy or make IC models, not in further use. Therefore, to overcome these issues, IC models need to adapt to changes caused by uncertainties in Imagery Streams. This paper focuses on the understanding the possible scenarios of such uncertainties in Color Imagery Streams, investigates the deleterious effects due to changes in Color Imagery Streams and provides the possible mitigation approach to overcome the issues in IC models. The contribution of this research is the first step towards an adaptive model development to mitigate the deleterious effects of uncertainty in Color Imagery Streams. This model will benefit many application areas and will directly contribute to the daily life of a society.

  • Conference Article
  • Cite Count Icon 4
  • 10.1109/iccci56745.2023.10128352
The Comparative Study of CNN models for Breast Histopathological Image Classification
  • Jan 23, 2023
  • R K Chandana Mani + 1 more

In Recent years, breast cancer has been considered as the prevalent type of cancer and second deadliest cancer. Histopathological image analysis is the gold standard for cancer diagnosis. Since CAD systems take less time compared to traditional approaches it is notable that automated cancer classification will be helpful for reliable and accurate diagnosis. Also CAD models assisted with deep learning techniques produce excellent results for classification of breast histopathological images. Convolutional Neural Network (CNN) models have achieved significant progress in medical image classification tasks including breast histology images. This paper makes a brief comparative study that illustrates performance of CNNs such as VGG-16, ResNet-50 and AlexNet for breast histological image classification. The study used BreakHis dataset for its model comparison. Further the paper illustrates application of data augmentation methods for the dataset as it improves the model accuracy. Also we extend our study to transfer learning methods and cross-check the results after using pre-trained models for breast histology image classification.

  • Conference Article
  • Cite Count Icon 5
  • 10.1109/icears53579.2022.9751974
Transfer Learning Based Breast Cancer Detection and Classification using Mammogram Images
  • Mar 16, 2022
  • R Remya + 1 more

Breast cancer can be considered a deadly disease affecting women over the globe. Since severity of breast cancer is high at advanced stages, early detection processes find useful to increase the survival rate. The recently presented medical imaging modalities and deep learning (DL) models pave the way to design effective breast cancer, classification models. In this view, this study develops a new deep transfer learning enabled breast cancer detection and classification (DTL-BCDC) model using mammogram images. The proposed DTL-BDCD technique mainly intends to identify the presence of breast cancer. Primarily, the DTL-BDCD model involves contrast enhancement using CLAHE technique and adaptive weighted segmentation (AWS) technique is used for determining the infected regions. Besides, Densely Connected Networks (DenseNet-169) model is employed for feature extraction and multilayer perceptron (MLP) is utilized for breast cancer classification. The design of DenseNet169 with MLP model for breast cancer detection shows the novelty of the work. In order to demonstrate the enhanced outcomes of the DTL-BDCD model, a wide range of simulations take place on benchmark datasets and the comparison study reported the betterment of the DTL-BDCD model over the recent approaches.

Save Icon
Up Arrow
Open/Close
Notes

Save Important notes in documents

Highlight text to save as a note, or write notes directly

You can also access these Documents in Paperpal, our AI writing tool

Powered by our AI Writing Assistant