Unbiased Feature Selection in Learning Random Forests for High-Dimensional Data

Abstract EN

Random forests (RFs) have been widely used as a powerful classification method.

However, with the randomization in both bagging samples and feature selection, the trees in the forest tend to select uninformative features for node splitting.

This makes RFs have poor accuracy when working with high-dimensional data.

Besides that, RFs have bias in the feature selection process where multivalued features are favored.

Aiming at debiasing feature selection in RFs, we propose a new RF algorithm, called xRF, to select good features in learning RFs for high-dimensional data.

We first remove the uninformative features using p-value assessment, and the subset of unbiased features is then selected based on some statistical measures.

This feature subset is then partitioned into two subsets.

A feature weighting sampling technique is used to sample features from these two subsets for building trees.

This approach enables one to generate more accurate trees, while allowing one to reduce dimensionality and the amount of data needed for learning RFs.

An extensive set of experiments has been conducted on 47 high-dimensional real-world datasets including image datasets.

The experimental results have shown that RFs with the proposed approach outperformed the existing random forests in increasing the accuracy and the AUC measures.