arrow
Return

A More Flexible and Robust Feature Selection Algorithm

delete2023-01-01
delete7
delete
OA
AI
T
Tianyi Tu *
Y
Ye Su
Y
Y. A. Tang
W
Wenxue Tan
S
Sheng Ren
DOI:10.1109/ACCESS.2023.3342044delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
With the increasing amount of real data, the challenges of large-scale model operations as well as poor generalization capacity, making selection of an appropriate feature set a significant concern. This study proposes ImprovedRFECV, an enhanced approach for cross-validated recursive feature elimination (RFECV). The algorithm first enhances the robustness of the optimal feature subset through random sampling of different data, building multiple models, and comparing their scores. Simultaneously, the L1 and L2 regularization terms are introduced to evaluate the value of each feature more comprehensively, thus reducing the impact on the anti-interference term and further improving the accuracy of the algorithm and its stability. Furthermore, a multi-model ensemble learning framework is employed to enhance generalization ability and effectively prevent overfitting. Lastly, a both-end expansion removal strategy is adopted to address the issue of strong covariance among features while enhancing the algorithm's flexibility. The experimental results demonstrate that, compared to the RFECV algorithm, the ImprovedRFECV algorithm achieves fewer optimal average features and outperforms the optimal feature subset across five datasets spanning five different domains, demonstrating the algorithm's high level of robustness and generalization ability.
Keywords:
Feature selection
machine learning
stability evaluation
regularization
RFECV

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

H
hunan university of science & engineering
Scholars:
561
Papers: 324
Citations: 0
H
hunan university of arts & science
Scholars:
743
Papers: 665
Citations: 0