Elevating Prediction Performance for Mechanical Properties of Hot-Rolled Strips by Using Semi-Supervised Regression and Deep Learning

Abstract

In the present work, to solve the problem of the lacking enough labeled training data for deep learning, a safe semi-supervised regression supporting Bayesian optimization deep neural network (SAFER-BODNN) model was proposed to establish mechanical property prediction model of hot-rolled strips. The Pearson correlation coefficient was applied to reduce the data dimension. The safe semi-supervised regression was implemented to add the pseudo labels to the unlabeled data for training dataset expansion. The deep neural network was trained with Bayesian optimization to determine the optimal hyper-parameters of the network. The results show that the SAFER-BODNN model achieves good performance for mechanical property prediction of hot-rolled strips with correlation coefficient of 0.9610 for yield strength, 0.9682 for tensile strength, and 0.8619 for elongation, respectively. Compared with the deep neural network trained on the labeled dataset, the SAFER-BODNN model obtains stable smaller predicted errors. Among all the variables, C content and Mn content have large influence on the yield strength and tensile strength, coiling temperature has the largest influence on the elongation. The investigation makes full use of unlabeled data to elevate the prediction performance of the deep neural network, and also provides a way for deep learning modeling when the data are insufficient.