| Literature DB >> 36203733 |
Yanfeng Wang1, Qing Liu1, Junwei Sun1, Lidong Wang2, Xin Song2, Xueke Zhao2.
Abstract
Deep neural network is a complex pattern recognition network system. It is widely favored by scholars for its strong nonlinear fitting ability. However, training deep neural network models on small datasets typically realizes worse performance than shallow neural network. In this study, a strategy to improve the sparrow search algorithm based on the iterative map, iterative perturbation, and Gaussian mutation is developed. This optimized strategy improved the sparrow search algorithm validated by fourteen benchmark functions, and the algorithm has the best search accuracy and the fastest convergence speed. An algorithm based on the iterative map, iterative perturbation, and Gaussian mutation improved sparrow search algorithm is designed to optimize deep neural networks. The modified sparrow algorithm is exploited to search for the optimal connection weights of deep neural network. This algorithm is implemented for the esophageal cancer dataset along with the other six algorithms. The proposed model is able to achieve 0.92 under all the eight scoring criteria, which is better than the performance of the other six algorithms. Therefore, an optimized deep neural network based on an improved sparrow search algorithm with iterative map, iterative perturbation, and Gaussian mutation is an effective approach to predict the survival rate of esophageal cancer.Entities:
Mesh:
Year: 2022 PMID: 36203733 PMCID: PMC9532078 DOI: 10.1155/2022/1036913
Source DB: PubMed Journal: Comput Intell Neurosci
Fourteen benchmark functions.
| Function name | Function | Dimension | Search space | Optimal value |
|---|---|---|---|---|
| Sphere |
| 30 | [−100, 100] | 0 |
| Schwefel 2.21 |
| 30 | [−100, 100] | 0 |
| Schwefel 2.22 |
| 30 | [−10, 10] | 0 |
| Rosenbrock |
| 30 | [−30, 30] | 0 |
| Step |
| 30 | [−100, 100] | 0 |
| Quartic |
| 30 | [−1.28, 1.28] | 0 |
| Alpine |
| 30 | [−10, 10] | 0 |
| Rastrigin |
| 30 | [−5.12, 5.12] | 0 |
| Sum squares |
| 30 | [−10, 10] | 0 |
| Ackley |
| 30 | [−100, 100] | 0 |
| Maytas |
| 2 | [−10, 10] | 0 |
| Levi |
| 2 | [−10, 10] | 0 |
| Booth |
| 2 | [−10, 10] | 0 |
| Three-Hump |
| 2 | [−10, 10] | 0 |
Figure 1Solution space diagram for nineteen benchmark functions. (a) Sphere function. (b) Schwefel 2.21 function. (c) Schwefel 2.22 function. (d) Rosenbrock function. (e) Step function. (f) Quartic function. (g) Alpine function. (h) Rastrigin function. (i) Sum squares function. (j) Ackley function. (k) Maytas function. (l) Levi function. (m) Booth function. (n) Three-Hump function.
Examine outcomes for fourteen benchmark functions.
| Function | Algorithms | Optimal value | Mean value | Standard deviation |
|---|---|---|---|---|
|
| SSA |
| 8.037E-114 | 3.594E-113 |
| Tent |
| 5.311E-202 |
| |
| Chebyshev |
| 7.017E-245 |
| |
| Circle |
|
|
| |
| Iterative |
| 4.101E-183 |
| |
| Sine |
| 1.298E-198 |
| |
| Singer |
| 7.994E-212 |
| |
| Sinusoidal |
| 1.202E-160 | 5.376E-160 | |
| Logistic |
| 4.07E-257 |
| |
| Cubic |
| 3.662E-163 | 2.223E-162 | |
|
| ||||
|
| SSA |
| 3.856E-45 | 1.724E-44 |
| Tent |
| 8.548E-155 | 3.823E-154 | |
| Chebyshev |
| 4.031E-130 | 1.803E-129 | |
| Circle |
| 1.481E-127 | 6.621E-127 | |
| Iterative |
|
|
| |
| Sine |
| 1.985E-136 | 8.877E-136 | |
| Singer |
| 9.621E-140 | 5.524E-140 | |
| Sinusoidal |
| 1.4722E-122 | 2.2693E-128 | |
| Logistic |
| 2.154E-133 | 1.312E-132 | |
| Cubic |
| 2.934E-133 | 2.0752E-79 | |
|
| ||||
|
| SSA |
| 1.673E-27 | 7.483E-27 |
| Tent |
| 2.714E-122 | 1.214E-121 | |
| Chebyshev |
| 9.729E-96 | 4.351E-95 | |
| Circle |
| 1.564E-74 | 6.994E-74 | |
| Iterative |
|
|
| |
| Sine |
| 2.99E-109 | 1.337E-108 | |
| Singer |
| 7.224E-105 | 3.231E-104 | |
| Sinusoidal |
| 1.316E-65 | 5.887E-65 | |
| Logistic |
| 1.106E-122 | 4.946E-122 | |
| Cubic |
| 1.165E-117 | 5.209E-117 | |
|
| ||||
|
| SSA | 1.223E-05 | 1.142E-03 | 2.778E-03 |
| Tent | 4.321E-08 | 5.348E-06 | 8.617E-06 | |
| Chebyshev | 5.946E-07 | 7.212E-05 | 8.992E-05 | |
| Circle | 1.033E-05 | 3.929E-04 | 5.017E-04 | |
| Iterative |
|
|
| |
| Sine | 2.714E-07 | 1.764E-05 | 1.828E-05 | |
| Singer | 1.128E-06 | 7.727E-05 | 8.092E-05 | |
| Sinusoidal | 2.735E-04 | 1.373E-05 | 2.526E-04 | |
| Logistic | 1.344E-07 | 1.124E-05 | 1.206E-05 | |
| Cubic | 3.213E-08 | 1.333E-05 | 1.568E-05 | |
|
| ||||
|
| SSA | 3486E-04 | 6.432E-03 | 5177E-03 |
| Tent | 4.018E-07 | 9.14E-05 | 9.994E-05 | |
| Chebyshev | 1.981E-08 | 1.796E-05 | 1.97E-05 | |
| Circle | 4.387E-08 | 8.52E-04 | 9.895E-05 | |
| Iterative |
|
|
| |
| Sine | 1.496E-06 | 9.726E-04 | 9.667E-05 | |
| Singer | 1.618E-06 | 7.313E-04 | 8.238E-05 | |
| Sinusoidal | 3.918E-06 | 1.317E-04 | 1.371E-04 | |
| Logistic | 9.34E-06 | 7.424E-04 | 6.168E-04 | |
| Cubic | 7.205E-06 | 1.032E-04 | 1.236E-04 | |
|
| ||||
|
| SSA | 3.1E-03 | 3.399E-02 | 2.059E-02 |
| Tent | 2.912E-04 | 6.554E-03 | 4.036E-03 | |
| Chebyshev |
| 4.367E-03 | 4.201E-03 | |
| Circle | 8.295E-04 | 8.557E-03 | 6.841E-03 | |
| Iterative | 3.63E-04 |
|
| |
| Sine | 1.218E-03 | 6.723E-03 | 4.694E-03 | |
| Singer | 5.581E-05 | 7.878E-03 | 5.441E-03 | |
| Sinusoidal | 5.971E-04 | 8.378E-03 | 7.758E-03 | |
| Logistic | 2.541E-04 | 7.907E-03 | 6.586E-03 | |
| Cubic | 4.965E-04 | 7.344E-03 | 4.835E-03 | |
|
| ||||
|
| SSA |
| 5.837E-60 | 2.611E-59 |
| Tent |
| 3.32E-119 | 1.485E-118 | |
| Chebyshev |
| 8.955E-90 | 4.005E-89 | |
| Circle |
| 1.03E-140 | 4.608E-140 | |
| Iterative |
| 9.454E-138 | 4.228E-137 | |
| Sine |
|
|
| |
| Singer |
| 1.417E-102 | 6.335E-102 | |
| Sinusoidal |
| 3.156E-99 | 1.412E-98 | |
| Logistic |
| 5.296E-85 | 2.368E-84 | |
| Cubic |
| 3.802E-100 | 1.701E-99 | |
|
| ||||
|
| SSA |
| 2.237E-01 | 1.688E-01 |
| Tent |
| 1.011E-01 | 3.295E-02 | |
| Chebyshev |
|
|
| |
| Circle |
|
| 1.034E-04 | |
| Iterative |
|
| 1.323E-04 | |
| Sine |
|
| 1.323E-04 | |
| Singer |
|
| 1.502E-04 | |
| Sinusoidal |
|
| 1.408E-04 | |
| Logistic |
|
| 1.608E-04 | |
| Cubic |
|
| 1.407E-04 | |
|
| ||||
|
| SSA |
| 6.364E-105 | 3.486E-104 |
| Tent |
| 5.818E-162 | 3.19E-161 | |
| Chebyshev |
| 2.456E-195 |
| |
| Circle |
| 1.067E-198 |
| |
| Iterative |
|
|
| |
| Sine |
| 3.023E-112 | 1.656E-111 | |
| Singer |
| 2.521E-141 | 1.381E-140 | |
| Sinusoidal |
| 5.721E-140 | 3.133E-139 | |
| Logistic |
| 3.719E-203 |
| |
| Cubic |
| 3.758E-173 |
| |
|
| ||||
|
| SSA |
|
|
|
| Tent |
|
|
| |
| Chebyshev |
|
|
| |
| Circle |
|
|
| |
| Iterative |
|
|
| |
| Sine |
|
|
| |
| Singer |
|
|
| |
| Sinusoidal |
|
|
| |
| Logistic |
|
|
| |
| Cubic |
|
|
| |
|
| ||||
|
| SSA |
| 4.594E-182 |
|
| Tent |
| 1.434E-245 |
| |
| Chebyshev |
|
|
| |
| Circle |
| 5.773E-238 |
| |
| Iterative |
| 6.493E-226 |
| |
| Sine |
| 1.862E-243 |
| |
| Singer |
| 4.438E-315 |
| |
| Sinusoidal |
| 2.401E-145 | 1.074E-144 | |
| Logistic |
| 2.014E-282 |
| |
| Cubic |
| 1.378E-190 |
| |
|
| ||||
|
| SSA | 2.324E-06 | 2.009E-05 | 9.768E-05 |
| Tent | 1.093E-07 | 1.083E-05 | 1.845E-05 | |
| Chebyshev | 4.92E-07 | 5.578E-06 | 6.295E-06 | |
| Circle | 9.458E-08 | 6.071E-06 | 6.68E-06 | |
| Iterative | 1.19E-08 |
|
| |
| Sine |
| 5.002E-06 | 4.757E-06 | |
| Singer | 2.386E-08 | 5.343E-06 | 5.608E-06 | |
| Sinusoidal | 4.846E-08 | 6.556E-06 | 5.957E-06 | |
| Logistic | 1.144E-07 | 6.205E-06 | 5.926E-06 | |
| Cubic | 2.044E-07 | 6.126E-06 | 8.436E-06 | |
|
| SSA | 2.647E-06 | 2.611E-05 | 2.979E-05 |
| Tent | 9.904E-07 | 7.571E-04 | 8.334E-04 | |
| Chebyshev | 8.105E-07 | 1.441E-05 | 1.482E-05 | |
| Circle | 5.826E-07 | 1.133E-05 | 1.619E-05 | |
| Iterative | 5.674E-07 |
|
| |
| Sine |
| 1.272E-05 | 1.293E-05 | |
| Singer | 1.144E-07 | 1.387E-05 | 1.481E-05 | |
| Sinusoidal | 2.314E-08 | 1.504E-05 | 2.021E-05 | |
| Logistic | 3.14E-08 | 1.363E-05 | 1.683E-05 | |
| Cubic | 2.296E-07 | 1.36E-05 | 1.427E-05 | |
|
| ||||
|
| SSA |
| 4.752E-133 | 2.125E-132 |
| Tent |
| 5.872E-268 |
| |
| Chebyshev |
| 7.727E-268 |
| |
| Circle |
| 2.862E-152 | 1.28E-151 | |
| Iterative |
|
|
| |
| Sine |
| 2.875E-311 |
| |
| Singer |
| 2.985E-122 | 1.335E-121 | |
| Sinusoidal |
| 1.003E-159 | 4.487E-159 | |
| Logistic |
| 9.108E-224 |
| |
| Cubic |
| 1.866E-260 |
| |
Figure 2Comparison of convergence curves of 10 algorithms on benchmark functions. (a) Sphere function. (b) Schwefel 2.21 function. (c) Schwefel 2.22 function. (d) Rosenbrock function. (e) Step function. (f) Quartic function. (g) Alpine function. (h) Rastrigin function. (i) Sum squares function. (j) Ackley function. (k) Maytas function. (l) Levi function. (m) Booth function. (n) Three-Hump function.
Figure 3DNN architecture.
Figure 4Training phase of the proposed IGSSA for DNN.
Continuous metrics in the dataset.
| Variable | Mean | Median (range) | Variance |
|---|---|---|---|
| Tumor length | 4.112 | 4 (1–11) | 3.208 |
| Tumor width | 2.649 | 2.5 (0.3–9) | 1.148 |
| Tumor thickness | 1.1776 | 1 (0.1–8) | 0.471 |
| WBC | 6.5366 | 6.2 (2.5–13.6) | 3.6958 |
| LY | 1.7622 | 1.8 (0–4) | 0.3652 |
| MONO | 0.3899 | 0.4 (0–1.4) | 0.06661 |
| NEUT | 4.0011 | 3.7 (0–9.8) | 2.8097 |
| EOS | 0.1238 | 0.1 (0–0.9) | 0.0198 |
| BASO | 0.04163 | 0 (0–5) | 0.005549 |
| RBC | 4.43 | 4.48 (2.73–5.75) | 0.2289 |
| HB | 137.4347 | 138 (64–169) | 223.7577 |
| PLT | 236.8518 | 231 (100–448) | 52.606 |
| TP | 71.0377 | 71 (50–92) | 54.4092 |
| ALB | 42.0201 | 42 (26–59) | 25.1281 |
| GLB | 29.1533 | 29 (16–45) | 28.8656 |
| PT | 10.2271 | 10.2 (7–16.6) | 2.4610 |
| APTT | 35.9095 | 35.1 (15.4–62.2) | 52.9934 |
| TT | 15.3420 | 15.5 (10.9–21.3) | 2.9607 |
| FIB | 387.3433 | 378.3960 (167.613–774.433) | 985.7021 |
| Age | 60 | 60 (38–82) | 70.099 |
| Survival time | 4 | 3 (0–11) | 12.873 |
The unit of tumor length, tumor width, tumor thickness is centimeter. The unit of WBC, LY, MONO, NEUT, EOS, BASO, RBC, and PLT is 109/L. The unit of HB, TP, ALB, and GLB is g/L. The unit of PT, APTT, and TT is second(s). The unit of FIB is mg/dL. The unit of survival time is year.
Discrete metrics in the dataset.
| Project | Category | Number of population | Percentage of population (%) |
|---|---|---|---|
| Gender | Male | 247 | 62 |
| Female | 151 | 38 | |
|
| |||
| Degree of differentiation | Poorly differentiated | 158 | 40 |
| Moderately differentiated | 217 | 54 | |
| Highly differentiated | 23 | 6 | |
|
| |||
| Tumor site | Lower thoracic | 78 | 20 |
| Mid thoracic | 267 | 67 | |
| Upper thoracic | 53 | 13 | |
|
| |||
| Transfer situation | Negative | 200 | 50 |
| Positive | 198 | 50 | |
|
| |||
| TNM stages | I | 39 | 10 |
| II | 172 | 43 | |
| III | 166 | 42 | |
| IV | 21 | 5 | |
|
| |||
| Survival status | Live | 101 | 25 |
| Dead | 297 | 75 | |
Figure 5The overall flow chart of IGSSA-DNN.
Results of the predictive model evaluation.
| Algorithms | Acc | FPR | REC | PRE | TNR | F1-M | AUC |
|
|---|---|---|---|---|---|---|---|---|
| DNN | 0.76 | 0.16 | 0.76 | 0.76 | 0.76 | 0.76 | 0.76 | 0.03 |
| PSO-DNN | 0.79 | 0.133 | 0.79 | 0.79 | 0.79 | 0.79 | 0.79 | 0.04 |
| GSA-DNN | 0.83 | 0.133 | 0.83 | 0.83 | 0.83 | 0.83 | 0.83 | 0.04 |
| GWO-DNN | 0.81 | 0.117 | 0.81 | 0.81 | 0.81 | 0.81 | 0.81 | 0.01 |
| WOA-DNN | 0.86 | 0.140 | 0.86 | 0.86 | 0.86 | 0.86 | 0.86 | 0.01 |
| SSA-DNN | 0.89 | 0.147 | 0.89 | 0.89 | 0.89 | 0.89 | 0.89 | 0.04 |
| IIGSSA-KNN | 0.75 | 0.21 | 0.75 | 0.75 | 0.75 | 0.75 | 0.75 | 0.04 |
| IIGSSA-SVM | 0.82 | 0.18 | 0.82 | 0.82 | 0.82 | 0.82 | 0.82 | 0.04 |
| IIGSSA-DNN | 0.92 | 0.100 | 0.92 | 0.92 | 0.92 | 0.92 | 0.92 | 0.01 |
Figure 6Five ROC curves of nine models. (a) Five ROC curves of DNN. (b) Five ROC curves of PSO-DNN. (c) Five ROC curves of GSA-DNN. (d) Five ROC curves of GWO-DNN. (e) Five ROC curves of WOA-DNN. (f) Five ROC curves of SSA-DNN. (g) Five ROC curves of IGSSA-KNN. (h) Five ROC curves of IGSSA-SVM. (i) Five ROC curves of IGSSA-DNN.
Figure 7Test accuracy of nine models.