Interpretable Predictive Model for Inclusions in Electroslag Remelting Based on XGBoost and SHAP Analysis

Yuxiao Liu,Yanwu Dong,Zhouhua Jiang,Xi Chen
DOI: https://doi.org/10.1007/s11663-024-03037-x
2024-03-26
Metallurgical and Materials Transactions B
Abstract:The use of machine learning techniques in the metallurgical field has been gradually expanding, but its application in the area of electroslag remelting (ESR) is limited, and the underlying predictive processes of current machine learning models lack exploration. In this study, a predictive method based on SHAP theory and XGBoost algorithm is proposed to forecast B-type inclusions in the process of ESR. The dataset used in this study consists of production data from ESR with 25 features. After data preprocessing, feature selection was performed using the mutual information method combined with metallurgical principles. Five machine learning models were constructed for predicting B-fine inclusions and B-thick inclusions separately using algorithms such as XGBoost. The performance of the five models was compared using metrics such as confusion matrix, ROC curve, and prediction accuracy. For the best performing XGBoost model, SHAP values were calculated, and the SHAP explanation framework was introduced to quantify and attribute the importance of various feature variables. This elucidates the significant factors influencing B-type inclusions and their modes of action, providing a new approach to exploring interpretable machine learning models and offering fresh insights into the prediction of inclusions in the ESR process.
materials science, multidisciplinary,metallurgy & metallurgical engineering
What problem does this paper attempt to address?