Shapley additive explanation on machine learning predictions of fatigue lifetimes in piston aluminum alloys under different manufacturing and loading conditions

Abstract

Various input variables, including corrosion time, fretting force, stress, lubrication, heat-treating, and nano-particles, were evaluated by modeling of stress-controlled fatigue lifetimes in AlSi12CuNiMg aluminum alloy of the engine pistons with different machine learning (ML) techniques. Bending fatigue experiments were conducted through cyclic loading with zero mean stress, and then experimental data was predicted by five different ML-based models. Moreover, when the optimal ML prediction model was found, it was analyzed using the Shapley additive explanation (SHAP) values method. Results illustrated that extreme gradient boosting (XGBoost) had superior data for estimations, with average training coefficients of determination of at least 63% and 90%, respectively for fatigue lifetime and its logarithmic value. The SHAP values interpretation of the XGBoost model revealed that fretting force, stress, and corrosion time were the most significant inputs in estimating the logarithm values of fatigue lifetimes, respectively.