Feature selection based on artificial bee colony and gradient boosting decision tree
Feature selection based on artificial bee colony and gradient boosting decision tree
复制标题
基于人工蜂群和梯度提升决策树的特征选择
DOI:
10.1016/j.asoc.2018.10.036
复制
发表时间:
2019-01-01
影响因子:
8.7
通讯作者:
Gu, Lichuan
中科院分区:
文献类型:
--
作者:
Rao, Haidi;Shi, Xianzhang;Gu, Lichuan
Data from many real-world applications can be high dimensional and features of such data are usually highly redundant. Identifying informative features has become an important step for data mining to not only circumvent the curse of dimensionality but to reduce the amount of data for processing. In this paper, we propose a novel feature selection method based on bee colony and gradient boosting decision tree aiming at addressing problems such as efficiency and informative quality of the selected features. Our method achieves global optimization of the inputs of the decision tree using the bee colony algorithm to identify the informative features. The method initializes the feature space spanned by the dataset. Less relevant features are suppressed according to the information they contribute to the decision making using an artificial bee colony algorithm. Experiments are conducted with two breast cancer datasets and six datasets from the public data repository. Experimental results demonstrate that the proposed method effectively reduces the dimensions of the dataset and achieves superior classification accuracy using the selected features. (C) 2018 Elsevier B.V. All rights reserved.