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

Autor: Siwei Wu, Jian Yang, Guangming Cao, Yunlong Qiu, Guoguang Cheng, Meiyi Yao, Jianxin Dong
Jazyk: angličtina
Rok vydání: 2020
Předmět:
Zdroj: IEEE Access, Vol 8, Pp 134124-134136 (2020)
Druh dokumentu: article
ISSN: 2169-3536
DOI: 10.1109/ACCESS.2020.3010506
Popis: 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.
Databáze: Directory of Open Access Journals