一种面向能源贫困识别的轻量可解释梯度提升树

分享到

【作者】王政，裔扬，史颖，赵兴旺，吴晨旭
【刊名】山西大学学报(自然科学版)
【作者单位】¹太原师范学院智能优化计算与区块链技术山西省重点实验室；²扬州大学信息工程学院；³太原师范学院计算机科学与技术学院；⁴山西大学计算机与信息技术学院
【ISSN】0253-2395
【关键词】LightGBM模型能源贫困预测特征关联分析模型解释方法
【摘要】为解决传统梯度提升树方法在解决能源贫困识别问题时容易出现的训练不足、过拟合、可解释性差等问题，本文提出了一种面向能源贫困识别任务的轻量可解释梯度提升树方法。该方法首先剔除原始数据中的缺失值、异常值等噪声样本，根据特征关联分析后的样本梯度进行排序，以实现梯度提升树内部节点的分割，随后采用特征绑定技术加速训练过程，以实现模型的轻量化；其次，引入模型解释方法进行影响因素分析，量化不同特征对能源贫困识别的影响程度，增强了模型的可解释性。在典型的能源贫困识别数据集上的实验结果表明，与其他方法相比，本文提出...
【基金】国家自然科学基金
【文献类型】期刊