Next Article in Journal
Nucleic Acid Aptamer-Based Biosensors: A Review
Previous Article in Journal
Efficient Treatment of Pulpitis via Transplantation of Human Pluripotent Stem Cell-Derived Pericytes Partially through LTBP1-Mediated T Cell Suppression
Previous Article in Special Issue
A Radiotherapy Dose Map-Guided Deep Learning Method for Predicting Pathological Complete Response in Esophageal Cancer Patients after Neoadjuvant Chemoradiotherapy Followed by Surgery
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Equilibrium Optimization Algorithm with Deep Learning Enabled Prostate Cancer Detection on MRI Images

1
Department of Financial Information Security, Kookmin University, Seoul 02707, Republic of Korea
2
Department of Computer Science and Engineering, Saveetha School of Engineering, Saveetha Institute of Medical and Technical Sciences, Chennai 602105, India
3
Big Data and Machine Learning Lab, South Ural State University, Chelyabinsk 454080, Russia
4
College of IBS, National University of Science and Technology, MISiS, Moscow 119049, Russia
5
Department of Convergence Science, Kongju National University, Gongju-si 32588, Republic of Korea
6
Department of Robotics & Mechatronics Engineering, Daegu Gyeongbuk Institute of Science & Technology (DGIST), Daegu 42988, Republic of Korea
*
Authors to whom correspondence should be addressed.
Biomedicines 2023, 11(12), 3200; https://doi.org/10.3390/biomedicines11123200
Submission received: 23 October 2023 / Revised: 22 November 2023 / Accepted: 28 November 2023 / Published: 1 December 2023
(This article belongs to the Special Issue Artificial Intelligence Applications in Cancer and Other Diseases)

Abstract

:
The enlargement of the prostate gland in the reproductive system of males is considered a form of prostate cancer (PrC). The survival rate is considerably improved with earlier diagnosis of cancer; thus, timely intervention should be administered. In this study, a new automatic approach combining several deep learning (DL) techniques was introduced to detect PrC from MRI and ultrasound (US) images. Furthermore, the presented method describes why a certain decision was made given the input MRI or US images. Many pretrained custom-developed layers were added to the pretrained model and employed in the dataset. The study presents an Equilibrium Optimization Algorithm with Deep Learning-based Prostate Cancer Detection and Classification (EOADL-PCDC) technique on MRIs. The main goal of the EOADL-PCDC method lies in the detection and classification of PrC. To achieve this, the EOADL-PCDC technique applies image preprocessing to improve the image quality. In addition, the EOADL-PCDC technique follows the CapsNet (capsule network) model for the feature extraction model. The EOA is based on hyperparameter tuning used to increase the efficiency of CapsNet. The EOADL-PCDC algorithm makes use of the stacked bidirectional long short-term memory (SBiLSTM) model for prostate cancer classification. A comprehensive set of simulations of the EOADL-PCDC algorithm was tested on the benchmark MRI dataset. The experimental outcome revealed the superior performance of the EOADL-PCDC approach over existing methods in terms of different metrics.

1. Introduction

Prostate cancer (PrC) is the second most common cause of death among males and the most frequently diagnosed cancer in males around the world [1]. Earlier diagnosis of PrC is essential for more effective disease management. In several countries, the standard practice for PrC diagnosis relies on high rates of prostate-specific antigen (PSA) in the blood and a digital rectal examination (DRE). In some cases, pre-biopsy magnetic resonance imaging (MRI) may be recommended to guide the biopsy process [2]. Automated computer-aided diagnosis (CAD) and identification systems can address the limitations of the standard radiological analysis by applying quantitative techniques for automated, standardized, and supported regenerative analyses of radiological images [3]. Although PrC has a wide variety of cancers when diagnosed in the earlier phases, the survival rates are increased because of the gradual development of cancer [4]. As a result, efficient monitoring and earlier identification are significant for enhanced survival of patients.
Current research has found that implementing the machine learning (ML) technique in prostate MRI can enhance diagnosis accuracy and decrease inter-reader changeability by focusing on suspected regions on the MRI [5], permitting a more attentive analysis by the radiologist in standard scan analysis. The ML method is also capable of forecasting cancer aggressiveness and therapeutic reactions [6]. Numerous studies have indicated similar implementations among ML techniques and proficient radiologists in head-to-head relations for MRI analysis. The ML approach has a domain of artificial intelligence (AI) to employ statistical methods for learning hidden models from present data and decision making around hidden records [7]. The major function of a machine learner is to make an overall system on the probable allocation of training samples and, further, generalize capability to unseen samples. The learning method relies on the presented data quality. A sample exists in a database with various features. Effective feature extraction is unfortunately challenging for any task [8]. A convolutional neural network (CNN) method could have greater individual effectiveness in real image analyses, which are expected to improve CAD in prostate MRIs. The CNN-based DL technique restructures and revolutionizes the present analytic model [9]. The main and real-time modules of medical prostate MRI analyses are diffusion-weighted imaging (DWI), T2-weighted imaging (T2WI), and apparent diffusion coefficient (ADC) classifications. Numerous earlier studies of the DL method engaged a PC analysis employing only one or two of the aforementioned systems and, therefore, were not directly compared with medical efficiency [10].
The study proposes an Equilibrium Optimization Algorithm with Deep Learning-based Prostate Cancer Detection and Classification (EOADL-PCDC) technique applied to MRIs. The main goal of the EOADL-PCDC method lies in the recognition and categorization of prostate cancer. To achieve this, the EOADL-PCDC technique applies image preprocessing to improve the image quality. In addition, the EOADL-PCDC technique follows the CapsNet (capsule network) for the feature extraction approach. The EOA-based hyperparameter tuning increases the efficiency of the CapsNet model. For PrC classification, the EOADL-PCDC technique makes use of a stacked bidirectional long short-term memory (SBiLSTM) model. A comprehensive set of simulations of the EOADL-PCDC technique was tested on the benchmark MRI data.

2. Related Works

Singh et al. [11] implemented the DL technique for diagnosing PrC by employing the notion of Gleason grading. A 3D-CNN is utilized for monitoring the affected area and forecasting the cancerous area by employing an Epithelial and Gleason grading network. In [12], a hybrid method was developed for accurately analyzing mpMRI inspection and forecasting PI-RADS scores. In this developed technique, feature mapping of mpMRI is extracted with the help of Darknet53, MobilenetV2, and Efficientnetb0 frameworks. Further, the feature maps acquired utilizing the above-mentioned frameworks are incorporated. The combined feature maps have been subjected to neighborhood components analysis (NCA) for extracting redundant features. Li et al. [13] introduced an imaging pattern that depends on multi-parameter MRI. Primarily, a multi-view radiomics framework approach was developed. Secondarily, logistic regression algorithms were exploited for extracting features and making a framework. Lastly, a Swin Transformer model was developed and trained through transfer learning (TL) methods.
In [14], a technique for automated classification of the prostate area and cancerous area with the help of SegNet, a deep-CNN (DCNN) framework, was introduced. This method employs the PROSTATEx database to train the framework and incorporate various categorizations into three channels of one image. For all subjects, every part consisting of the PCa area, transition zone (TZ), and peripheral zone (PZ) is preferred. Ye et al. [15] presented a PrC identification that depends on the DL method, PSP-Net+VGG16.
The DCNN segmentation technique dependent upon the PSP-Net was used to make an atrous convolutional residual architecture framework extraction network. Primarily, a 3D prostate MRI was transformed into 2D image parts and trained to rely on the PSP-Net and the VGG_16 framework, which could be employed for detecting the targeted area and classifying the normal prostate and PrC. Ragab et al. [16] presented an Archimedes Optimizer Algorithm with DL-based PrC Classification (AOADLB-P2C) technique to execute preprocessing in two phases. Primarily, the technique extracts features through a DenseNet-161 architecture with the RMSProp optimization. Lastly, the method categorizes PCa utilizing the AOA with the least-square SVM (LS-SVM) approach.
In [17], a technique for computational identification of EPE on multivariant MRI exploiting DL was developed. This technique can include two phases. Initially, DL algorithms are trained by implementing the MRI as input for producing cancerous possibility maps both outside and inside the prostate. Next, this analysis makes an image postprocessing pipeline to produce estimates for the EPE position. Bouslimi and Echi [18] suggested a CAD technique. This developed method examines the convolutional neural networks (UNet) algorithm to identify PCa cancers and segmentation for accurately collecting cancer categories. This research provides a completely automated technique through MultiResUNet, which was primarily developed for the segmentation of skin cancers.
In [19], a new automated classification model was developed that combined various DL methods to identify PrC from ultrasound (US) and MRI images. Furthermore, the presented approach describes why a certain decision is made in the input of US or MRI images. In [20], a strong DL-CNN is applied using a transfer learning algorithm. The outcomes are compared to different ML approaches (different kernels, Decision Tree (DT), and SVM). In [21], a new DL model is used for creating a pipeline for the classification and segmentation of MRI images. The two steps of the DL technique are given as follows: a U-Net model to segment ROI in the first stage and an LSTM model for categorizing the ROI as non-cancerous or cancerous.
Several automated tools have been proposed in this work for efficient prostate cancer detection and classification. Although ML and DL techniques exist in the prior research, still it is essential to increase the efficiency of PrC classification. The number of parameters of DL techniques also quickly increases, leading to model over-fitting due to the continual deepening of the model. Simultaneously, dissimilar hyperparameters have a considerable influence on the CNN model performance. Thus, we apply EOA for the parameter selection of the CapsNet architecture. A summary of reviewed works is given in Table 1.

3. The Proposed Model

In this article, we introduce an automatic prostate cancer diagnosis model using the EOADL-PCDC technique on MRI Images. The main goal of the EOADL-PCDC algorithm lies in the detection and classification of PrC. The presented model involves four major stages, namely, image preprocessing, feature extractor, EOA-based hyperparameter tuning, and SBiLSTM-based classification. Figure 1 represents the overall working flow of the EOADL-PCDC technique.

3.1. Image Preprocessing

The CLAHE technique is used to improve the contrast level to preprocess the input images. It is an image enhancement method that has two fundamental aspects: (1) clip limit C l i m i t and (2) non-overlapping regions Y c o n t e x t u a l [22]. In this model, both parameters are accountable for controlling the improved quality of the image. X a v refers to the average number of pixels in the grayscale as follows:
X a v = X c r X × X c r Y X g
where X   c r X indicates the pixel count in the x dimensions of Y c o n t e x t u a l , X g denotes the gray level count in the Y c o n t e x t u a l , and X c r Y shows the pixel count in the y dimensions of Y c o n t e x t u a l .
X a c i s = X c X g
The distributed pixel is evaluated as follows:
P d = X g X l p
In Equation (3), X l p denotes the residual number of clipped image pixels.

3.2. Feature Extraction Using CapsNet

For the feature extraction process, the EOADL-PCDC technique uses the CapsNet model. The Conv layer, PrimaryCaps layer, and DigitCaps layer are the three fundamental layers of CapsNet [23]. Furthermore, using three FC layers, CapsNet forms a reconstructed stage. The Conv layer extracts the main key of input images. The original Sabour structure chooses the ReLU function, with 256 filters, or channels, having a size of 9 × 9 and stride of 1. Hence, the 256 channels given with the dimension of 20 × 20 from the Conv layers, the size of 9 × 9 with a depth of 256, and a stride of 2 are used, which leads to 32 PrimaryCaps layers of dimension 6 × 6 × 8 , where PrimaryCap contains 8D (eight dimensions). This operation generates 1152 capsules. All the capsules have two different components: orientation and magnitude. The orientation comprises the instantiate properties or parameters of the entities, and the magnitude is the probability where the entity occurs.
When the capsule is calculated, it decides which data will be passed to the following layer. In addition, the squashing function attains a vector with 0 and 1 values, which ensures that a smaller vector takes values closer to 0 and a larger vector has a value below 1. Equation (4) demonstrates the squashing function, in which v j denotes the output vector of the jth capsules and s j is its overall input.
v j = s q u a s h s j = s j 2 1 + s j 2 s j s j
For the PrimaryCaps layer, the overall input to s j capsules is a weighted sum through each prediction vector u ^ j | i from the capsule by multiplying the output u i in the below layer by the weighted matrices W i j , which implies u ^ j | i = W i j u j and s j = Σ i u ^ j | i . This function exploits a weight transform matrix that encodes the spatial significance and other relationships amongst the features of the present one and the low-level capsule. If the estimated prediction vector has a higher value than the potential parent, then the coupling coefficient c i j value is adjusted to select the right connection path through the dynamic routing mechanism. The coupling coefficient between the ith capsule and the capsule in the top layer sums to 1 and is defined by the routing softmax function, where initial logits b i j are log -likelihoods that ith capsule must be coupled with the jth capsule:
c i j = s o f t m a x b i = e x p b i j Σ k e x p b i k

3.3. Parameter Tuning Using EOA

The EOA is used to adjust the hyperparameter based on the CapsNet model. EOA is a newly established metaheuristic optimization strategy which attempts to keep the balance between the exploitation and exploration phases [24]. Exploitation is used to locally search the space to obtain the best solution and increase the search quality, while exploration is used to globally search the space but prevents the local optimal solution. The EOA technique is derived from the dynamic mass balance of the control volume system. Similar to other metaheuristic algorithms, EOA begins with population initialization, which is generated based on the dimensions of the feature size and the number of particles:
V d P d t = Q P e q Q P + G
The equation signifies the random initial population as follows:
p i i n i t i a l = p m i n   + r a n d i p m a x   p m i n
where P m i n   and P m a x denote the minimal and maximal concentration of particles, correspondingly, p i i n i t i a l shows the initial concentration vectors of the ith particles,   n refers to the number of particles in the population, and r a n d i belongs to 0 ,   1 .
The equilibrium state is used to conclude the optimization technique as it is globally enhanced, and there is no knowledge of optimization at the beginning. Consider four different particles that remain the best throughout the entire optimization algorithm. The number of particles selected is random and dissimilar from that in other optimization techniques. The five chosen objects that help to create a vector called equilibrium pooling are discussed below.
p e q . p o o l = p e q . 1 , P e q . 2 l , P e q . 3 , P e q . 4 , P e q . a v g
Eventually, an exponential term (E) updates the concentration and balance between exploitation and exploration, and finally aims at achieving better optimization. ε denotes the random vector and ranges within 0 ,   1 , and is given as follows:
E = e ε t t 0
In Equation (9), t differs with the difference in iteration (i), which can be signified as:
t = 1 i   max i k 2 i   max i
where i shows the existing iteration and max i denotes the maximum iteration number. k 2 denotes the parameter that manages the exploitation capability. The succeeding formula demonstrates that if the searching performance is delayed by augmenting the exploitation and exploration capabilities, then the convergence is easier to obtain.
t 0 = 1 l n k 1 s i g n m 0.5 1 e . t + t
In the above equation, k 1 shows the exploration capability. The greater the value of k 2 , the lower the exploration and the higher the exploitation capability. s i g n m 0.5 denotes the direction of exploitation and exploration. The values of m lie within [ 0 ,   1 ]. The amended version of Equation (9) is given below:
E = k . s i g n m 0.5 e t 1
Then, the generation rate is the next important stage, which gives an accurate solution to the optimization process by ensuring a better exploitation stage. There exist several algorithms for computing the generation rate amongst well-known models for 1D space, as given below.
H G = H 0 · e t t 0
In Equation (13), H 0 refers to the initial value and ε denotes the decay constant:
H G = H 0 · E
E 0 = G C P P e q P
G C P = 0.5 m i f   m > G P 0 e l s e
Now, G C P refers to the generation control parameter. Finally, the EOA updating rule can be given in the following equation:
P = p e q + P p e q E + F   V 1 E
The EOA system derives an FF to accomplish higher efficacy of classification. It determines a positive integer to be the greater outcome of the solution candidate. Thus, the decay of the classifier error rate is assumed as an FF as follows:
f i t n e s s x i = C l a s s i f i e r E r r o r R a t e x i = n o .   o f   m i s c l a s s i f i e d   s a m p l e s T o t a l   n o .   o f   s a m p l e s 100

3.4. Image Classification Using SBiLSTM Model

The SBiLSTM model is applied for prostate cancer classification. BiLSTM is obtained via the data sequence of time-dependent input, together with latent relations between the input features and the destination [25]. It is proposed to record the long-term prior experience and manage it by applying memory units. LSTM provides several benefits while analyzing and projecting time series data. In the network module, a chain structure exists in LSTM and RNN. The RNN model encompasses a single neuron structure, while the LSTM model includes cells with three gates, namely, forget, output, and input gates. The succeeding equation demonstrates the computation method for the above three gates:
i n p u t   t = σ W i x t + V i h t 1 + b i
In Equation (19), W i and y i signify the weights of the input gate, h t 1 symbolizes the output of prior cells, x t signifies the input of existing cells, and σ denotes the sigmoid function:
f o r g e t   t = σ W f x t + V f h t 1 + b f
This gate states that data in a cell must be forgotten, and w f and y f are the weights of the forgotten gate. By using the following expression, the update procedure was carried out:
C ˜ t = t a n h W c x t + V c h t 1 + b c
C t = f o r g e t   t C t 1 + i n p u f t C ˜ t
Equation (21) shows the candidate unit for memory which generates present data. Moreover, Equation (22) illustrates the procedure of renewing conditions of the cell. i V V c and y c show the weight of the substitute and existing condition and shows the Hadamard product.
o u t p u t   t = σ W 0 x t + V 0 h t 1 + b 0
h t = o u t p u t t t a n h C t
Equations (23) and (24) analyze the output gate. Initially, the sigmoid layer is obtained via the cell states to be outputted. Through the t a n h function, the cell state upgraded is processed, and the upgraded states are multiplied using output ( t ) to attain h t . y 0 shows the weight of the output gate. These frameworks create a BLSTM network to extract data features. In comparison to classical LSTM, BiLSTM is used to extract more context data. Backwards and forward time series are used to gain data regarding the present timestamp in the prior period and the future to create accurate time series predictions. Figure 2 shows the structure of the BiLSTM model.
The BiLSTM layer is added in the SBiLSTM mechanism to the stacked layer. Accordingly, output representations from the stacking layer are transferred to FC and the regression layer determines the temperature values. The ReLU function is used in the output layer to mitigate gradient vanishing problems:
t e m p e r a t u r e = R e L U W 0 h f + b 0
In Equation (25),   W 0 and b 0 correspondingly show the weighted matrices and bias in the regression layer. h f indicates the output of the FC layer:
L o s s = 1 T r = 1 T t e m p e r a t u r e r t e m p e r a t u r e r 2
The proposed model was trained through the BP model.

4. Results and Discussion

The presented approach was simulated using the Python 3.6.5 tool. The prostate cancer classification outcomes of the EOADL-PCDC method were validated on the dataset [16], including 400 samples with two class labels as described in Table 2.
Figure 3 illustrates the confusion matrices made by the EOADL-PCDC method on 80:20 and 70:30 of the TR phase (TRAP)/TS phase (TESP). The experimental data specify the efficient identification of the Prostate and Brachytherapy samples under every class.
In Table 3 and Figure 4, the prostate cancer detection outcome values of the EOADL-PCDC method are shown at 80:20 of the TRAP/TESP. The stimulation values highlighted that the EOADL-PCDC system recognizes two classes. With 80% TRAP, the EOADL-PCDC technique gains average a c c u y , r e c a l , s p e c y , F s c o r e , and MCC of 99.69%, 99.70%, 99.70%, 99.69%, and 99.38%, correspondingly. Moreover, with a 20% TESP, the EOADL-PCDC method attains average a c c u y , r e c a l , s p e c y , F s c o r e , and MCC of 98.75%, 98.61%, 98.61%, 98.73%, and 97.50%, correspondingly.
As shown in Figure 5, TR and TS a c c u y curves are defined to calculate the performance of the EOADL-PCDC technique on 80:20 of TRAP/TESP. The TR and TS a c c u y curves show the outcomes of the EOADL-PCDC system over numerous epochs. The figure provides relevant details about the learning process and generalizability of the EOADL-PCDC technique. The TR and TS a c c u y curves are enriched with the improvement in epoch count. It can be noticeable that the EOADL-PCDC algorithm yields enhanced testing a c c u y and has the proficiency to identify the patterns in the TR and TS datasets.
The comprehensive TR and TS loss values of the EOADL-PCDC system on 80:20 of TRAP/TESP over epochs are shown in Figure 6. The TR loss states the model loss minimized over epochs. Initially, the loss value is minimized as the model adapts the weight to minimize the predictive error on the TR and TS datasets. The loss curve exhibits the level where the model fits the training dataset. The TR and TS loss reduced progressively and showed that the EOADL-PCDC method efficiently learns the patterns given in the TR and TS datasets. The EOADL-PCDC technique adapts the parameters for the reduction in the variance between the predictive and original training labels.
As shown in Figure 7, the PR analysis of the EOADL-PCDC algorithm on 80:20 of TRAP/TESP is confirmed by plotting precision against recall. The simulated outcomes confirm that the EOADL-PCDC method attains increased PR values in each class. The figure describes the model learned for recognizing diverse class labels. The EOADL-PCDC technique achieves better experimental outcomes in the detection of positive samples with reduced false positives.
The ROC examination presented by the EOADL-PCDC model on 80:20 of TRAP/TESP is demonstrated in Figure 8, which has the ability to discriminate the classes. The figure provides insights into the tradeoff between the TPR and FPR rates over dissimilar classification thresholds and differing counts of epochs. It represents the accurately forecasted outcomes of the EOADL-PCDC method on the classification of dissimilar class labels.
In Table 4 and Figure 9, the prostate cancer recognition-simulated investigation of the EOADL-PCDC model is demonstrated at 70:30 of the TRAP/TESP. The observation data highlighted that the EOADL-PCDC models identify two classes. With 70% TRAP, the EOADL-PCDC system attains average a c c u y , r e c a l , s p e c y , F s c o r e , and MCC of 99.29%, 99.29%, 99.29%, 99.29%, and 98.57%, respectively. Moreover, with a 30% TESP, the EOADL-PCDC method gains average a c c u y , r e c a l , s p e c y , F s c o r e , and MCC of 99.17%, 99.14%, 99.14%, 99.17%, and 98.34%, correspondingly.
As presented in Figure 10, TR and TS a c c u y curves are shown to calculate the performance of the EOADL-PCDC method on 70:30 of TRAP/TESP. The TR and TS a c c u y curves show the outcomes of the EOADL-PCDC algorithm over dissimilar epochs. The figure provides significant details about the learning task and generalizability of the EOADL-PCDC method. With the improvement in epoch count, the TR and TS a c c u y curves acquire superior outcomes. The EOADL-PCDC method achieves maximum testing accuracy, which has the proficiency to identify the patterns in the TR and TS datasets.
Figure 11 illustrates the detailed TR and TS loss values of the EOADL-PCDC model on 70:30 of TRAP/TESP over epochs. The TR loss shows the model loss minimized over epochs. The loss values were minimized as a model adjusted the weight for the reduction in the predictive error on the TR and TS datasets. The loss curves exhibit the extent to which the model fits the trained dataset. The TR and TS loss reduced progressively and showed that the EOADL-PCDC method efficiently learns the patterns given in the TR and TS dataset. Note that the EOADL-PCDC algorithm adapts the parameters to diminish the dissimilarity between the original and predictive training labels.
As shown in Figure 12, the PR examination of the EOADL-PCDC model on 70:30 of TRAP/TESP is represented by plotting precision against recall. The experimental data confirm that the EOADL-PCDC methodology enhanced PR values under each class. The figure demonstrates that the model learns to detect dissimilar class labels. The EOADL-PCDC system accomplishes higher experimental outcomes in the detection of positive samples with diminished false positives.
The ROC investigation provided by the EOADL-PCDC method at 70:30 of the TRAP/TESP is demonstrated in Figure 13, which has the capability to discriminate the classes. The figure provides valued insights into the tradeoff between the TPR and FPR rates over dissimilar classification thresholds and varying counts of epochs. It obtains the accurately forecasted outcomes of the EOADL-PCDC method on the classification of several class labels.
The comparison study of the EOADL-PCDC method with current models is provided in Table 5 and Figure 14 [16]. The results indicate that the DT model achieves ineffectual performance. Next, the NB, SVM-Gaussian, SVM-RBF, and GoogleNet models obtained slightly enhanced outcomes. Although the AOADLB-P2C model reaches near-optimal performance, the EOADL-PCDC technique obtains maximum performance with a c c u y , s e n s y , s p e c y , and F s c o r e of 99.69%, 99.70%, 99.70%, and 99.69%, correspondingly. These outcomes ensured the superior outcome of the EOADL-PCDC method on the PC classification algorithm.

5. Conclusions

In this article, we introduced an automated prostate cancer diagnoses method using the EOADL-PCDC method on MRI images. The main goal of the EOADL-PCDC technique lies in the recognition and categorization of prostate cancer. The presented model involves four major stages, namely, image preprocessing, CapsNet feature extraction, EOA-based hyperparameter tuning, and SBiLSTM-based classification. Primarily, the EOADL-PCDC technique applies image preprocessing to improve the image quality. In addition, the EOADL-PCDC technique follows CapsNet for the feature extraction process. The EOA-based hyperparameter tuning is used to enhance the performance of CapsNet. The EOADL-PCDC technique makes use of the SBiLSTM model for prostate cancer classification. A comprehensive set of simulations of the EOADL-PCDC technique was tested on the benchmark MRI dataset. The experimental outcomes revealed the superior performance of the EOADL-PCDC method over the existing techniques under various metrics. In the future, the EOADL-PCDC method can be further optimized to accommodate larger and more diverse datasets, enhancing its generalizability across different patient populations. Additionally, exploring the model’s potential for real-time or near-real-time diagnosis in clinical settings could significantly contribute to improving patient outcomes through timely interventions.

Author Contributions

Conceptualization, E.Y.; Data curation, K.S. and S.K.; Formal analysis, E.Y. and S.K.; Funding acquisition, C.S.; Investigation, K.S. and I.M.; Methodology, E.Y. and K.S.; Project administration, C.S.; Resources, C.S.; Software, S.K.; Supervision, C.S.; Validation, K.S., C.S. and I.M.; Visualization, S.K. and I.M.; Writing—original draft, E.Y.; Writing—review and editing, C.S. and I.M. All authors have read and agreed to the published version of the manuscript.

Funding

This research was supported by the Basic Science Research Program through the National Research Foundation of Korea (NRF) funded by the Ministry of Science, ICT and Future Planning (No. RS-2023-00253613).

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

Dataset information is available in the manuscript.

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. Hamm, C.A.; Baumgärtner, G.L.; Biessmann, F.; Beetz, N.L.; Hartenstein, A.; Savic, L.J.; Froböse, K.; Dräger, F.; Schallenberg, S.; Rudolph, M.; et al. Interactive Explainable Deep Learning Model Informs Prostate Cancer Diagnosis at MRI. Radiology 2023, 307, e222276. [Google Scholar] [CrossRef]
  2. Bosma, J.S.; Saha, A.; Hosseinzadeh, M.; Slootweg, I.; de Rooij, M.; Huisman, H. Annotation-efficient cancer detection with report-guided lesion annotation for deep learning-based prostate cancer detection in bpMRI. arXiv 2021, arXiv:2112.05151. [Google Scholar]
  3. Mehralivand, S.; Yang, D.; Harmon, S.A.; Xu, D.; Xu, Z.; Roth, H.; Masoudi, S.; Kesani, D.; Lay, N.; Merino, M.J. Deep learning-based artificial intelligence for prostate cancer detection at biparametric MRI. Abdom. Radiol. 2022, 47, 1425–1434. [Google Scholar] [CrossRef] [PubMed]
  4. Bosma, J.S.; Saha, A.; Hosseinzadeh, M.; Slootweg, I.; de Rooij, M.; Huisman, H. Semi-supervised Learning with Report-guided Pseudo Labels for Deep Learning-based Prostate Cancer Detection Using Biparametric MRI. Radiol. Artif. Intell. 2023, 5, e230031. [Google Scholar] [CrossRef] [PubMed]
  5. Li, D.; Han, X.; Gao, J.; Zhang, Q.; Yang, H.; Liao, S.; Guo, H.; Zhang, B. Deep learning in prostate cancer diagnosis using multiparametric magnetic resonance imaging with whole-mount histopathology referenced delineations. Front. Med. 2022, 8, 810995. [Google Scholar] [CrossRef] [PubMed]
  6. Sun, Z.; Wu, P.; Cui, Y.; Liu, X.; Wang, K.; Gao, G.; Wang, H.; Zhang, X.; Wang, X. Deep-Learning Models for Detection and Localization of Visible Clinically Significant Prostate Cancer on Multi-Parametric MRI. J. Magn. Reson. Imaging 2023, 58, 1067–1081. [Google Scholar] [CrossRef] [PubMed]
  7. Yi, Z.; Ou, Z.; Hu, J.; Qiu, D.; Quan, C.; Othmane, B.; Wang, Y.; Wu, L. Computer-aided diagnosis of prostate cancer based on deep neural networks from multi-parametric magnetic resonance imaging. Front. Physiol. 2022, 13, 918381. [Google Scholar] [CrossRef]
  8. Liu, X.; Han, C.; Cui, Y.; Xie, T.; Zhang, X.; Wang, X. Detection and segmentation of pelvic bones metastases in MRI images for patients with prostate cancer based on deep learning. Front. Oncol. 2021, 11, 773299. [Google Scholar] [CrossRef]
  9. Zhao, L.; Bao, J.; Wang, X.; Qiao, X.; Shen, J.; Zhang, Y.; Jin, P.; Ji, Y.; Zhang, J.; Su, Y.; et al. Detecting Adverse Pathology of Prostate Cancer with a Deep Learning Approach Based on a 3D Swin-Transformer Model and Biparametric MRI: A Multicenter Retrospective Study. J. Magn. Reson. Imaging 2023. [Google Scholar] [CrossRef]
  10. Min, Z.; Bianco, F.J.; Yang, Q.; Rodell, R.; Yan, W.; Barratt, D.; Hu, Y. Controlling false Positive/Negative rates for deep-Learning-Based prostate cancer detection on multiparametric MR images. In Proceedings of the Medical Image Understanding and Analysis: 25th Annual Conference, MIUA 2021, Oxford, UK, 12–14 July 2021; Proceedings 25. Springer International Publishing: Berlin/Heidelberg, Germany, 2021; pp. 56–70. [Google Scholar]
  11. Singh, S.K.; Sinha, A.; Singh, H.; Mahanti, A.; Patel, A.; Mahajan, S.; Pandit, A.K.; Varadarajan, V. A novel deep learning-based technique for detecting prostate cancer in MRI images. Multimed. Tools Appl. 2023, 1–15. [Google Scholar] [CrossRef]
  12. Yildirim, K.; Yildirim, M.; Eryesil, H.; Talo, M.; Yildirim, O.; Karabatak, M.; Ogras, M.S.; Artas, H.; Acharya, U.R. Deep learning-based PI-RADS score estimation to detect prostate cancer using multiparametric magnetic resonance imaging. Comput. Electr. Eng. 2022, 102, 108275. [Google Scholar] [CrossRef]
  13. Li, C.; Deng, M.; Zhong, X.; Ren, J.; Chen, X.; Chen, J.; Xiao, F.; Xu, H. Multi-view radiomics and deep learning modeling for prostate cancer detection based on multi-parametric MRI. Front. Oncol. 2023, 13, 1198899. [Google Scholar] [CrossRef] [PubMed]
  14. Lai, C.C.; Wang, H.K.; Wang, F.N.; Peng, Y.C.; Lin, T.P.; Peng, H.H.; Shen, S.H. Autosegmentation of prostate zones and cancer regions from biparametric magnetic resonance images by using deep-learning-based neural networks. Sensors 2021, 21, 2709. [Google Scholar] [CrossRef] [PubMed]
  15. Ye, L.Y.; Miao, X.Y.; Cai, W.S.; Xu, W.J. Medical image diagnosis of prostate tumor based on PSP-Net+ VGG16 deep learning network. Comput. Methods Programs Biomed. 2022, 221, 106770. [Google Scholar] [CrossRef] [PubMed]
  16. Ragab, M.; Kateb, F.; El-Sawy, E.K.; Binyamin, S.S.; Al-Rabia, M.W.; Mansouri, R.A. Archimedes Optimization Algorithm with Deep Learning-Based Prostate Cancer Classification on Magnetic Resonance Imaging. Healthcare 2023, 11, 590. [Google Scholar] [CrossRef] [PubMed]
  17. Moroianu, Ş.L.; Bhattacharya, I.; Seetharaman, A.; Shao, W.; Kunder, C.A.; Sharma, A.; Ghanouni, P.; Fan, R.E.; Sonn, G.A.; Rusu, M. Computational Detection of Extraprostatic Extension of Prostate Cancer on Multiparametric MRI Using Deep Learning. Cancers 2022, 14, 2821. [Google Scholar] [CrossRef] [PubMed]
  18. Bouslimi, Y.; Echi, A.K. Deep Learning Based Localisation and Segmentation of Prostate Cancer from mp-MRI Images. ELCVIA Electron. Lett. Comput. Vis. Image Anal. 2023, 22, 52–70. [Google Scholar]
  19. Hassan, M.R.; Islam, M.F.; Uddin, M.Z.; Ghoshal, G.; Hassan, M.M.; Huda, S.; Fortino, G. Prostate cancer classification from ultrasound and MRI images using deep learning based Explainable Artificial Intelligence. Future Gener. Comput. Syst. 2022, 127, 462–472. [Google Scholar] [CrossRef]
  20. Abbasi, A.A.; Hussain, L.; Awan, I.A.; Abbasi, I.; Majid, A.; Nadeem, M.S.A.; Chaudhary, Q.A. Detecting prostate cancer using deep learning convolution neural network with transfer learning approach. Cogn. Neurodynamics 2020, 14, 523–533. [Google Scholar] [CrossRef]
  21. Gavade, A.B.; Nerli, R.; Kanwal, N.; Gavade, P.A.; Pol, S.S.; Rizvi, S.T.H. Automated diagnosis of prostate cancer using mpmri images: A deep learning approach for clinical decision support. Computers 2023, 12, 152. [Google Scholar] [CrossRef]
  22. Narayan, V.; Mall, P.K.; Alkhayyat, A.; Abhishek, K.; Kumar, S.; Pandey, P. Enhance-Net: An Approach to Boost the Performance of Deep Learning Model Based on Real-Time Medical Images. J. Sens. 2023, 2023, 8276738. [Google Scholar] [CrossRef]
  23. Rangel, G.; Cuevas-Tello, J.C.; Rivera, M.; Renteria, O. A Deep Learning Model Based on Capsule Networks for COVID Diagnostics through X-ray Images. Diagnostics 2023, 13, 2858. [Google Scholar] [CrossRef] [PubMed]
  24. Dey, A.; Chattopadhyay, S.; Singh, P.K.; Ahmadian, A.; Ferrara, M.; Sarkar, R. A hybrid meta-heuristic feature selection method using golden ratio and equilibrium optimization algorithms for speech emotion recognition. IEEE Access 2020, 8, 200953–200970. [Google Scholar] [CrossRef]
  25. Latif, A.; Mehedi, I.M.; Vellingiri, M.T.; Meem, R.J.; Palaniswamy, T. Enhanced Remora Optimization with Deep Learning Model for Intelligent PMSM Drives Temperature Prediction in Electric Vehicles. Axioms 2023, 12, 852. [Google Scholar] [CrossRef]
Figure 1. The overall working flow of the EOADL-PCDC technique.
Figure 1. The overall working flow of the EOADL-PCDC technique.
Biomedicines 11 03200 g001
Figure 2. BiLSTM structure.
Figure 2. BiLSTM structure.
Biomedicines 11 03200 g002
Figure 3. Confusion matrices of (a,b) 80% of TRAP/20% of TESP and (c,d) 70% of TRAP/30% of TESP.
Figure 3. Confusion matrices of (a,b) 80% of TRAP/20% of TESP and (c,d) 70% of TRAP/30% of TESP.
Biomedicines 11 03200 g003
Figure 4. Average of the EOADL-PCDC system on 80:20 of TRAP/TESP.
Figure 4. Average of the EOADL-PCDC system on 80:20 of TRAP/TESP.
Biomedicines 11 03200 g004
Figure 5. A c c u y curve of the EOADL-PCDC algorithm on 80:20 of TRAP/TESP.
Figure 5. A c c u y curve of the EOADL-PCDC algorithm on 80:20 of TRAP/TESP.
Biomedicines 11 03200 g005
Figure 6. Loss curve of the EOADL-PCDC system on 80:20 of TRAP/TESP.
Figure 6. Loss curve of the EOADL-PCDC system on 80:20 of TRAP/TESP.
Biomedicines 11 03200 g006
Figure 7. PR analysis of EOADL-PCDC technique on 80:20 of TRAP/TESP.
Figure 7. PR analysis of EOADL-PCDC technique on 80:20 of TRAP/TESP.
Biomedicines 11 03200 g007
Figure 8. ROC analysis of the EOADL-PCDC technique on 80:20 of TRAP/TESP.
Figure 8. ROC analysis of the EOADL-PCDC technique on 80:20 of TRAP/TESP.
Biomedicines 11 03200 g008
Figure 9. Average of the EOADL-PCDC method on 70:30 of TRAP/TESP.
Figure 9. Average of the EOADL-PCDC method on 70:30 of TRAP/TESP.
Biomedicines 11 03200 g009
Figure 10. A c c u y curve of the EOADL-PCDC technique on 70:30 of TRAP/TESP.
Figure 10. A c c u y curve of the EOADL-PCDC technique on 70:30 of TRAP/TESP.
Biomedicines 11 03200 g010
Figure 11. Loss curve of the EOADL-PCDC technique on 70:30 of TRAP/TESP.
Figure 11. Loss curve of the EOADL-PCDC technique on 70:30 of TRAP/TESP.
Biomedicines 11 03200 g011
Figure 12. PR curve of the EOADL-PCDC algorithm on 70:30 of TRAP/TESP.
Figure 12. PR curve of the EOADL-PCDC algorithm on 70:30 of TRAP/TESP.
Biomedicines 11 03200 g012
Figure 13. ROC curve of the EOADL-PCDC model on 70:30 of TRAP/TESP.
Figure 13. ROC curve of the EOADL-PCDC model on 70:30 of TRAP/TESP.
Biomedicines 11 03200 g013
Figure 14. Comparative outcome of the EOADL-PCDC algorithm with recent systems.
Figure 14. Comparative outcome of the EOADL-PCDC algorithm with recent systems.
Biomedicines 11 03200 g014
Table 1. Summary of existing works.
Table 1. Summary of existing works.
ReferenceObjectiveMethodologyDatasetMeasuresMeritsDemerits
Singh et al. [11]Classify PrC which belongs to the Gleason grade group Faster RCNN with Inception-Resnet-V2Prostate-2 datasetAccuracy, sensitivity, specificityEffective segmentation of lesionLess experimentation
Yildirim et al. [12]Aimed to detect the PI-RADS groups using mpMR images.MobilenetV2, Efficientnetb0, and Darknet53Own dataAccuracyCan be used to reduce unnecessary biopsiesmoderate to lower level agreement in PI-RADS scoring evaluation
Li et al. [13]Classify PCa and prostate hyperplasiaSwin TransformerOwn dataAUC, ROCBetter predictive outcomesHard to detect the accurate area on MRI images
Lai et al. [14]For auto-segmenting the prostate zone and cancer regionSegNet, DCNNOwn dataAccuracy, DSC, RecallSuperior results for PCa auto segmentationLess amount of training data and requires model fine-tuning
Ye et al. [15]Develop a prostate tumor diagnosis modelPSP-Net+VGG16Own dataAccuracySuperior in accuracy and processing timeLess experimentation
Ragab et al. [16]Investigate MRI images for prostate cancer detectionDenseNet-161, LS-SVM, AOAOwn dataAccuracy, sensitivity, specificity, F-ScoreEnhanced performanceComputational complexity analysis is needed
Moroianu et al. [17]Identify PCa that spreads outside the prostateU-NetOwn dataROC, AUCParameter tuning is accomplishedComputational complexity analysis is needed
Bouslimi, and Echi [18]Offer a fully automatic system for prostate detection MultiResUnet Radboudumc prostate cancer datasetAccuracyEnhanced performanceLess experimentation
Hassan et al. [19]Detect prostate cancer using a fusion of different DL modelsSVM, Adaboost, K-NN, and Random ForestsOwn dataAccuracyExamined by XAIRequires fine-tuning of model parameters
Abbasi et al. [20]Employ transfer learning model for prostate cancer detectionGoogleNet+ML classifiersHarvard University prostate datasetSensitivity, specificity, PPV, NPV, and total accuracy Enhanced performanceRequires fine-tuning of model parameters
Gavade et al. [21]Classify prostate cancerU-Net architecture+LSTMI2CVB datasetAccuracy, F1 score, precision, recall, ROC, dice Reduce bias and enhance the generalization ability Less experimentation
Table 2. Details of the database.
Table 2. Details of the database.
ClassesNo. of Instances
Prostate200
Brachytherapy200
Total Samples400
Table 3. Prostate cancer detection outcomes of the EOADL-PCDC algorithm on 80:20 of TRAP/TESP.
Table 3. Prostate cancer detection outcomes of the EOADL-PCDC algorithm on 80:20 of TRAP/TESP.
Class A c c u y R e c a l S p e c y F s c o r e MCC
TRAP (80%)
Prostate99.6999.39100.0099.6999.38
Brachytherapy99.69100.0099.3999.6899.38
Average99.6999.7099.7099.6999.38
TESP (20%)
Prostate98.7597.22100.0098.5997.50
Brachytherapy98.75100.0097.2298.8897.50
Average98.7598.6198.6198.7397.50
Table 4. Prostate cancer detection outcome of the EOADL-PCDC model at 70:30 of TRAP/TESP.
Table 4. Prostate cancer detection outcome of the EOADL-PCDC model at 70:30 of TRAP/TESP.
Class A c c u y R e c a l S p e c y F s c o r e MCC
TRAP (70%)
Prostate99.2999.2899.3099.2898.57
Brachytherapy99.2999.3099.2899.3098.57
Average99.2999.2999.2999.2998.57
TESP (30%)
Prostate99.17100.0098.2899.2098.34
Brachytherapy99.1798.28100.0099.1398.34
Average99.1799.1499.1499.1798.34
Table 5. Comparative outcome of the EOADL-PCDC method with existing techniques [16].
Table 5. Comparative outcome of the EOADL-PCDC method with existing techniques [16].
Methods A c c u y s e n s y S p e c y F s c o r e
EOADL-PCDC99.6999.7099.7099.69
AOADLB-P2C99.5099.5099.5099.50
NB98.4698.4798.6498.81
DT 97.2997.2698.4798.83
SVM-Gaussian 98.3698.4398.5497.91
SVM-RBF 98.1298.6397.8998.52
GoogleNet98.2898.2898.4998.69
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

Yang, E.; Shankar, K.; Kumar, S.; Seo, C.; Moon, I. Equilibrium Optimization Algorithm with Deep Learning Enabled Prostate Cancer Detection on MRI Images. Biomedicines 2023, 11, 3200. https://doi.org/10.3390/biomedicines11123200

AMA Style

Yang E, Shankar K, Kumar S, Seo C, Moon I. Equilibrium Optimization Algorithm with Deep Learning Enabled Prostate Cancer Detection on MRI Images. Biomedicines. 2023; 11(12):3200. https://doi.org/10.3390/biomedicines11123200

Chicago/Turabian Style

Yang, Eunmok, K. Shankar, Sachin Kumar, Changho Seo, and Inkyu Moon. 2023. "Equilibrium Optimization Algorithm with Deep Learning Enabled Prostate Cancer Detection on MRI Images" Biomedicines 11, no. 12: 3200. https://doi.org/10.3390/biomedicines11123200

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop