Prediction model for corrosion rate of low-alloy steels under atmospheric conditions using machine learning algorithms

This work constructed a machine learning (ML) model to predict the atmospheric corrosion rate of low-alloy steels (LAS). The material properties of LAS, environmental factors, and exposure time were used as the input, while the corrosion rate as the output. 6 different ML algorithms were used to construct the proposed model. Through optimization and filtering, the eXtreme gradient boosting (XGBoost) model exhibited good corrosion rate prediction accuracy. The features of material properties were then transformed into atomic and physical features using the proposed property transformation approach, and the dominant descriptors that affected the corrosion rate were filtered using the recursive feature elimination (RFE) as well as XGBoost methods. The established ML models exhibited better prediction performance and generalization ability via property transformation descriptors. In addition, the SHapley additive exPlanations (SHAP) method was applied to analyze the relationship between the descriptors and corrosion rate. The results showed that the property transformation model could effectively help with analyzing the corrosion behavior, thereby significantly improving the generalization ability of corrosion rate prediction models.