Classified dataset, regression and machine learning modeling for prediction of phase transformation temperatures in steels

Abstract

The prediction of the characteristic Martensite Start (M_s) temperature and Austenitic Nose Tip Temperature (ANTT) in steels is of scientific and technological importance; however, it faces significant challenges due to multiphysical complexity.

In this study, we introduced a structured framework for data classification and hierarchical iterations aimed at predicting Ms (Martensite start temperature) and ANTT (Austenite non-transforming temperature). This framework was incorporated into two optimization models, leading to enhancements in accuracy, extrapolation capabilities, and generalization performance. First, we classified the collected Ms datasets hierarchically based on the alloying elements presented in steels, including carbon, austenite stabilizers, non-austenitization elements, and data credibility. Regression analyses of Ms temperatures concerning chemical compositions were then carried out using phenomenological variables from binary systems to multi-component systems in alignment with the spirit of CALPHAD modeling, which is renowned for its robust extrapolation abilities. By iteratively fitting the hierarchically classified datasets and implementing hierarchical iterations, we developed the CALPHAD-guided phenomenological variable (CGPV) Ms regression model. This model achieved improved accuracy levels, with R² values of 0.9 for training and 0.87 for testing, surpassing most conventional regression models that do not account for compositional interactions. Furthermore, the CALPHAD-guided machine learning (CGML) model, constructed based on the classified datasets and hierarchical iterations but without utilizing phenomenological variables, demonstrated strong performance with R² values of 0.98 and 0.86 for training and testing, respectively. The CGML model was demonstrated not only to reliably filter out problematic data in a dataset but also to unveil the unnoticed coupling between carbon and other alloying elements on M_s. Finally, the CGML method has been readily transferred to predict ANTT with high accuracy as well.