DRBoost: A Learning-Based Method for Steel Quality Prediction

Steel products play an important role in daily production and life as a common production material. Currently, the quality of steel products is judged by manual experience. However, various inspection criteria employed by human operators and complex factors and mechanisms in the steelmaking process may lead to inaccuracies. To address these issues, we propose a learning-based method for steel quality prediction, which is named DRBoost,based on multiple machine learning techniques, including Decision tree, Random forest, and the LSBoost algorithm. In our method, the decision tree clearly captures the nonlinear relationships between features and serves as a solid baseline for making preliminary predictions. Random forest enhances the model’s robustness and avoids overfitting by aggregating multiple decision trees. LSBoost uses gradient descent training to assign contribution coefficients to different kinds of raw materials to obtain more accurate predictions. Five key chemical elements, including carbon, silicon, manganese, phosphorus, and sulfur, which significantly influence the major performance characteristics of steel products, are selected. Steel quality prediction is conducted by predicting the contents of these chemical elements. Multiple models are constructed to predict the contents of five key chemical elements in steel products. These models are symmetrically complementary, meeting the requirements of different production scenarios and forming a more accurate and universal method for predicting the steel product’s quality. In addition, the prediction method provides a symmetric quality control system for steel product production. Experimental evaluations are conducted based on a dataset of 2012 samples from a steel plant in Liaoning Province, China. The input variables include various raw material usages, while the outputs are the content of five key chemical elements that influence the quality of steel products. The experimental results show that the models demonstrate their advantages in different performance metrics and are applicable to practical steelmaking scenarios.