<p>In response to the challenges of inadequate predictive accuracy and limited generalization capability in data-driven modeling for the mechanical properties of the cold-rolled strip steel, a predictive modeling method named RFR-WOA is developed based on random forest regression (RFR) and whale optimization algorithm (WOA). Firstly, using Pearson and Spearman correlation analysis and Gini coefficient importance ranking on an actual production dataset containing 37,878 samples, 22 key variables are selected as model inputs from 112 variables that affect mechanical properties. Subsequently, an RFR-based predictive model for the mechanical properties of cold-rolled strip steel is constructed. Then, with the combination of the coefficient of determination (<i>R</i><sup>2</sup>) and root mean square error as the optimization objective, the hyperparameters of RFR model are iteratively optimized using WOA, and better predictive effectiveness is obtained. Finally, the mechanical properties prediction model based on RFR-WOA is compared with models established using deep neural networks, convolutional neural networks, and other methods. The test results on 9469 samples of actual production data show that the model developed present has better predictive accuracy and generalization capability.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Predictive modeling for mechanical properties of cold-rolled strip steel based on random forest regression and whale optimization algorithm

  • Hong-Lei Cai,
  • Yi-Ming Fang,
  • Le Liu,
  • Li-Hui Ren,
  • Zhen-Dong Liu,
  • Xiao-Dong Zhao

摘要

In response to the challenges of inadequate predictive accuracy and limited generalization capability in data-driven modeling for the mechanical properties of the cold-rolled strip steel, a predictive modeling method named RFR-WOA is developed based on random forest regression (RFR) and whale optimization algorithm (WOA). Firstly, using Pearson and Spearman correlation analysis and Gini coefficient importance ranking on an actual production dataset containing 37,878 samples, 22 key variables are selected as model inputs from 112 variables that affect mechanical properties. Subsequently, an RFR-based predictive model for the mechanical properties of cold-rolled strip steel is constructed. Then, with the combination of the coefficient of determination (R2) and root mean square error as the optimization objective, the hyperparameters of RFR model are iteratively optimized using WOA, and better predictive effectiveness is obtained. Finally, the mechanical properties prediction model based on RFR-WOA is compared with models established using deep neural networks, convolutional neural networks, and other methods. The test results on 9469 samples of actual production data show that the model developed present has better predictive accuracy and generalization capability.