Prediction Model of Mechanical Properties of Hot-Rolled Strip Based on Improved Feature Selection Method

Zhi-wei Gao,Guang-ming Cao,Si-wei Wu,Deng Luo,Hou-xin Wang,Zhen-yu Liu
DOI: https://doi.org/10.1007/s42243-024-01254-x
2024-01-01
Journal of Iron and Steel Research International
Abstract:Selecting proper descriptors (also known feature selection, FS) is key in the process of establishing mechanical properties prediction model of hot-rolled microalloyed steels by using machine learning (ML) algorithm. FS methods based on data-driving can reduce the redundancy of data features and improve the prediction accuracy of mechanical properties. Based on the collected data of hot-rolled microalloyed steels, the association rules are used to mine the correlation information between the data. High-quality feature subsets are selected by the proposed FS method (FS method based on genetic algorithm embedding, GAMIC). Compared with the common FS method, it is shown on dataset that GAMIC selects feature subsets more appropriately. Six different ML algorithms are trained and tested for mechanical properties prediction. The result shows that the root-mean-square error of yield strength, tensile strength and elongation based on limit gradient enhancement (XGBoost) algorithm is 21.95 MPa, 20.85 MPa and 1.96
What problem does this paper attempt to address?