Skip to main content
Log in

Optimizing the neural network hyperparameters utilizing genetic algorithm

利用遗传算法优化神经网络超参数

  • Article
  • Published:
Journal of Zhejiang University-SCIENCE A Aims and scope Submit manuscript

Abstract

Neural networks (NNs), as one of the most robust and efficient machine learning methods, have been commonly used in solving several problems. However, choosing proper hyperparameters (e.g. the numbers of layers and neurons in each layer) has a significant influence on the accuracy of these methods. Therefore, a considerable number of studies have been carried out to optimize the NN hyperparameters. In this study, the genetic algorithm is applied to NN to find the optimal hyperparameters. Thus, the deep energy method, which contains a deep neural network, is applied first on a Timoshenko beam and a plate with a hole. Subsequently, the numbers of hidden layers, integration points, and neurons in each layer are optimized to reach the highest accuracy to predict the stress distribution through these structures. Thus, applying the proper optimization method on NN leads to significant increase in the NN prediction accuracy after conducting the optimization in various examples.

Abstract

目的

证明超参数优化对深度能量方法 (DEM) 精度的影响以及DEM在预测不同荷载作用下梁和板等结构的应力分布方面的能力。

创新点

1. 为了提高DEM的准确性, 各种超参数组合被输入遗传算法 (GA) 并找到最佳组合。2. 为了防止重复计算以及提高这种元启发式算法的效率, GA过程中还考虑了超参数组合的禁忌列表。

方法

1. 实施非均匀有理样条 (NURBS) 以生成穿过结构体和边界的积分点。2. 采用DEM计算位移和应力分布。3. 利用遗传算法优化DEM的超参数, 以对模型在预测结构内应力和位移传播的准确性方面具有显着影响。

结论

1. 在不同的优化器和激活函数中, Adam和L-BFGS-B方法以及ReLU2函数的组合使得DEM模型的准确率最高。2. 其他对模型预测准确性有影响的超参数包括隐藏层的数量, 每层神经元的数量以及通过上述结构集成的点数。3. 优化DEM的超参数可以使相对应变能误差降低近50%, 提高了DEM模型对应力和位移分布的预测能力。

This is a preview of subscription content, log in via an institution to check access.

Access this article

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Similar content being viewed by others

References

Download references

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Timon Rabczuk.

Additional information

Contributors

Timon RABCZUK devised the project, verified the computational process, and contributed to the final version of the manuscript. Mohammad SALAVATI contributed to the simulation process. Arvin MOJAHEDIN worked out the technical details, performed the computational calculations, and wrote the manuscript.

Conflict of interest

Arvin MOJAHEDIN, Mohammad SALAVATI, and Timon RABCZUK declare that they have no conflict of interest.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Nikbakht, S., Anitescu, C. & Rabczuk, T. Optimizing the neural network hyperparameters utilizing genetic algorithm. J. Zhejiang Univ. Sci. A 22, 407–426 (2021). https://doi.org/10.1631/jzus.A2000384

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1631/jzus.A2000384

Key words

关键词

CLC number

Navigation