面向大型数据集的高效决策树参数剪枝算法  

High-Efficient Parameter-Pruning Algorithm of Decision Tree for Large Dataset

作　　者：谢兆贤 邹兴敏 张文静[1] XIE Zhaoxian;ZOU Xingmin;ZHANG Wenjing(School of Cyber Science and Engineering,Qufu Normal University,Qufu 273165,Shandong,China)

机构地区：[1]曲阜师范大学网络空间安全学院,山东曲阜273165

出　　处：《计算机工程》2024年第1期156-165,共10页Computer Engineering

基　　金：山东省自然科学基金面上项目(ZR2020MF048)。

摘　　要：决策树在数据分类上具有较好的效果,但容易产生过拟合的现象,解决方案是对决策树进行剪枝处理,然而传统剪枝算法普遍存在预剪枝容易欠拟合、后剪枝时间消耗多、网络搜索剪枝仅适用于小型数据集等问题。为了解决以上问题,提出一种高效的决策树参数剪枝算法。根据网络安全态势感知模型,建立剪枝决策树态势感知系统架构,分析网络数据流。在生成决策树的过程中,利用枚举与二分搜索算法找出决策树最大深度,采用深度优先搜索算法找到节点最小分裂数和最大特征数,最终结合这3个最优参数自上而下完成剪枝。实验结果表明,所提算法在大型数据集上的过拟合风险较小,训练集与测试集准确率都在95%以上,同时相比于后剪枝算法中表现较好的悲观错误剪枝算法快了近20倍。Decision tree(DT)have a good effect on data classification but easily develop overfitting.The solution to this problem is to prune the DT.However,the pruning algorithm has shortcomings;for example,prepruning is prone to underfitting,the postpruning time is extended,and Web-search pruning is only suitable for small datasets.This study proposes an efficient parameter-pruning algorithm for the DT to solve the above problems.Based on the network security situation awareness model,the architecture of the pruned decision-tree situation awareness system is established,and the data flow of the network is analyzed.During the process of generating the DT,enumeration and binary search algorithms are used to determine the maximum depth of the DT,and a depth-first search algorithm is used to determine the minimum number of split nodes and the maximum number of features.Finally,the three optimal parameters are combined to complete the pruning from top to bottom.The experimental results show that this algorithm has a low risk of overfitting in large datasets.The accuracy of the training and testing sets exceed 95%.Compared to the Pessimistic Error-Pruning(PEP)algorithm that exhibits the best performance in post-pruning algorithms,the pruning algorithm is almost 20 times faster.

关 键 词：决策树 剪枝 过拟合 安全态势感知 泛化性 

分 类 号：TP313.13[自动化与计算机技术—计算机软件与理论]