基于优化XGBoost算法的电信用户流失预测  

作　　者：胡晓东 吕铭春 阿克弘 靳梦凡 张金海 

机构地区：[1]青海交通职业技术学院,青海西宁810003 [2]中国电信集团有限公司西宁分公司,青海西宁810001 [3]国家计算机网络与信息安全管理中心青海分中心,青海西宁810000

出　　处：《科技与创新》2024年第10期36-39,44,共5页Science and Technology & Innovation

摘　　要：电信市场竞争激烈的现状下,用户流失对运营商非常不利。为提前精准有效地预测用户流失倾向,提出了基于贝叶斯优化极度梯度提升树(ExtremeGradientBoosting,XGBoost)的电信用户流失预测方法。首先,利用有效的数据预处理方法处理电信用户数据中的异常数据。其次,利用XGBoost算法训练用户流失概率预测模型。同时,针对XGBoost算法学习超参数众多、缺乏高效准确的超参数寻优方法的问题,利用贝叶斯优化算法、随机搜索算法、网格搜索算法3种方法进行超参数寻优。最后,对比分析3种优化方法的优化效果。实验结果表明,该方法能够有效预测用户流失倾向,且贝叶斯优化XGBoost算法的电信用户流失模型准确率均优于基于随机搜索优化和基于网格搜索优化,XGBoost算法所建立的模型,准确度比优化前提升了4.84%。

关 键 词：XGBoost算法 用户流失 数据挖掘 贝叶斯优化 

分 类 号：TP18[自动化与计算机技术—控制理论与控制工程] F626[自动化与计算机技术—控制科学与工程]