| Literature DB >> 36185439 |
Yongtao Lu1,2, Yi Huo1, Zhuoyue Yang3, Yibiao Niu1, Ming Zhao1, Sergei Bosiakov4, Lei Li5.
Abstract
In recent years, the convolutional neural network (CNN) technique has emerged as an efficient new method for designing porous structure, but a CNN model generally contains a large number of parameters, each of which could influence the predictive ability of the CNN model. Furthermore, there is no consensus on the setting of each parameter in the CNN model. Therefore, the present study aimed to investigate the sensitivity of the parameters in the CNN model for the prediction of the mechanical property of porous structures. 10,500 samples of porous structure were randomly generated, and their effective compressive moduli obtained from finite element analysis were used as the ground truths to construct and train a CNN model. 8,000 of the samples were used to train the CNN model, 2000 samples were used for the cross-validation of the CNN model and the remaining 500 new structures, which did not participate in the CNN training process, were used to test the predictive power of the CNN model. The sensitivity of the number of convolutional layers, the number of convolution kernels, the number of pooling layers, the number of fully connected layers and the optimizer in the CNN model were then investigated. The results showed that the optimizer has the largest influence on the training speed, while the fully connected layer has the least impact on the training speed. Additionally, the pooling layer has the largest impact on the predictive ability while the optimizer has the least impact on the predictive ability. In conclusion, the parameters of the CNN model play an important role in the performance of the CNN model and the parameter sensitivity analysis can help optimize the CNN model to increase the computational efficiency.Entities:
Keywords: bone scaffold; compressive modulus; convolutional neural network; finite element modeling; sensitivity analysis
Year: 2022 PMID: 36185439 PMCID: PMC9520359 DOI: 10.3389/fbioe.2022.985688
Source DB: PubMed Journal: Front Bioeng Biotechnol ISSN: 2296-4185
FIGURE 1Schematic diagram of porous composite structure (The shaded areas are the pores).
FIGURE 2Loading and constraint defined in the finite element analysis (“ε” refers to the strain).
FIGURE 3The workflow for the (A) training and (B) cross-validation of the convolutional neural network (CNN) model.
FIGURE 4The convolutional neural network model constructed in the present study.
FIGURE 5The relationship between the mean absolute error and the Epoch.
FIGURE 6The relationship between the mean absolute error and the epoch. (A) Sensitivity of the convolution layer; (B) Sensitivity of the pooling layer; (C) Sensitivity of the fully connected layer and (D) Sensitivity of the optimizer (‘Best’ refers to the case with the fastest iterative convergence speed and the corresponding best parameters are listed in Table 1).
The parameters used in the fastest convergent scenario.
| Number of convolutional layers | Number of convolutional kernels | Number of pooling layers | Number of convolutional layers | Optimizer | |
|---|---|---|---|---|---|
| (a) | 2 | 2, 4 | 2 | 3 | RMSprop |
| (b) | 2 | 2, 4 | 1 | 3 | RMSprop |
| (c) | 2 | 2, 4 | 1 | 2 | RMSprop |
| (d) | 2 | 2, 4 | 1 | 2 | Adam |
FIGURE 7The relationship between the cumulative percentile and the relative prediction error from the convolutional neural network model. (A) Sensitivity of the convolution layer; (B) Sensitivity of the pooling layer; (C) Sensitivity of the fully connected layer and (D) Sensitivity of the optimizer (‘Best’ refers to the case with the lowest 95th percentile of relative prediction error and the corresponding best parameters are listed in Table 2).
The parameters with the lowest 95th percentile of relative prediction error.
| Number of convolution layers | Number of convolution kernels | Number of pooling layers | Number of convolution layers | Optimizer | |
|---|---|---|---|---|---|
| (a) | 4 | 2, 4, 6, 8 | 2 | 3 | RMSprop |
| (b) | 4 | 2, 4, 6, 8 | 0 | 3 | RMSprop |
| (c) | 4 | 2, 4, 6, 8 | 0 | 2 | RMSprop |
| (d) | 4 | 2, 4, 6, 8 | 0 | 2 | Adam |