Interpretability study on prediction models for alloy pitting based on ensemble learning

Yuhui Song,Qinying Wang,Yuchen Xi,Wenqi Ma,Xingshou Zhang,Lijin Dong,Yiping Wu
DOI: https://doi.org/10.1016/j.corsci.2023.111790
IF: 7.72
2024-01-05
Corrosion Science
Abstract:Ensemble learning (EL) models were developed to predict the pitting potential ( E p ) of alloys. By carefully selecting features and optimizing parameters, the predictive accuracy of the models was improved and experimentally validated. Interpretability methods, SHapley Additive exPlanations (SHAP), Partial Dependence Plot (PDP), and Accumulated Local Effect (ALE), explained the model's prediction logic effectively and portrayed the effects of alloying elements, thermodynamic parameters, and environmental factors on E p in detail. Furthermore, performance differences between interpretability methods, errors in model prediction, and failure cases of interpretability methods were discussed, which provided insights for model optimization and appropriate Interpretability methods selection.
materials science, multidisciplinary,metallurgy & metallurgical engineering
What problem does this paper attempt to address?