Prediction of Mg Alloy Corrosion Based on Machine Learning Models

Abstract

Magnesium alloy is a potential biodegradable metallic material characterized by bone-like elastic modulus, which has great application prospects in medical, automotive, and aerospace industries owing to its bone-like elastic modulus, biocompatibility, and lightweight properties. However, the rapid corrosion rates of magnesium alloys seriously limit their applications. This study collected magnesium alloys’ corrosion data and developed a model to predict the corrosion potential, based on the chemical composition of magnesium alloys. We compared four machine learning algorithms: random forest (RF), multiple linear regression (MLR), support vector machine regression (SVR), and extreme gradient boosting (XGBoost). The RF algorithm offered the most accurate predictions than the other three machine learning algorithms. The input effects on corrosion potential have been investigated. Moreover, we used feature creation (transforming chemical component characteristics into atomic and physical characteristics) so that the input characteristics were not limited to specific chemical compositions. From this result, the model’s application range was widened, and machine learning was used to verify the accuracy and feasibility of predicting corrosion of magnesium alloys.