TITLE:
Research on Telecom Customer Churn Prediction Based on GA-XGBoost and SHAP
AUTHORS:
Ke Peng, Yan Peng
KEYWORDS:
GA-XGBoost Algorithm, SHAP, Genetic Algorithm, Customer Churn, Prediction, Machine Learning
JOURNAL NAME:
Journal of Computer and Communications,
Vol.10 No.11,
November
28,
2022
ABSTRACT: To address the prominent problems faced by customer churn in telecom enterprise management, a telecom customer churn prediction model integrating GA-XGBoost and SHAP is proposed. By using the ADASYN algorithm for data processing on the unbalanced sample set; based on the GA-XGBoost model, the XGBoost algorithm is used to construct the telecom customer churn prediction model, and the hyperparameters of the model are optimized by using the genetic algorithm. The experimental results show that compared with traditional machine learning methods such as GBDT, decision tree, KNN and single XGBoost model, the improved XGBoost model has better performance in recall, F1 value and AUC value; the GA-XGBoost model is integrated with SHAP framework to analyze and explain the important features affecting telecom customer churn, which is more in line with the telecom industry to predict customer the actual situation of churn.