Next Article in Journal
Teaching–Learning Optimization Algorithm Based on the Cadre–Mass Relationship with Tutor Mechanism for Solving Complex Optimization Problems
Next Article in Special Issue
YOLOv5-MS: Real-Time Multi-Surveillance Pedestrian Target Detection Model for Smart Cities
Previous Article in Journal
Comparison of Prophylactic Effects between Localized Biomimetic Minocycline and Systematic Amoxicillin on Implants Placed Immediately in Infected Sockets
Previous Article in Special Issue
YOLO-DRS: A Bioinspired Object Detection Algorithm for Remote Sensing Images Incorporating a Multi-Scale Efficient Lightweight Attention Mechanism
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Intelligent Breast Mass Classification Approach Using Archimedes Optimization Algorithm with Deep Learning on Digital Mammograms

Information Technology Department, Faculty of Computing and Information Technology, King Abdulaziz University, Jeddah 21589, Saudi Arabia
Biomimetics 2023, 8(6), 463; https://doi.org/10.3390/biomimetics8060463
Submission received: 7 September 2023 / Revised: 22 September 2023 / Accepted: 27 September 2023 / Published: 1 October 2023
(This article belongs to the Special Issue Bioinspired Artificial Intelligence Applications)

Abstract

:
Breast cancer (BC) has affected many women around the world. To accomplish the classification and detection of BC, several computer-aided diagnosis (CAD) systems have been introduced for the analysis of mammogram images. This is because analysis by the human radiologist is a complex and time-consuming task. Although CAD systems are used to primarily analyze the disease and offer the best therapy, it is still essential to enhance present CAD systems by integrating novel approaches and technologies in order to provide explicit performances. Presently, deep learning (DL) systems are outperforming promising outcomes in the early detection of BC by creating CAD systems executing convolutional neural networks (CNNs). This article presents an Intelligent Breast Mass Classification Approach using the Archimedes Optimization Algorithm with Deep Learning (BMCA-AOADL) technique on Digital Mammograms. The major aim of the BMCA-AOADL technique is to exploit the DL model with a bio-inspired algorithm for breast mass classification. In the BMCA-AOADL approach, median filtering (MF)-based noise removal and U-Net segmentation take place as a pre-processing step. For feature extraction, the BMCA-AOADL technique utilizes the SqueezeNet model with AOA as a hyperparameter tuning approach. To detect and classify the breast mass, the BMCA-AOADL technique applies a deep belief network (DBN) approach. The simulation value of the BMCA-AOADL system has been studied on the MIAS dataset from the Kaggle repository. The experimental values showcase the significant outcomes of the BMCA-AOADL technique compared to other DL algorithms with a maximum accuracy of 96.48%.

1. Introduction

Breast cancer (BC) is a major serious problem which is extremely common worldwide. Early diagnosis and identification perform the main protection from this disease, and these are attained through imaging methods, namely mammography [1]. The diagnostic procedures for BC need medical checkups. Breast mass classification refers to the process of categorizing breast masses or tumors into different classes or categories based on certain characteristics or features. This classification is a critical step in breast cancer diagnosis and plays a crucial role in determining the appropriate treatment and management strategies. Computer-aided diagnosis (CAD) methods depend upon medical screening which can be suitable and which can minimize the specialist’s efforts. CAD analysis plays a considerable part in the diagnosis of breast cancer and can be effectively employed in biomedical science [2]. The widespread CAD-based medical checkups are Magnetic Resonance Imaging (MRI), ultrasound, and mammogram. Mammograms can be a highly efficient, lower-dose, reliable, and less invasive screening method for the early identification of BC [3]. The use of automated CAD methods with mammograms increases the accuracy rate of identification; the operational expedient accelerates the diagnosing process and retains the medical appliances. Additionally, the breast mass shows unique symptoms in BC detection [4]. The marginal data of the breast mass display the growth patterns and biological features. The non-uniform margins of breast mass can usually be malignant cancers, and the accuracy rate of the masses’ segmentation impacts the Malignant Breast Masses Classification (MBMC). Hence, mass classification is an important process utilized in CAD methods to diagnose BC [5], as it supports physicians in treating and analyzing BCs in the earlier phases. Different characteristics exist in breast mass like infection margins, various sizes, and various shapes. Thus, accurate segmentation classification is a common and challenging issue in CAD methods [6].
Digital mammograms often depict difficult and subtle patterns that can be challenging for human radiologists to consistently interpret. Deep learning (DL) models can effectively capture intricate image features and patterns, resulting in significantly enhanced classification accuracy. The incorporation of the DL models into breast mass classification can greatly improve the performance of breast cancer screening programs. Automated systems for BC detection using DL can process a large volume of digital mammograms rapidly, making screening more scalable and accessible. At present, BC detection is challenging because of the variation in breasts in texture, size, location, and shape, and there has been a crucial necessity to decrease false negative and positive conditions and enhance the all-inclusive effectiveness of CAD techniques [7]. With the current development in computers and their improved computational speed and capability, the DL approach is widely recommended in biomedical applications and especially in CAD techniques for mammograms [8]. In recent years, the DL algorithm has made achievements in several computer vision (CV) tasks and has established the ability to address difficult issues in the medical imaging field [9]. Therefore, numerous types of research are proposed to be implemented, specifically in mammography cancer diagnosis, BC segmentation, and classification [10]. Currently, DL approaches have surpassed the basic alteration of convolutional neural network (CNN) methods for existing various complex frameworks, and have also surpassed the image classification outcomes.
This article presents an Intelligent Breast Mass Classification Approach using the Archimedes Optimization Algorithm with Deep Learning (BMCA-AOADL) technique on Digital Mammograms. In the present BMCA-AOADL approach, median filtering (MF)-based noise removal and U-Net segmentation take place as a pre-processing step. For feature extraction, the BMCA-AOADL technique utilizes the SqueezeNet model with AOA as a hyperparameter tuning approach. To detect and classify the breast mass, the BMCA-AOADL technique applies a deep belief network (DBN) approach. The simulation value of the BMCA-AOADL system can be studied on two BC databases.
  • An Intelligent BMCA-AOADL technique comprising pre-processing, the SqueezeNet feature extractor, AOA-based hyperparameter tuning, and DBN classification of BC has been developed. To the best of our knowledge, the BMCA-AOADL technique did not exist in the literature.
  • Pre-processing steps help to enhance the quality of the digital mammograms and isolate the regions of interest, which are the breast masses.
  • The SqueezeNet model is employed for the feature extraction process. The SqueezeNet is known for its efficiency and compact architecture, whilst maintaining a competitive performance, making it a suitable choice for resource-constrained applications like medical image analysis.
  • AOA with the SqueezeNet model for feature extraction is designed, where the hyperparameter optimization process using cross-validation helps to boost the predictive outcome of the BMCA-AOADL model for unseen data.
  • DBNs are employed for the classification process, which have the capability of modeling complex data distributions, making them suitable for BC image classification tasks.

2. Related Works

In [11], a new CAD technique included various phases. The images were segmented in the pre-processing phase and a mask of a lesion was acquired; then, the feature extraction of the DL method was executed using DenseNet-201. Furthermore, hand-crafted features were attained from an image. This technique utilized the CNN framework for extraction, as well as the mutual data selection method, and was supported by the numerous classifiers implemented for selecting the highly sensitive dataset between the features. Malebary and Hashmi [12] introduced a new Breast Mass Classification system termed BMC. This is an enhanced framework that depends on the incorporation of k-mean clustering, the LSTM network of RNN, CNN, and the RF, boosting algorithms for the classification of the breast mass into normal, malignant, and benign.
Jiang et al. [13] presented a three-phase DL approach that depended on an anchor-free object detection technique, called the Probabilistic Anchor Assignment (PAA). Firstly, a single-phase PAA-based detector fully determined suspect BCs in mammography. Secondly, a two-stage ROI detector was developed for more classification and regression of these cancers. Finally, a threshold-adaptive, post-processing method was presented. Eventually, the cancers were categorized using the ROI method. Ragab [14] suggested that a novel deep transfer learning system be developed to accomplish remote sensing scene classification and security. For scene classification, the presented model employed NasNet feature extraction with hyperparameter tuning using a metaheuristic algorithm. The design of the hyperparameter optimization technique helped to adjust the hyperparameter values, which resulted in the achievement of enhanced classification performance. In [15], a DL technique was proposed which included three stages. The pre-processing stage primarily considered normalizing the input images to follow the computation. A novel attention mechanism named the global channel attention method supported the DL architecture for learning inter-channel attention and, therefore, enhanced the segmentation effectiveness of DL frameworks. Narayanan et al. [16] introduced a new DL approach that incorporated the DCNNs and the RF approaches for identifying and classifying BCs. The feature extraction could be conducted using the AlexNet algorithm of the DCNNs and the classifier correctness was improved via the RF method. The images were gathered from the different mammography images of predetermined databases.
In [17], a DL-based convolutional neural network (ConvNet) was introduced to integrate both the mammogram and the medical variation for predicting and classifying breast masses to support the specialist’s decision-making methods. This study was designed to perform an automated and overall representation of breast masses employing the suitable layers in the deep ConvNet technique and by executing data augmentation methods. Chakravarthy et al. [18] used a method of TL to extract the features. The classification efficiency of the weighted KNN (w-KNN) method was enhanced by employing the dragonfly optimizer algorithm (DFOA), crow search optimizer algorithm (CSOA), and PSO as a transformation approach.
In [19], a PSO approach was deployed for selecting important features in the data with ANN for the classification model on a University of California Irvine ML database. In Adebiyi et al.’s study [20], the purpose of this examination was to improve the accuracy of ML approaches for BC analysis. This search executed the ML approaches of RF and SVM with the feature extraction model of linear discriminant analysis (LDA) to the Wisconsin BC (WBC) Database. In [21], PSO was recommended to enhance the efficacy of the classifier with the DT approach on a WBC database.

3. The Proposed Model

In this article, we presented an automated breast mass classification approach named the BMCA-AOADL technique on Digital Mammograms. The main intention of the BMCA-AOADL technique was to exploit the DL model with a bio-inspired algorithm for breast mass classification. In the presented BMCA-AOADL approach, MF-based noise removal, U-Net segmentation, the SqueezeNet feature extractor, AOA-based hyperparameter tuning, and the DBN classification were involved. Figure 1 portrays the workflow of the BMCA-AOADL system.

3.1. Image Pre-Processing

In this work, two stages of image pre-processing were involved: MF-based noise filtering and U-Net segmentation. The MF is the filter approach utilized for noise removal in images and signals [22]. It is very critical in the image processing domain as it was established for the maintenance of edges from noise removal. The basic concept of the MF system is to run over the signal entry by entry, exchanging all the entries with the median of neighboring entries. Next, the U-Net segmentation approach was used to segment the affected regions. The U-net is a type of convolutional neural network (CNN) framework that was developed for the segmentation of life science images containing constrained trained data and to categorize the tumors in breast images [23]. The U-Net architecture comprises expansion and contraction. The expansion part is nearly the same as the CNN operation that consists of a convolution layer, and the objective is to frequently apply 3 × 3 convolution filters on the input images. There is a linear activator or ReLU layer after every convolution layer. The ReLU layer employs an activation function to all the neurons, i.e., m a x ( 0 , x ) (which sets the threshold value to 0 ). Because of the reduction in the number of calculations and parameters within the network, there is a 2 × 2 max-pooling layer with a stride of 2 to decrease the spatial dimension of the image, which causes over-fitting. On the other hand, there is an incremental sampler or in other words, upsampling, together with 2 × 2 convolution layers that is accountable to reduce the particular channel. There is a 3 × 3 convolution layer with a ReLU layer. The objective is to locate the object. During the final layer, there is a 1 × 1 convolutional layer which maps property vectors to a certain class. The steps of the U-Net structure are given below:
(1)
Define U-Net structure with the encoding and decoding part;
(2)
Input the image to the encoding part and downsample it;
(3)
Pass downsampling mapping features to the decoding part;
(4)
Upsample the mapping feature and concatenate it with the mapping feature in the respective encoding layer;
(5)
Repeat step 4 until image size is attained;
(6)
Execute convolution layer to concatenated mapping feature;
(7)
Output segmented images.

3.2. Feature Extraction Using Optimal SqueezeNet Model

For feature extraction, the BMCA-AOADL algorithm utilizes the SqueezeNet algorithm. Early DL architectures had the drawback of necessitating considerable amounts of parameters for training [24]. Thus, researcher workers from Deep-scale, University of California, Berkeley, and Stanford University introduced a novel structure named “SqueezeNet” that has a small-scale model and requires fewer training parameters. The SqueezeNet structure includes eight fire modules and one convolution layer. Figure 2 demonstrates the structure of the SqueezeNet.
The fire module has a convolution layer with “ n ” filters and 1 × 1 filter size, represented as a “Squeeze” layer. Then, there are two cohesive layers, named “Expand” layers. The initial “Expand” layer comprises 4 n filters and   1 × 1   filter sizes. Likewise, the next “Expand” layer comprises 4 n filters and   3 × 3   filter sizes. These “Squeeze” and “Expand” layers are in sequence. Then, in the first convolutional layer and the second and fourth fire models, a max-pooling layer is utilized (stride = 2), which decreases the feature map size. An FC layer (weight = 1000) is used after the concluding fire module, which transforms the feature maps into a 1D vector. These feature vectors pass over the avgpooling layers. Each prior convolution layer follows a ReLu function. Finally, a softmax layer converts the feature vectors into dual classification.
The hyperparameter tuning of the SqueezeNet approach occurs by employing AOA. AOA was inspired by the rule of Archimedes which is regarded as the law of physics [25]. The Archimedes rules are focused on the object that is completely or partially occupied by the fluid. In AOA, the immersed objects can be supposed to be the population individuals (candidate solutions). This process derives from the initialization of a population with objects; in addition, the position of every object can be initialized from the arbitrary model inside the problem-searching space. Afterwards, the equivalent fitness function (FF) is measured. Iteratively, AOA upgrades the object volumes and densities but its accelerations can upgrade the essential of its collisions with neighbor objects. The initializing method of each object can be executed by exploiting it as:
O i = l i + r a n d × u i l i , i = 1,2 , , N
u i and l i denote the upper as well as lower limits of the i t h object and N implies the object counts. The density and volume of all the objects are initialized as:
d e n i = r a n d , v o l i = r a n d
In this example, r a n d indicates the vector of D -dimension with values within zero and one. The acceleration of all the objects is measured as:
a c c i = l b i + r a n d × u b i l b i
A main FF can be measured, and an object with better fitness is distributed as x b e s t ,   d e n b e s t ,   v o l b e s t , and a c c b e s t .
The upgrading procedure of the i t h object volume and density is executed depending on the following equation:
d e n i t + 1 = d e n i t + r a n d × d e n b e s t d e n i t
v o l i t + 1 = v o l i t + r a n d × v o l b e s t v o l i t
where t denotes the present iteration and r a n d implies the random number. Primarily, it is a collision amongst the objects once the object obtains the equilibrium condition. It can be performed as proposed in AOA, utilizing a transfer function that supports a change from the exploration to the exploitation phases.
The formulation of this transfer function is represented as:
T F = e x p t t m a x t m a x
where r m a x denotes the maximal iteration counts. Now, the T F value is enhanced slowly with iterations. However, it still accomplishes unity.
d t + 1 = e x p t t m a x t m a x t t m a x
The value of d t + 1 diminishes with time, although the suitable transmission of this variable supports accomplishing exploration and exploitation balances. The exploration step can be defined by collision among the objects; this step can be assumed once the transfer function is 0.5. The acceleration of the i t h object at iteration t + 1 can be upgraded by choosing random material ( m r ) as:
a c c i t + 1 = d e n m r + ι ƒ m o l m r × a c c m r d e n i t + 1 × v o l i t + 1
where v o l m r , a c c m r , and e n m r stand for the volume, acceleration, and density of m r . The exploitation stage from the AOA assumes no collision among the objects; this step can be executed if the value of the transfer function is superior to 0.5. The acceleration of the i t h object from the exploitation stage is calculated as:
a c c i t + 1 = d e n b e s t + v o l b e s t × a c c b e s t d e n i t + 1 × v o l i t + 1
where a c c b e s t , e n b e s t , and v o l b e s t indicate the optimum object acceleration, density, and volume, correspondingly. The normalized acceleration is represented as:
a c c i n o r m t + 1 = u × a c c i t + 1 min a c c max a c c min a c c + l
where l and u represent the normalized range; it could be assigned as [0.1,0.9]. Once the object is away from the global optimal, the acceleration values are higher. During case exploration, this step can lead to the exploitation stage being proposed.
The position of the i t h particle can be upgraded in the exploration stage by exploiting it as:
x i t + 1 = x i t + C 1 × r a n d × a c c i n o r m t + 1 × d × x r a n d x i t
In contrast, the upgrading procedure of particle positions in the exploitation stage is projected as:
x i t + 1 = x i t + F × C 2 × r a n d × a c c i n o r m t + 1 × d × T × x b e s t x i t
where C 1 and C 2 denote the constants represented by the user, T implies the parameter that is based on the transfer function T = C 3 × T F ,   C 3 represents the constant value, x b e s t stands for the best particle position, and F defines the flag deployed for changing the particle motion direction.
F = + 1 P 0.5 1 P > 0.5
where the value of P is distributed arbitrarily by the user. Finally, the FF can be estimated at the upgrade particle positions after the optimal performance is recorded.
The fitness selection is a crucial factor in the AOA approach. Solution encoding is exploited for assessing the aptitude (goodness) of the candidate solution. Now, the accuracy value is the main condition utilized for designing a fitness function.
F i t n e s s = m a x P
P = T P T P + F P
where F P and T P signify the false and true positive values.

3.3. Image Classification Using the DBN Model

For breast mass classification, the DBN approach can be utilized. The DBN model efficiently leverages the residuals based on the predicted values of the MLP model and the corresponding actual observation to enable prediction [26]. The preliminary step of this component includes the visible layer (VL) of RBM receiving the input datasets—viz., the difference between the observed and predicted values of MLP models. Consequently, these data traverse towards the low hidden layer (HL), where a complex feature extraction system unfolds. The feature-laden output journeys onwards to the HL of the succeeding RBM, which instigates a cascading layer-wise training model. This final RBM incorporates its predecessor’s HL, thereby providing the seamless progression and continuity of extraction features through the network.
Eventually, the concluding RBM funnels the information handled into the topmost output layer. Now, the complex BP model takes the helm, which orchestrates the finetuning of the overall network. In the following, the prediction method is given:
The input vector X   is allocated to the y 1 VL and the following equation computes the probability that the neuron from the HL is turned on:
P H 1 V 1 = σ W H 1 V 1 + b 1
In Equation (16), b shows the bias of VL, and σ indicates the sigmoid function.
H 1 P H 1 V 1
In Equation (17), H 1 is used to create the VL of the next RBM and compute the probability that the neuron from the VL is turned on:
P V 2 H 1 = σ W V 2 H 1 + a 2
In Equation (18), a is the bias of HL, and after computing the probability density, the Gibbs sampling is carried out:
V 2 P V 2 H 1
Update weight and bias.
W W + l r [ P H 1 V 1 V 1 T P H 2 V 2 V 2 T ] b b + l r [ V 1 V 2 ] a a + l r [ H 1 H 2 ]
In Equation (20), the learning rate can be represented as l r .
The BP algorithm is used to minimalize the objective function, viz., mean square error, and to finetune the weight of the network:
E = 1 m i = 1 m ( Y i ' Y i ) 2
In Equation (21),   m represents the number of samples and Y i and Y i exemplify the actual and ideal outputs of the resultant layer.

4. Results and Discussion

In this section, the simulation analysis of the BMCA-AOADL methodology takes place using the MIAS dataset [27], comprising 1000 instances with three classes, as depicted in Table 1. Figure 3 exemplifies the sample images.
Figure 4 reveals the confusion matrices achieved via the BMCA-AOADL method under 60:40 and 70:30 of the TR set/TS set. The outcome demonstrated the effective recognition and classification of three class labels.
In Table 2 and Figure 5, the BC classification outcome of the BMCA-AOADL system is demonstrated with 60:40 of the TR set/TS set provided. The results indicate that the BMCA-AOADL technique effectually categorized three classes. On the 60% TR set, the BMCA-AOADL technique gained average a c c u y , s e n s y , s p e c y , and F m e a s u r e of 96.11%, 94.09%, 97.10%, and 94.07%, respectively. Moreover, on the 40% TS set, the BMCA-AOADL algorithm offered average a c c u y , s e n s y , s p e c y , and F m e a s u r e of 95.83%, 93.74%, 96.88%, and 93.73%, correspondingly.
In Table 3 and Figure 6, the BC classifier outcome of the BMCA-AOADL algorithm is depicted with 70:30 of the TR set/TS set provided. The simulation value showed that the BMCA-AOADL system effectively categorized three classes. On the 70% TR set, the BMCA-AOADL system reached average a c c u y , s e n s y , s p e c y , and F m e a s u r e of 96.48%, 94.71%, 97.38%, and 94.66%, correspondingly. Furthermore, on the 30% TS set, the BMCA-AOADL methodology obtained average a c c u y , s e n s y , s p e c y , and F m e a s u r e of 94.89%, 92.29%, 96.20%, and 92.19%, correspondingly.
Figure 7 illustrates the training accuracy T R _ a c c u y and V L _ a c c u y of the BMCA-AOADL system at 70:30 of the TR set/TS set. The T L _ a c c u y is defined by the assessment of the BMCA-AOADL system on the TR dataset whereas the V L _ a c c u y is computed by estimating the performance on a separate testing dataset. The outcomes displayed that T R _ a c c u y and V L _ a c c u y caused an upsurge with an increase in epochs. Therefore, the performance of the BMCA-AOADL technique improved upon the TR and TS dataset with a rise in the number of epochs.
In Figure 8, the T R _ l o s s and V R _ l o s s curves of the BMCA-AOADL system at 70:30 of the TR set/TS set are depicted. The T R _ l o s s defines the error among the predictive solution and original values on the TR data. The V R _ l o s s represents the measure of the performance of the BMCA-AOADL method on individual validation data. The outcomes show that the T R _ l o s s and V R _ l o s s generally decreased with rising epochs. This portrayed the enhanced performance of the BMCA-AOADL methodology and its ability to produce accurate classification. The lesser value of T R _ l o s s and V R _ l o s s demonstrated the enhanced performance of the BMCA-AOADL method in capturing patterns and relationships.
A detailed PR analysis of the BMCA-AOADL algorithm is defined at 70:30 of the TR set/TS set in Figure 9. The simulation values demonstrated that the BMCA-AOADL algorithm had better outcomes in enhancing PR values. Thus, it can be clear that the BMCA-AOADL algorithm attained greater performances of PR on three class labels.
In Figure 10, a ROC outcome of the BMCA-AOADL method is demonstrated at 70:30 of the TR set/TS set. The outcome value exposed that the BMCA-AOADL approach led to better values of ROC. Moreover, it was obvious that the BMCA-AOADL technique achieved superior values of ROC in three classes.
The performance of the BMCA-AOADL technique is compared with recent DL models in Table 4 and Figure 11 [12]. The result stated that the MCNN-MIC and CMIM-CNN models showed the lowest performance. At the same time, the ResNet, ResNet-VGG, and BMCS-DL ELDM models illustrated moderately closer results. Additionally, the DL-CADSBM model reached considerable performance with a c c u y , s e n s y , s p e c y , and F m e a s u r e of 82%, 90.13%, 96.31%, and 94.76%, respectively. But the BMCA-AOADL technique surpassed the other ones with improved a c c u y , s e n s y , s p e c y , and F m e a s u r e of 96.48%, 94.71%, 97.38%, and 94.66%, respectively. Thus, the BMCA-AOADL technique can be applied to the automated breast mass classification process.

5. Conclusions

In this article, we presented an automated breast mass classification approach named the BMCA-AOADL technique on Digital Mammograms. The main intention of the BMCA-AOADL technique is to exploit the DL model with a bio-inspired algorithm for breast mass classification. In the presented BMCA-AOADL approach, the MF-based noise removal, U-Net segmentation, the SqueezeNet feature extractor, AOA-based hyperparameter tuning, and DBN classification are involved. For feature extraction, the BMCA-AOADL technique utilizes the SqueezeNet model with AOA as a hyperparameter tuning approach. To detect and classify the breast mass, the BMCA-AOADL technique applies the DBN algorithm. The simulation value of the BMCA-AOADL approach can be studied on the MIAS dataset. The experimental values displayed the significant outcomes of the BMCA-AOADL technique compared to other DL models with a maximum accuracy of 96.48%. Future work can perform comprehensive clinical validation and testing of the BMCA-AOADL approach on a large and diverse dataset of digital mammograms. This will help assess its real-world performance and generalizability. In addition, future work can enhance the interpretability and explainability of the BMCA-AOADL model’s predictions. This will help build trust among healthcare professionals and patients by providing insights into why a particular classification decision was made.

Funding

This research work was funded by Institutional Fund Projects under grant no. (IFPIP: 402-611-1443). Therefore, the author gratefully acknowledges technical and financial support provided by the Ministry of Education and Deanship of Scientific Research (DSR), King Abdulaziz University (KAU), Jeddah, Saudi Arabia.

Institutional Review Board Statement

Not Applicable.

Data Availability Statement

Data sharing does not apply to this article as no datasets were generated during the current study.

Conflicts of Interest

The authors declare that they have no conflict of interest. The manuscript was written through the contributions of all authors. All authors have approved the final version of the manuscript.

References

  1. Yurdusev, A.A.; Adem, K.; Hekim, M. Detection and classification of microcalcifications in mammograms images using difference filter and Yolov4 deep learning model. Biomed. Signal Process. Control. 2023, 80, 104360. [Google Scholar] [CrossRef]
  2. Ragab, M.; Nahhas, A.F. Optimal Deep Transfer Learning Model for Histopathological Breast Cancer Classification. Comput. Mater. Contin. 2022, 73, 2849–2864. [Google Scholar]
  3. Nemade, V.; Pathak, S.; Dubey, A.K. Deep learning-based ensemble model for classification of breast cancer. Microsyst. Technol. 2023, 1–15. [Google Scholar] [CrossRef]
  4. Al-Mansour, E.; Hussain, M.; Aboalsamh, H.A.; Amin, F.E. An Efficient Method for Breast Mass Classification Using Pre-Trained Deep Convolutional Networks. Mathematics 2022, 10, 2539. [Google Scholar] [CrossRef]
  5. Atrey, K.; Singh, B.K.; Bodhey, N.K.; Pachori, R.B. Mammography and ultrasound based dual modality classification of breast cancer using a hybrid deep learning approach. Biomed. Signal Process. Control 2023, 86, 104919. [Google Scholar] [CrossRef]
  6. Kumar, S.; Bhupati Bhambu, P.; Pachar, S.; Cotrina-Aliaga, J.C.; Arias-Gonzáles, J.L. Deep Learning-Based Comput-er-Aided Diagnosis Model for the Identification and Classification of Mammography Images. SN Comput. Sci. 2023, 4, 502. [Google Scholar] [CrossRef]
  7. Baccouche, A.; Garcia-Zapirain, B.; Elmaghraby, A.S. An integrated framework for breast mass classification and diagnosis using stacked ensemble of residual neural networks. Sci. Rep. 2022, 12, 12259. [Google Scholar] [CrossRef]
  8. Chakravarthy, S.S.; Rajaguru, H. Automatic Detection and Classification of Mammograms Using Improved Extreme Learning Machine with Deep Learning. IRBM 2022, 43, 49–61. [Google Scholar] [CrossRef]
  9. Li, H.; Niu, J.; Li, D.; Zhang, C. Classification of breast mass in two-view mammograms via deep learning. IET Image Process. 2021, 15, 454–467. [Google Scholar] [CrossRef]
  10. Mahmood, T.; Li, J.; Pei, Y.; Akhtar, F.; Rehman, M.U.; Wasti, S.H. Breast lesions classifications of mammographic images using a deep convolutional neural network-based approach. PLoS ONE 2022, 17, e0263126. [Google Scholar] [CrossRef]
  11. Cruz-Ramos, C.; García-Avila, O.; Almaraz-Damian, J.-A.; Ponomaryov, V.; Reyes-Reyes, R.; Sadovnychiy, S. Benign and Malignant Breast Tumor Classification in Ultrasound and Mammography Images via Fusion of Deep Learning and Handcraft Features. Entropy 2023, 25, 991. [Google Scholar] [CrossRef] [PubMed]
  12. Malebary, S.J.; Hashmi, A. Automated Breast Mass Classification System Using Deep Learning and Ensemble Learning in Digital Mammogram. IEEE Access 2021, 9, 55312–55328. [Google Scholar] [CrossRef]
  13. Jiang, J.; Peng, J.; Hu, C.; Jian, W.; Wang, X.; Liu, W. Breast cancer detection and classification in mammogram using a three-stage deep learning framework based on PAA algorithm. Artif. Intell. Med. 2022, 134, 102419. [Google Scholar] [CrossRef] [PubMed]
  14. Ragab, M. Leveraging mayfly optimization with deep learning for secure remote sensing scene image classification. Comput. Electr. Eng. 2023, 108, 108672. [Google Scholar] [CrossRef]
  15. Yu, X. Automatic Breast Mass Detection and Classification for Mammograms with Deep Learning. Ph.D. Thesis, University of Leicester, Leicester, UK, 2023. [Google Scholar]
  16. Narayanan, K.L.; Krishnan, R.S.; Robinson, Y.H. A Hybrid Deep Learning Based Assist System for Detection and Classification of Breast Cancer from Mammogram Images. Int. Arab. J. Inf. Technol. 2022, 19, 965–974. [Google Scholar] [CrossRef]
  17. Mahmood, T.; Li, J.; Pei, Y.; Akhtar, F.; Jia, Y.; Khand, Z.H. Breast Mass Detection and Classification Using Deep Convolutional Neural Networks for Radiologist Diagnosis Assistance. In Proceedings of the 2021 IEEE 45th Annual Computers, Software, and Applications Conference (COMPSAC), Madrid, Spain, 12–16 July 2021; pp. 1918–1923. [Google Scholar]
  18. Chakravarthy, S.S.; Bharanidharan, N.; Rajaguru, H. Deep Learning-Based Metaheuristic Weighted K-Nearest Neighbor Algorithm for the Severity Classification of Breast Cancer. IRBM 2023, 44, 100749. [Google Scholar] [CrossRef]
  19. Adebiyi, M.O.; Afolayan, J.O.; Arowolo, M.O.; Tyagi, A.K.; Adebiyi, A.A. Breast Cancer Detection Using a PSO-ANN Machine Learning Technique. In Using Multimedia Systems, Tools, and Technologies for Smart Healthcare Services; IGI Global: Hershey, PA, USA, 2023; pp. 96–116. [Google Scholar]
  20. Adebiyi, M.O.; Arowolo, M.O.; Mshelia, M.D.; Olugbara, O.O. A Linear Discriminant Analysis and Classification Model for Breast Cancer Diagnosis. Appl. Sci. 2022, 12, 11455. [Google Scholar] [CrossRef]
  21. Afolayan, J.O.; Adebiyi, M.O.; Arowolo, M.O.; Chakraborty, C.; Adebiyi, A.A. Breast Cancer Detection Using Particle Swarm Optimization and Decision Tree Machine Learning Technique. In Intelligent Healthcare: Infrastructure, Algorithms and ManagementI; Springer Nature Singapore: Singapore, 2022; pp. 61–83. [Google Scholar]
  22. Noor, A.; Zhao, Y.; Khan, R.; Wu, L.; Abdalla, F.Y. Median filters combined with denoising convolutional neural network for Gaussian and impulse noises. Multimed. Tools Appl. 2020, 79, 18553–18568. [Google Scholar] [CrossRef]
  23. Ragab, M.; Albukhari, A.; Alyami, J.; Mansour, R.F. Ensemble Deep-Learning-Enabled Clinical Decision Support System for Breast Cancer Diagnosis and Classification on Ultrasound Images. Biology 2022, 11, 439. [Google Scholar] [CrossRef]
  24. Bernardo, L.S.; Damaševičius, R.; Ling, S.H.; de Albuquerque, V.H.C.; Tavares, J.M.R.S. Modified SqueezeNet Architecture for Parkinson’s Disease Detection Based on Keypress Data. Biomedicines 2022, 10, 2746. [Google Scholar] [CrossRef]
  25. Fathy, A.; Alharbi, A.G.; Alshammari, S.; Hasanien, H.M. Archimedes optimization algorithm based maximum power point tracker for wind energy generation system. Ain Shams Eng. J. 2021, 13, 101548. [Google Scholar] [CrossRef]
  26. Jiang, K.; Huang, Z.; Zhou, X.; Tong, C.; Zhu, M.; Wang, H. Deep belief improved bidirectional LSTM for multivariate time series forecasting. Math. Biosci. Eng. 2023, 20, 16596–16627. [Google Scholar] [CrossRef]
  27. Available online: https://www.kaggle.com/datasets/kmader/mias-mammography (accessed on 12 June 2023).
Figure 1. Workflow of BMCA-AOADL algorithm.
Figure 1. Workflow of BMCA-AOADL algorithm.
Biomimetics 08 00463 g001
Figure 2. SqueezeNet Architecture.
Figure 2. SqueezeNet Architecture.
Biomimetics 08 00463 g002
Figure 3. (a) Normal, (b) benign, (c) malignant.
Figure 3. (a) Normal, (b) benign, (c) malignant.
Biomimetics 08 00463 g003
Figure 4. Confusion matrices of (a,b) 60:40 of TR set/TS set and (c,d) 70:30 of TR set/TS set.
Figure 4. Confusion matrices of (a,b) 60:40 of TR set/TS set and (c,d) 70:30 of TR set/TS set.
Biomimetics 08 00463 g004
Figure 5. Average of BMCA-AOADL algorithm at 60:40 of TR set/TS set.
Figure 5. Average of BMCA-AOADL algorithm at 60:40 of TR set/TS set.
Biomimetics 08 00463 g005
Figure 6. Average of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Figure 6. Average of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Biomimetics 08 00463 g006
Figure 7. A c c u y curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Figure 7. A c c u y curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Biomimetics 08 00463 g007
Figure 8. Loss curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Figure 8. Loss curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Biomimetics 08 00463 g008
Figure 9. PR curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Figure 9. PR curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Biomimetics 08 00463 g009
Figure 10. ROC curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Figure 10. ROC curve of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Biomimetics 08 00463 g010
Figure 11. Comparative outcome of BMCA-AOADL approach with other DL methods.
Figure 11. Comparative outcome of BMCA-AOADL approach with other DL methods.
Biomimetics 08 00463 g011
Table 1. Details on the database.
Table 1. Details on the database.
ClassNo. of Samples
Normal300
Benign350
Malignant350
Total Samples1000
Table 2. BC classifier outcome of BMCA-AOADL algorithm at 60:40 of TR set/TS set.
Table 2. BC classifier outcome of BMCA-AOADL algorithm at 60:40 of TR set/TS set.
Class A c c u y S e n s y S p e c y F M e a s u r e
TR set (60%)
Normal95.5093.1496.4792.35
Benign96.5092.8698.4694.89
Malignant96.3396.2896.3694.95
Average96.1194.0997.1094.07
TS set (40%)
Normal95.7592.8097.0993.17
Benign96.5092.8698.4694.89
Malignant95.2595.5695.0993.14
Average95.8393.7496.8893.73
Table 3. BC classifier outcome of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Table 3. BC classifier outcome of BMCA-AOADL algorithm at 70:30 of TR set/TS set.
Class A c c u y S e n s y S p e c y F M e a s u r e
TR set (70%)
Normal95.7194.2696.3392.92
Benign99.0097.9499.5698.55
Malignant94.7191.9496.2492.49
Average96.4894.7197.3894.66
TS set (30%)
Normal94.6793.4195.2291.40
Benign97.6797.2097.9396.74
Malignant92.3386.2795.4588.44
Average94.8992.2996.2092.19
Table 4. Comparative outcome of BMCA-AOADL approach with other DL methods [12].
Table 4. Comparative outcome of BMCA-AOADL approach with other DL methods [12].
Methods A c c u y S e n s y S p e c y F M e a s u r e
DL-CADSBM93.0093.0095.5894.34
MCNN-MIC82.0090.1396.3194.76
CMIM-CNN83.0093.0095.0193.70
ResNet-ResNet92.0087.0089.0091.00
ResNet-VGG94.0088.0092.0093.00
BMCS-DL ELDM95.0093.0097.0094.00
BMCA-AOADL96.4894.7197.3894.66
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Basheri, M. Intelligent Breast Mass Classification Approach Using Archimedes Optimization Algorithm with Deep Learning on Digital Mammograms. Biomimetics 2023, 8, 463. https://doi.org/10.3390/biomimetics8060463

AMA Style

Basheri M. Intelligent Breast Mass Classification Approach Using Archimedes Optimization Algorithm with Deep Learning on Digital Mammograms. Biomimetics. 2023; 8(6):463. https://doi.org/10.3390/biomimetics8060463

Chicago/Turabian Style

Basheri, Mohammed. 2023. "Intelligent Breast Mass Classification Approach Using Archimedes Optimization Algorithm with Deep Learning on Digital Mammograms" Biomimetics 8, no. 6: 463. https://doi.org/10.3390/biomimetics8060463

Article Metrics

Back to TopTop