| Literature DB >> 35310591 |
Abstract
Neural network algorithms and intelligent algorithms are hot topics in the field of deep learning. In this study, the neural network algorithm and intelligence are optimized, and it is used in simulation experiments to improve the target image recognition ability of the algorithm in the machine vision environment. First, this paper introduces the application of neural networks in the field of machine vision. Second, in the experiment, the improved VGG-16 convolutional neural network (CNN) model is applied to metal block defect detection. Experimental results show that the optimized network can classify metal block defects with the maximum accuracy of 99.28%. Then, the intelligent algorithm based on neural network is studied, and the CIFAR-10 data set is taken as the experimental target for training test and verification test. Using BP algorithm, particle swarm optimization algorithm (PSO-BP), and improved neural network algorithm, respectively, the convergence speed of ICS algorithm based on BP neural network is compared. In contrast, ICS-BP algorithm has the fastest convergence speed and converges when the number of iterations is 32, followed by PSO-BP algorithm.Entities:
Mesh:
Year: 2022 PMID: 35310591 PMCID: PMC8926490 DOI: 10.1155/2022/6154453
Source DB: PubMed Journal: Comput Intell Neurosci
Figure 1Neuron structure.
Figure 2BP neural network structure.
Figure 3CNN structure.
Figure 4Inner product extraction of image features.
Figure 5ReLU function curve.
Figure 6Sigmoid output function.
Figure 7Image processing CNN structure.
Error rate indicators of each model using the ImageNet data set.
| Model | Top-1 validation set | Top-5 validation set | Top-5 test set |
|---|---|---|---|
| VGG | 23.7 | 6.8 | 6.8 |
| OverFeat | 34.0 | 13.2 | 13.6 |
| AlexNet | 38.1 | 16.4 | 16.4 |
| MSRA | 27.9 | 9.1 | 9.1 |
| GoogLeNet | - | 6.7 | 6.7 |
VGG-16 convolutional structure parameters.
| Layer (Type) | Output shape | Parameters |
|---|---|---|
| Input | 224 | 0 |
| Conv1 | 224 | 1728 |
| Conv2 | 224 | 36864 |
| Pool | 112 | 0 |
| Conv1 | 112 | 73728 |
| Conv2 | 112 | 147456 |
| Pool | 56 | 0 |
| Conv1 | 56 | 294912 |
| Conv2 | 56 | 589824 |
| Conv3 | 56 | 589824 |
| Pool | 28 | 0 |
| Conv1 | 28 | 1179648 |
| Conv2 | 28 | 2359296 |
| Conv3 | 28 | 2359296 |
| Pool | 14 | 0 |
| Conv1 | 14 | 2359296 |
| Conv2 | 14 | 2359296 |
| Conv3 | 14 | 2359296 |
| Pool | 7 | 0 |
| FC | 1 | 102760448 |
| FC | 1 | 16777216 |
| FC | 1 | 4096000 |
| Total parameters | 138M parameters | |
Figure 8Image prediction classification results.
Figure 9Experimental results.
Comparison of test results.
| Type | Four categories (%) | Accuracy (%) | Recall (%) |
|---|---|---|---|
| VGG-16 | 74.76 | 98.15 | 95.03 |
| AlexNet | 71.24 | 97.92 | 68.77 |
| GoogLeNet | 92.33 | 99.16 | 99.26 |
| Optimized VGG-16 | 79.73 | 99.21 | 99.17 |
| Optimized AlexNet | 68.85 | 98.58 | 98.06 |
| Optimized GoogLeNet | 84.46 | 99.60 | 99.49 |
Test functions.
| Sphere | Levy | |
|---|---|---|
| Search area | [−25,25] | [−10,10] |
| Optimal location |
|
|
| Optimal value |
|
|
Figure 10Optimal value change graph.
Figure 11Neural network image classification training convergence graph.