Nonparametric feature selection by random forests and deep neural networks

Xiaojun Mao,Liuhua Peng,Zhonglei Wang
DOI: https://doi.org/10.1016/j.csda.2022.107436
2022-06-01
Abstract:Random forests are a widely used machine learning algorithm, but their computational efficiency is undermined when applied to large-scale datasets with numerous instances and useless features. Herein, we propose a nonparametric feature selection algorithm that incorporates random forests and deep neural networks, and its theoretical properties are also investigated under regularity conditions. Using different synthetic models and a real-world example, we demonstrate the advantage of the proposed algorithm over other alternatives in terms of identifying useful features, avoiding useless ones, and the computation efficiency. Although the algorithm is proposed using standard random forests, it can be widely adapted to other machine learning algorithms, as long as features can be sorted accordingly.
statistics & probability,computer science, interdisciplinary applications
What problem does this paper attempt to address?