A two-stage feature selection method with its application

作者：Zhao, Xuehua; Li, Daoliang^*; Yang, Bo; Chen, Huiling; Yang, Xinbin; Yu, Chenglong; Liu, Shuangyin

来源：Computers & Electrical Engineering, 2015, 47: 114-125.

DOI：10.1016/j.compeleceng.2015.08.011

• Foreign fibers
• Feature selection
• Information gain
• Binary particle swarm optimization

Foreign fibers in cotton seriously affect the quality of cotton products. Online detection systems of foreign fibers based on machine vision are the efficient tools to minimize the harmful effects of foreign fibers. The optimum feature set with small size and high accuracy can efficiently improve the performance of online detection systems. To find the optimal feature sets, a two-stage feature selection algorithm combining IG (Information Gain) approach and BPSO (Binary Particle Swarm Optimization) is proposed for foreign fiber data. In the first stage, IG approach is used to filter noisy features, and the BPSO uses the classifier accuracy as a fitness function to select the highly discriminating features in the second stage. The proposed algorithm is tested on foreign fiber dataset The experimental results show that the proposed algorithm can efficiently find the feature subsets with smaller size and higher accuracy than other algorithms.

• 出版日期2015-10
• 单位深圳信息职业技术学院; 吉林大学; 广东海洋大学; 中国农业大学; 温州大学