Physically-and Knowledge-Informed Deep Learning for Robust Prediction of Martensite Start Temperature in Steels

Pengxin Wang and G. M. A. M. El-Fallah

Abstract

Accurate prediction of the martensite start temperature (Mₛ) is essential for optimising the mechanical performance of steels and enabling data-driven alloy design. This study proposes, for the first time, a physics- and knowledge-informed multi-branch Transformer fusion model that integrates chemical composition, physical features, and empirical equation-based features for precise Mₛ prediction. A curated dataset of 1,100 steel samples was compiled from peer-reviewed studies and technical reports published between 1973 and 2023, with a primary focus on structural steels, encompassing a comprehensive range of compositional and physical variables relevant to phase transformation behaviour. The model architecture employs dedicated multilayer perceptron (MLP) branches for each feature group, followed by a Transformer encoder to capture global dependencies and nonlinear interactions. Compared with baseline models, including conventional MLPs and lasso regression, the proposed model achieves superior predictive performance (RMSE = 33.9 °C, MAE = 14 °C, R² = 0.94). SHAP-based interpretability analysis reveals that the model identifies C, Ni, and Ceq as key contributors, forming a dominant Ni–Ceq–C interaction network modulated by structural descriptors. Sensitivity analyses confirm these trends and support the model’s physical consistency. To experimentally validate the model, dilatometry tests on two newly designed steels confirmed that the predicted Mₛ values (231 °C and 165 °C) closely matched the experimental results (238 °C and 169 °C), with deviations of only 7 °C and 4 °C, respectively. This work establishes a novel, physically grounded, and interpretable machine learning framework for Mₛ prediction, highlighting the value of integrating physics-based knowledge with advanced deep learning to accelerate alloy design and mechanical property optimisation.

Published in Materials Today Communications, Vol. 49 (2025), Article 113743.

Martensite Start Temperature \(M_s\) Predictor

Empirical Equation:

$$ \begin{aligned} M_s =\ & 419.5 - 195.5C - 0.27Cr - Co - 29.66C \cdot Mn - 13.4C \cdot Ni - 2.99Mn \cdot Cr \\ & - 1.29Si^2 - 1.5Cr \cdot Ni - 0.5Ni^2 + 7.25Mo \cdot Cu + 3.3Mo \cdot Nb \\ & + 53.5V \cdot Cu + 2592.5Al \cdot Cu + 90W \cdot Nb + 11.2W \cdot Ti + 118.7Nb \cdot Ti \end{aligned} $$