节点文献

基于LightGBM的特征选择算法

Feature Selection Algorithm Based on LightGBM

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 李占山姚鑫刘兆赓张家晨

【Author】 LI Zhan-shan;YAO Xin;LIU Zhao-geng;ZHANG Jia-chen;College of Computer Science and Technology,Jilin University;

【机构】 吉林大学计算机科学与技术学院

【摘要】 为解决过滤式和基于演化学习的包裹式两类特征选择算法的缺陷,提出一种新型包裹式特征选择算法LGBFS(LightGBM feature selection).首先引入LightGBM对原始特征构建迭代提升树模型并对特征重要度进行度量;随后结合提出的LR序列前向搜索策略LRSFFS对特征进行选择;最后将所提出算法与9种对比算法在21个标准数据集上进行对比,结果显示LGBFS在21个标准数据集中的16个取得最优分类精度、18个取得最优维度缩减率和最优CPU运行时间.还进行了时间复杂度分析与显著性检验,检验表明LGBFS相较6种对比算法具有显著性差异,也说明LGBFS能够同时兼顾特征子集的计算效率和分类精度.

【Abstract】 In order to solve the shortcomings of the following two types of feature selection algorithms,filtering and wrapping based on evolutionary learning,a new wrapping feature selection algorithm LGBFS( LightGBM feature selection) was proposed. First,LightGBM was introduced to construct an gradient boosting tree model for the original features and measure the importance of features; then the proposed LR sequential forward search strategy LRSFFS was combined to select features; finally,the proposed algorithm was compared with nine algorithms in21 standard datasets. The results show that 16 of the 21 standard data sets of LGBFS have achieved the best classification accuracy,and 18 standard data sets have achieved the best dimensionality reduction rate and the best CPU running time. In addition,time complexity analysis and significance test were carried out. The test shows that LGBFS is significantly different from the six comparison algorithms,and it also shows that LGBFS can balance the calculation efficiency and classification accuracy of feature subsets.

【基金】 国家自然科学基金资助项目(61802056);吉林省自然科学基金资助项目(20180101043JC);吉林省发展和改革委员会产业技术研究与开发项目(2019C053-9)
  • 【文献出处】 东北大学学报(自然科学版) ,Journal of Northeastern University(Natural Science) , 编辑部邮箱 ,2021年12期
  • 【分类号】TP18
  • 【被引频次】11
  • 【下载频次】1717
节点文献中: 

本文链接的文献网络图示:

本文的引文网络