An optimized machine-learning model for mechanical properties prediction and domain knowledge clarification in quenched and tempered steels

Abstract

Clarifying the relationship between compositions, heat treatment processes, and mechanical properties of carbon steel, as the basis of material design, is challengeable, while machine learning (ML) makes this complex correlation explicit. In this work, three different mechanical properties (ultimate tensile strength, yield strength, and total elongation) were predicted based on the collected quenched and tempered (Q&T) steel dataset by six ML algorithms, in which the optimal Gaussian process regression (GPR) combined with the key descriptors by feature engineering to train an optimized ML model. Such a simplified ML model shows even better prediction accuracy. In the above training process, Bayesian optimization (BO) searches the hyperparameters efficiently. The newly collected data also achieve small prediction errors, showing good generalization capacity. To maximize the application value of the current ML model, the grid prediction of composition and process, and local interpretable model-agnostic explanations (LIME) were utilized to reveal some new insights about the quenched and tempered steels, which could shed light on the ongoing new material design. Besides, the overfitting tendency of the ML model was examined to ensure the rationality of prediction, and the influence of data amount on the prediction performance was discussed.