The Scientific World Journal
Volume 2015 (2015), Article ID 471371, 18 pages
http://dx.doi.org/10.1155/2015/471371

Research Article

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

Thanh-Tung Nguyen,1,2,3 Joshua Zhexue Huang,1,4 and Thuy Thi Nguyen⁵

¹Shenzhen Key Laboratory of High Performance Data Mining, Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, Shenzhen 518055, China
²University of Chinese Academy of Sciences, Beijing 100049, China
³School of Computer Science and Engineering, Water Resources University, Hanoi 10000, Vietnam
⁴College of Computer Science and Software Engineering, Shenzhen University, Shenzhen 518060, China
⁵Faculty of Information Technology, Vietnam National University of Agriculture, Hanoi 10000, Vietnam

Received 20 June 2014; Accepted 20 August 2014

Academic Editor: Shifei Ding

Copyright © 2015 Thanh-Tung Nguyen et al. This is an open access article distributed under the Creative Commons Attribution License, which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.

Abstract.

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 -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.

1. Introduction

Random forests (RFs) [1] are a nonparametric method that builds an ensemble model of decision trees from random subsets of features and bagged samples of the training data.

RFs have shown excellent performance for both classification and regression problems. RF model works well even when predictive features contain irrelevant features (or noise); it can be used when the number of features is much larger than the number of samples. However, with randomizing mechanism in both bagging samples and feature selection, RFs could give poor accuracy when applied to high dimensional data. The main cause is that, in the process of growing a tree from the bagged sample data, the subspace of features randomly sampled from thousands of features to split a node of the tree is often dominated by uninformative features (or noise), and the tree grown from such bagged subspace of features will have a low accuracy in prediction which affects the final prediction of the RFs. Furthermore, Breiman et al. noted that feature selection is biased in the classification and regression tree (CART) model because it is based on an information criteria, called multivalue problem [2]. It tends in favor of features containing more values, even if these features have lower importance than other ones or have no relationship with the response feature (i.e., containing less missing values, many categorical or distinct numerical values) [3, 4].

In this paper, we propose a new random forests algorithm using an unbiased feature sampling method to build a good subspace of unbiased features for growing trees. We first use random forests to measure the importance of features and produce raw feature importance scores. Then, we apply a statistical Wilcoxon rank-sum test to separate informative features from the uninformative ones. This is done by neglecting all uninformative features by defining threshold ; for instance, . Second, we use the Chi-square statistic test () to compute the related scores of each feature to the response feature. We then partition the set of the remaining informative features into two subsets, one containing highly informative features and the other one containing weak informative features. We independently sample features from the two subsets and merge them together to get a new subspace of features, which is used for splitting the data at nodes. Since the subspace always contains highly informative features which can guarantee a better split at a node, this feature sampling method enables avoiding selecting biased features and generates trees from bagged sample data with higher accuracy. This sampling method also is used for dimensionality reduction, the amount of data needed for training the random forests model. Our experimental results have shown that random forests with this weighting feature selection technique outperformed recently the proposed random forests in increasing of the prediction accuracy; we also applied the new approach on microarray and image data and achieved outstanding results.

The structure of this paper is organized as follows. In Section 2, we give a brief summary of related works. In Section 3 we give a brief summary of random forests and measurement of feature importance score. Section 4 describes our new proposed algorithm using unbiased feature selection. Section 5 provides the experimental results, evaluations, and comparisons. Section 6 gives our conclusions.

2. Related Works

Random forests are an ensemble approach to make classification decisions by voting the results of individual decision trees. An ensemble learner with excellent generalization accuracy has two properties, high accuracy of each component learner and high diversity in component learners [5]. Unlike other ensemble methods such as bagging [1] and boosting [6, 7], which create basic classifiers from random samples of the training data, the random forest approach creates the basic classifiers from randomly selected subspaces of data [8, 9]. The randomly selected subspaces increase the diversity of basic classifiers learnt by a decision tree algorithm.

Feature importance is the importance measure of features in the feature selection process [1, 10–14]. In RF frameworks, the most commonly used score of importance of a given feature is the mean error of a tree in the forest when the observed values of this feature are randomly permuted in the out-of-bag samples. Feature selection is an important step to obtain good performance for an RF model, especially in dealing with high dimensional data problems.

For feature weighting techniques, recently Xu et al. [13] proposed an improved RF method which uses a novel feature weighting method for subspace selection and therefore enhances classification performance on high dimensional data. The weights of feature were calculated by information gain ratio or -test; Ye et al. [14] then used these weights to propose a stratified sampling method to select feature subspaces for RF in classification problems. Chen et al. [15] used a stratified idea to propose a new clustering method. However, implementation of the random forest model suggested by Ye et al. is based on a binary classification setting, and it uses linear discriminant analysis as the splitting criteria. This stratified RF model is not efficient on high dimensional datasets with multiple classes. With the same way for solving two-class problem, Amaratunga et al. [16] presented a feature weighting method for subspace sampling to deal with microarray data, the -test of variance analysis is used to compute weights for the features. Genuer et al. [12] proposed a strategy involving a ranking of explanatory features using the RFs score weights of importance and a stepwise ascending feature introduction strategy. Deng and Runger [17] proposed a guided regularized RF (GRRF), in which weights of importance scores from an ordinary random forest (RF) are used to guide the feature selection process. They found that the regularized least subset selected by their GRRF with minimal regularization ensures better accuracy than the complete feature set. However, regular RF was used as a classifier due to the fact that regularized RF may have higher variance than RF because the trees are correlated.

Several methods have been proposed to correct bias of importance measures in the feature selection process in RFs to improve the prediction accuracy [18–21]. These methods intend to avoid selecting an uninformative feature for node splitting in decision trees. Although the methods of this kind were well investigated and can be used to address the high dimensional problem, there are still some unsolved problems, such as the need to specify in advance the probability distributions, as well as the fact that they struggle when applied to large high dimensional data.

In summary, in the reviewed approaches, the gain at higher levels of the tree is weighted differently than the gain at lower levels of the tree. In fact, at lower levels of the tree, the gain is reduced because of the effect of splits on different features at higher levels of the tree. That affects the final prediction performance of RFs model. To remedy this, in this paper we propose a new method for unbiased feature subsets selection in high dimensional space to build RFs. Our approach differs from previous approaches in the techniques used to partition a subset of features. All uninformative features (considered as noise) are removed from the system and the best feature set, which is highly related to the response feature, is found using a statistical method. The proposed sampling method always provides enough highly informative features for the subspace feature at any levels of the decision trees. For the case of growing an RF model on data without noise, we used in-bag measures. This is a different importance score of features, which requires less computational time compared to the measures used by others. Our experimental results showed that our approach outperformed recently the proposed RF methods.

3. Background

3.1. Random Forest Algorithm

Given a training dataset , where are features (also called predictor variables), is a class response feature, is the number of training samples, and is the number of features and a random forest model described in Algorithm 1, let be the prediction of tree given input . The prediction of random forest with trees is

    input:   : the training data set,

                : the number of trees,

                mtry: the size of the subspaces.

    output: A random forest RF

(1) for    to    do

(2)     Draw a bagged subset of samples from .

(4)     while (stopping  criteria  is  not  met) do

(5)         Select randomly mtry features.

(6)         for    to    do

(7)             Compute the decrease in the node impurity.

(8)         Choose the feature which decreases the impurity the most and

             the node is divided into two children nodes.

(9) Combine the trees to form a random forest.

Algorithm 1: Random forest algorithm.

Since each tree is grown from a bagged sample set, it is grown with only two-thirds of the samples in , called in-bag samples. About one-third of the samples is left out and these samples are called out-of-bag (OOB) samples which are used to estimate the prediction error.

The OOB predicted value is where , and are in-bag and out-of-bag sampled indices, is the size of OOB subdataset, and the OOB prediction error is

where is an error function and is OOB samples’ size.

3.2. Measurement of Feature Importance Score from an RF

Breiman presented a permutation technique to measure the importance of features in the prediction [1], called an out-of-bag importance score. The basic idea for measuring this kind of importance score of features is to compute the difference between the original mean error and the randomly permuted mean error in OOB samples. The method rearranges stochastically all values of the th feature in OOB for each tree and uses the RF model to predict this permuted feature and get the mean error. The aim of this permutation is to eliminate the existing association between the th feature and values and then to test the effect of this on the RF model. A feature is considered to be in a strong association if the mean error decreases dramatically.

The other kind of feature importance measure can be obtained when the random forest is growing. This is described as follows. At each node in a decision tree, the split is determined by the decrease in node impurity . The node impurity is the gini index. If a subdataset in node contains samples from classes, gini() is defined as

where is the relative frequency of class in . Gini() is minimized if the classes in are skewed. After splitting into two child nodes and with sample sizes and , the gini index of the split data is defined as

The feature providing smallest is chosen to split the node. The importance score of feature in a single decision tree is

and it is computed over all trees in a random forest, defined as

It is worth noting that a random forest uses in-bag samples to produce a kind of importance measure, called an in-bag importance score. This is the main difference between the in-bag importance score and an out-of-bag measure, which is produced with the decrease of the prediction error using RF in OOB samples. In other words, the in-bag importance score requires less computation time than the out-of-bag measure.

4. Our Approach

4.1. Issues in Feature Selection on High Dimensional Data

When Breiman et al. suggested the classification and regression tree (CART) model, they noted that feature selection is biased because it is based on an information gain criteria, called multivalue problem [2]. Random forest methods are based on CART trees [1]; hence this bias is carried to random forest RF model. In particular, the importance scores can be biased when very high dimensional data contains multiple data types. Several methods have been proposed to correct bias of feature importance measures [18–21]. The conditional inference framework (referred to as cRF [22]) could be successfully applied for both the null and power cases [19, 20, 22]. The typical characteristic of the power case is that only one predictor feature is important, while the rest of the features are redundant with different cardinality. In contrast, in the null case all features used for prediction are redundant with different cardinality. Although the methods of this kind were well investigated and can be used to address the multivalue problem, there are still some unsolved problems, such as the need to specify in advance the probability distributions, as well as the fact that they struggle when applied to high dimensional data.

Another issue is that, in high dimensional data, when the number of features is large, the fraction of importance features remains so small. In this case the original RF model which uses simple random sampling is likely to perform poorly with small , and the trees are likely to select an uninformative feature as a split too frequently ( denotes a subspace size of features). At each node of a tree, the probability of uninformative feature selection is too high.

To illustrate this issue, let be the number of noisy features, denote by the total number of predictor features, and let the features be important ones which have a high correlation with values. Then, if we use simple random sampling when growing trees to select a subset of features (), the total number of possible uninformative a and the total number of all subset features is . The probability distribution of selecting a subset of () important features is given by

Because the fraction of important features is too small, the probability in (7) tends to , which means that the important features are rarely selected by the simple sampling method in RF [1]. For example, with 5 informative and 5000 noise or uninformative features, assuming , the probability of an informative feature to be selected at any split is .

4.2. Bias Correction for Feature Selection and Feature Weighting

The bias correction in feature selection is intended to make the RF model to avoid selecting an uninformative feature. To correct this kind of bias in the feature selection stage, we generate shadow features to add to the original dataset. The shadow features set contains the same values, possible cut-points, and distribution with the original features but have no association with values. To create each shadow feature, we rearrange the values of the feature in the original dataset times to create the corresponding shadow. This disturbance of features eliminates the correlations of the features with the response value but keeps its attributes. The shadow feature participates only in the competition for the best split and makes a decrease in the probability of selecting this kind of uninformative feature. For the feature weight computation, we first need to distinguish the important features from the less important ones. To do so, we run a defined number of random forests to obtain raw importance scores, each of which is obtained using (6). Then, we use Wilcoxon rank-sum test [23] that compares the importance score of a feature with the maximum importance scores of generated noisy features called shadows. The shadow features are added to the original dataset and they do not have prediction power to the response feature. Therefore, any feature whose importance score is smaller than the maximum importance score of noisy features is considered less important. Otherwise, it is considered important. Having computed the Wilcoxon rank-sum test, we can compute the -value for the feature. The -value of a feature in Wilcoxon rank-sum test is assigned a weight with a feature , -value   , and this weight indicates the importance of the feature in the prediction. The smaller the -value of a feature, the more correlated the predictor feature to the response feature, and therefore the more powerful the feature in prediction. The feature weight computation is described as follows.

Let be the number of features in the original dataset, and denote the feature set as . In each replicate , shadow features are generated from features in , and we randomly permute all values of times to get a corresponding shadow feature ; denote the shadow feature set as . The extended feature set is denoted by .

Let the importance score of at replicate be where and are the importance scores of and at the th replicate, respectively. We built a random forest model from the dataset to compute importance scores for features. We repeated the same process times to compute replicates getting and . From the replicates of shadow features, we extracted the maximum value from th row of and put it into the comparison sample denoted by . For each data feature , we computed Wilcoxon test and performed hypothesis test on to calculate the -value for the feature. Given a statistical significance level, we can identify important features from less important ones. This test confirms that if a feature is important, it consistently scores higher than the shadow over multiple permutations. This method has been presented in [24, 25].

In each node of trees, each shadow shares approximately the same properties of the corresponding , but it is independent on and consequently has approximately the same probability of being selected as a splitting candidate. This feature permutation method can reduce bias due to different measurement levels of according to -value and can yield correct ranking of features according to their importance.

4.3. Unbiased Feature Weighting for Subspace Selection

Given all -values for all features, we first set a significance level as the threshold , for instance . Any feature whose -value is greater than is considered a uninformative feature and is removed from the system; otherwise, the relationship with is assessed. We now consider the set of features obtained from after neglecting all uninformative features.

Second, we find the best subset of features which is highly related to the response feature; a measure correlation function is used to test the association between the categorical response feature and each feature . Each observation is allocated to one cell of a two-dimensional array of cells (called a contingency table) according to the values of . If there are rows and columns in the table and is the number of total samples, the value of the test statistic is

For the test of independence, a chi-squared probability of less than or equal to 0.05 is commonly interpreted for rejecting the hypothesis that the row variable is independent of the column feature.

Let be the best subset of features, we collect all feature whose -value is smaller or equal to as a result from the statistical test according to (8). The remaining features are added to , and this approach is described in Algorithm 2. We independently sample features from the two subsets and put them together as the subspace features for splitting the data at any node, recursively. The two subsets partition the set of informative features in data without irrelevant features. Given and , at each node, we randomly select features from each group of features. For a given subspace size, we can choose proportions between highly informative features and weak informative features that depend on the size of the two groups. That is and , where and are the number of features in the groups of highly informative features and weak informative features , respectively. is the number of informative features in the input dataset. These are merged to form the feature subspace for splitting the node.

    input:   The training data set and a random forest RF.

                , : The number of replicates and the threshold.

    output: and .

(1)   Let , .

(2)   for    to    do

(3)       .

(4)       .

(5)       Build RF model from to produce ,

(6)        and , .

(7)   Set .

(8)   for    to    do

(9)       Compute Wilcoxon rank-sum test with and .

(10)     Compute values for each feature .

(11)     if    then

(12)         

(13) Set , .

(14) Compute statistic to get value

(15) for    to    do

(16)     if () then

(17)         

(18)

(19) return  

Algorithm 2: Feature subspace selection.

4.4. Our Proposed RF Algorithm

In this section, we present our new random forest algorithm called , which uses the new unbiased feature sampling method to generate splits at the nodes of CART trees [2]. The proposed algorithm includes the following main steps: (i) weighting the features using the feature permutation method, (ii) identifying all unbiased features and partitioning them into two groups and , (iii) building RF using the subspaces containing features which are taken randomly and separately from , , and (iv) classifying a new data. The new algorithm is summarized as follows.

(1)Generate the extended dataset of dimensions by permuting the corresponding predictor feature values for shadow features.(2)Build a random forest model RF from and compute replicates of raw importance scores of all predictor features and shadows with RF. Extract the maximum importance score of each replicate to form the comparison sample of elements.(3)For each predictor feature, take importance scores and compute Wilcoxon test to get -value, that is, the weight of each feature.(4)Given a significance level threshold , neglect all uninformative features.(5)Partition the remaining features into two subsets and described in Algorithm 2.(6)Sample the training set with replacement to generate bagged samples .(7)For each , grow a CART tree as follows.

(a)At each node, select a subspace of features randomly and separately from and and use the subspace features as candidates for splitting the node.(b)Each tree is grown nondeterministically, without pruning until the minimum node size is reached.

(8)Given a , use (1) to predict the response value.

5. Experiments

5.1. Datasets

Real-world datasets including image datasets and microarray datasets were used in our experiments. Image classification and object recognition are important problems in computer vision. We conducted experiments on four benchmark image datasets, including the Caltech categories (http://www.vision.caltech.edu/html-files/archive.html) dataset, the Horse (http://pascal.inrialpes.fr/data/horses/) dataset, the extended YaleB database [26], and the AT&T ORL dataset [27].

For the Caltech dataset, we use a subset of 100 images from the Caltech face dataset and 100 images from the Caltech background dataset following the setting in ICCV (http://people.csail.mit.edu/torralba/shortCourseRLOC/). The extended YaleB database consists of 2414 face images of 38 individuals captured under various lighting conditions. Each image has been cropped to a size of 192 × 168 pixels and normalized. The Horse dataset consists of 170 images containing horses for the positive class and 170 images of the background for the negative class. The AT&T ORL dataset includes of 400 face images of 40 persons.

In the experiments, we use a bag of words for image features representation for the Caltech and the Horse datasets. To obtain feature vectors using bag-of-words method, image patches (subwindows) are sampled from the training images at the detected interest points or on a dense grid. A visual descriptor is then applied to these patches to extract the local visual features. A clustering technique is then used to cluster these, and the cluster centers are used as visual code words to form visual codebook. An image is then represented as a histogram of these visual words. A classifier is then learned from this feature set for classification.

In our experiments, traditional -means quantization is used to produce the visual codebook. The number of cluster centers can be adjusted to produce the different vocabularies, that is, dimensions of the feature vectors. For the Caltech and Horse datasets, nine codebook sizes were used in the experiments to create 18 datasets as follows: {CaltechM300, CaltechM500, CaltechM1000, CaltechM3000, CaltechM5000, CaltechM7000, CaltechM1000, CaltechM12000, CaltechM15000}, and {HorseM300, HorseM500, HorseM1000, HorseM3000, HorseM5000, HorseM7000, HorseM1000, HorseM12000, HorseM15000}, where M denotes the number of codebook sizes.

For the face datasets, we use two type of features: eigenface [28] and the random features (randomly sample pixels from the images). We used four groups of datasets with four different numbers of dimensions . Totally, we created 16 subdatasets as {YaleB.EigenfaceM30, YaleB.EigenfaceM56, YaleB.EigenfaceM120, YaleB.EigenfaceM504}, {YaleB.RandomfaceM30, YaleB.RandomfaceM56, YaleB.RandomfaceM120, YaleB.RandomfaceM504}, {ORL.EigenfaceM30, ORL.EigenM56, ORL.EigenM120, ORL.EigenM504}, and {ORL.RandomfaceM30, ORL.RandomM56, ORL.RandomM120, ORL.RandomM504}.

The properties of the remaining datasets are summarized in Table 1. The Fbis dataset was compiled from the archive of the Foreign Broadcast Information Service and the La1s, La2s datasets were taken from the archive of the Los Angeles Times for TREC-5 (http://trec.nist.gov/). The ten gene datasets are used and described in [11, 17]; they are always high dimensional and fall within a category of classification problems which deal with large number of features and small samples. Regarding the characteristics of the datasets given in Table 1, the proportion of the subdatasets, namely, Fbis, La1s, La2s, was used individually for a training and testing dataset.

Table 1: Description of the real-world datasets sorted by the number of features and grouped into two groups, microarray data and real-world datasets, accordingly.

Dataset
No. of features
No. of training
No. of tests
No. of classes

Colon
2,000
62
—
2

Srbct
2,308
63
—
4

Leukemia
3,051
38
—
2

Lymphoma
4,026
62
—
3

breast.2.class
4,869
78
—
2

breast.3.class
4,869
96
—
3

nci
5,244
61
—
8

Brain
5,597
42
—
5

Prostate
6,033
102
—
2

adenocarcinoma
9,868
76
—
2

Fbis
2,000
1,711
752
17

La2s
12,432
1,855
845
6

La1s
13,195
1,963
887
6

5.2. Evaluation Methods

We calculated some measures such as error bound (), strength (), and correlation () according to the formulas given in Breiman’s method [1]. The correlation measures indicate the independence of trees in a forest, whereas the average strength corresponds to the accuracy of individual trees. Lower correlation and higher strength result in a reduction of general error bound measured by () which indicates a high accuracy RF model.

The two measures are also used to evaluate the accuracy of prediction on the test datasets: one is the area under the curve (AUC) and the other one is the test accuracy (Acc), defined as

where is the indicator function and is the number of votes for on class , is the th tree classifier, is the number of samples in test data , and indicates the true class of .

5.3. Experimental Settings

The latest -packages random Forest and RRF [29, 30] were used in environment to conduct these experiments. The GRRF model was available in the RRF -package. The wsRF model, which used weighted sampling method [13] was intended to solve classification problems. For the image datasets, the 10-fold cross-validation was used to evaluate the prediction performance of the models. From each fold, we built the models with 500 trees and the feature partition for subspace selection in Algorithm 2 was recalculated on each training fold dataset. The and parameters were set to and , respectively. The experimental results were evaluated in two measures AUC and the test accuracy according to (9).

We compared across a wide range the performances of the 10 gene datasets, used in [11]. The results from the application of GRRF, varSelRF, and LASSO logistic regression on the ten gene datasets are presented in [17]. These three gene selection methods used RF -package [30] as the classifier. For the comparison of the methods, we used the same settings which are presented in [17], for the coefficient we used value of , because GR-RF(0.1) has shown competitive accuracy [17] when applied to the 10 gene datasets. The 100 models were generated with different seeds from each training dataset and each model contained 1000 trees. The and parameters were of the same settings on the image dataset. From each of the datasets two-thirds of the data were randomly selected for training. The other one-third of the dataset was used to validate the models. For comparison, Breiman’s RF method, the weighted sampling random forest wsRF model, and the xRF model were used in the experiments. The guided regularized random forest GRRF [17] and the two well-known feature selection methods using RF as a classifier, namely, varSelRF [31] and LASSO logistic regression [32], are also used to evaluate the accuracy of prediction on high-dimensional datasets.

In the remaining datasets, the prediction performances of the ten random forest models were evaluated, each one was built with 500 trees. The number of features candidates to split a node was . The minimal node size was 1. The xRF model with the new unbiased feature sampling method is a new implementation. We implemented the xRF model as multithread processes, while other models were run as single-thread processes. We used to call the corresponding C/C++ functions. All experiments were conducted on the six 64-bit Linux machines, with each one being equipped with Intel Xeon CPU E5620 2.40 GHz, 16 cores, 4 MB cache, and 32 GB main memory.

5.4. Results on Image Datasets

Figures 1 and 2 show the average accuracy plots of recognition rates of the models on different subdatasets of the datasets YaleB and ORL. The GRRF model produced slightly better results on the subdataset ORL.RandomM120 and ORL dataset using eigenface and showed competitive accuracy performance with the xRF model on some cases in both YaleB and ORL datasets, for example, YaleB.EigenM120, ORL.RandomM56, and ORL.RandomM120. The reason could be that truly informative features in this kind of datasets were many. Therefore, when the informative feature set was large, the chance of selecting informative features in the subspace increased, which in turn increased the average recognition rates of the GRRF model. However, the xRF model produced the best results in the remaining cases. The effect of the new approach for feature subspace selection is clearly demonstrated in these results, although these datasets are not high dimensional.

(a)

(b)

Figure 1: Recognition rates of the models on the YaleB subdatasets, namely, YaleB.EigenfaceM30, YaleB.EigenfaceM56, YaleB.EigenfaceM120, YaleB.EigenfaceM504, and YaleB.RandomfaceM30, YaleB.RandomfaceM56, YaleB.RandomfaceM120, and YaleB.RandomfaceM504.

(a)

(b)

Figure 2: Recognition rates of the models on the ORL subdatasets, namely, ORL.EigenfaceM30, ORL.EigenM56, ORL.EigenM120, ORL.EigenM504, and ORL.RandomfaceM30, ORL.RandomM56, ORL.RandomM120, and ORL.RandomM504.

Figures 3 and 5 present the box plots of the test accuracy (mean std-dev%); Figures 4 and 6 show the box plots of the AUC measures of the models on the 18 image subdatasets of the Caltech and Horse, respectively. From these figures, we can observe that the accuracy and the AUC measures of the models GRRF, wsRF, and xRF were increased on all high-dimensional subdatasets when the selected subspace was not so large. This implies that when the number of features in the subspace is small, the proportion of the informative features in the feature subspace is comparatively large in the three models. There will be a high chance that highly informative features are selected in the trees so the overall performance of individual trees is increased. In Brieman’s method, many randomly selected subspaces may not contain informative features, which affect the performance of trees grown from these subspaces. It can be seen that the xRF model outperformed other random forests models on these subdatasets in increasing the test accuracy and the AUC measures. This was because the new unbiased feature sampling was used in generating trees in the xRF model; the feature subspace provided enough highly informative features at any levels of the decision trees. The effect of the unbiased feature selection method is clearly demonstrated in these results.

Figure 3: Box plots: the test accuracy of the nine Caltech subdatasets.

Figure 4: Box plots of the AUC measures of the nine Caltech subdatasets.

Figure 5: Box plots of the test accuracy of the nine Horse subdatasets.

Figure 6: Box plots of the AUC measures of the nine Horse subdatasets.

Table 2 shows the results of against the number of codebook sizes on the Caltech and Horse datasets. In a random forest, the tree was grown from a bagging training data. Out-of-bag estimates were used to evaluate the strength, correlation, and . The GRRF model was not considered in this experiment because this method aims to find a small subset of features, and the same RF model in -package [30] is used as a classifier. We compared the xRF model with two kinds of random forest models RF and wsRF. From this table, we can observe that the lowest values occurred when the wsRF model was applied to the Caltech dataset. However, the xRF model produced the lowest error bound on the dataset. These results demonstrate the reason that the new unbiased feature sampling method can reduce the upper bound of the generalization error in random forests.

Table 2: The () error bound results of random forest models against the number of codebook size on the Caltech and Horse datasets. The bold value in each row indicates the best result.

Dataset
Model
300
500
1000
3000
5000
7000
10000
12000
15000

Caltech
xRF
.0312
.0271
.0280
.0287
.0357
.0440
.0650
.0742
.0789

RF
.0369
.0288
.0294
.0327
.0435
.0592
.0908
.1114
.3611

wsRF
.0413
.0297
.0268
.0221
.0265
.0333
.0461
.0456
.0789

Horse
xRF
.0266
.0262
.0246
.0277
.0259
.0298
.0275
.0288
.0382

RF
.0331
.0342
.0354
.0374
.0417
.0463
.0519
.0537
.0695

wsRF
.0429
.0414
.0391
.0295
.0288
.0333
.0295
.0339
.0455

Table 3 presents the prediction accuracies (mean std-dev%) of the models on subdatasets CaltechM3000, HorseM3000, YaleB.EigenfaceM504, YaleB.randomfaceM504, ORL.EigenfaceM504, and ORL.randomfaceM504. In these experiments, we used the four models to generate random forests with different sizes from 20 trees to 200 trees. For the same size, we used each model to generate 10 random forests for the 10-fold cross-validation and computed the average accuracy of the 10 results. The GRRF model showed slightly better results on YaleB.EigenfaceM504 with different tree sizes. The wsRF model produced the best prediction performance on some cases when applied to small subdatasets YaleB.EigenfaceM504, ORL.EigenfaceM504, and ORL.randomfaceM504. However, the xRF model produced, respectively, the highest test accuracy on the remaining subdatasets and AUC measures on high-dimensional subdatasets CaltechM3000 and HorseM3000, as shown in Tables 3 and 4. We can clearly see that the xRF model also outperformed other random forests models in classification accuracy on most cases in all image datasets. Another observation is that the new method is more stable in classification performance because the mean and variance of the test accuracy measures were minor changed when varying the number of trees.

Table 3: The prediction test accuracy (mean% ± std-dev%) of the models on the image datasets against the number of trees . The number of feature dimensions in each subdataset is fixed. Numbers in bold are the best results.

Dataset
Model
= 20
= 50
= 80
= 100
= 200

CaltechM3000
xRF
95.50 ± .2
96.50 ± .1
96.50 ± .2
97.00 ± .1
97.50 ± .2

RF
70.00 ± .7
76.00 ± .9
77.50 ± 1.2
82.50 ± 1.6
81.50 ± .2

wsRF
91.50 ± .4
91.00 ± .3
93.00 ± .2
94.50 ± .4
92.00 ± .9

GRRF
93.00 ± .2
96.00 ± .2
94.50 ± .2
95.00 ± .3
94.00 ± .2

HorseM3000
xRF
80.59 ± .4
81.76 ± .2
79.71 ± .6
80.29 ± .1
77.65 ± .5

RF
50.59 ± 1.0
52.94 ± .8
56.18 ± .4
58.24 ± .5
57.35 ± .9

wsRF
62.06 ± .4
68.82 ± .3
67.65 ± .3
67.65 ± .5
65.88 ± .7

GRRF
65.00 ± .9
63.53 ± .3
68.53 ± .3
63.53 ± .9
71.18 ± .4

YaleB.EigenfaceM504
xRF
75.68 ± .1
85.65 ± .1
88.08 ± .1
88.94 ± .0
91.22 ± .0

RF
71.93 ± .1
79.48 ± .1
80.69 ± .1
81.67 ± .1
82.89 ± .1

wsRF
77.60 ± .1
85.61 ± .0
88.11 ± .0
89.31 ± .0
90.68 ± .0

GRRF
74.73 ± .0
84.70 ± .1
87.25 ± .0
89.61 ± .0
91.89 ± .0

YaleB.randomfaceM504
xRF
94.71 ± .0
97.64 ± .0
98.01 ± .0
98.22 ± .0
98.59 ± .0

RF
88.00 ± .0
92.59 ± .0
94.13 ± .0
94.86 ± .0
96.06 ± .0

wsRF
95.40 ± .0
97.90 ± .0
98.17 ± .0
98.14 ± .0
98.38 ± .0

GRRF
95.66 ± .0
98.10 ± .0
98.42 ± .0
98.92 ± .0
98.84 ± .0

ORL.EigenfaceM504
xRF
76.25 ± .6
87.25 ± .3
91.75 ± .2
93.25 ± .2
94.75 ± .2

RF
71.75 ± .2
78.75 ± .4
82.00 ± .3
82.75 ± .3
85.50 ± .5

wsRF
78.25 ± .4
88.75 ± .3
90.00 ± .1
91.25 ± .2
92.50 ± .2

GRRF
73.50 ± .6
85.00 ± .2
90.00 ± .1
90.75 ± .3
94.75 ± .1

ORL.randomfaceM504
xRF
87.75 ± .3
92.50 ± .2
95.50 ± .1
94.25 ± .1
96.00 ± .1

RF
77.50 ± .3
82.00 ± .7
84.50 ± .2
87.50 ± .2
86.00 ± .2

wsRF
87.00 ± .5
93.75 ± .2
93.75 ± .0
95.00 ± .1
95.50 ± .1

GRRF
87.25 ± .1
93.25 ± .1
94.50 ± .1
94.25 ± .1
95.50 ± .1

Table 4: AUC results (mean ± std-dev%) of random forest models against the number of trees on the CaltechM3000 and HorseM3000 subdatasets. The bold value in each row indicates the best result.

Dataset
Model
= 20
= 50
= 80
= 100
= 200

CaltechM3000
xRF
.995 ± .0
.999 ± .5
1.00 ± .2
1.00 ± .1
1.00 ± .1

RF
.851 ± .7
.817 ± .4
.826 ± 1.2
.865 ± .6
.864 ± 1

wsRF
.841 ± 1
.845 ± .8
.834 ± .7
.850 ± .8
.870 ± .9

GRRF
.846 ± .1
.860 ± .2
.862 ± .1
.908 ± .1
.923 ± .1

HorseM3000
xRF
.849 ± .1
.887 ± .0
.895 ± .0
.898 ± .0
.897 ± .0

RF
.637 ± .4
.664 ± .7
.692 ± 1.5
.696 ± .3
.733 ± .9

wsRF
.635 ± .8
.687 ± .4
.679 ± .6
.671 ± .4
.718 ± .9

GRRF
.786 ± .3
.778 ± .3
.785 ± .8
.699 ± .1
.806 ± .4

5.5. Results on Microarray Datasets

Table 5 shows the average test results in terms of accuracy of the 100 random forest models computed according to (9) on the gene datasets. The average number of genes selected by the xRF model, from 100 repetitions for each dataset, is shown on the right of Table 5, divided into two groups (strong) and (weak). These genes are used by the unbiased feature sampling method for growing trees in the xRF model. LASSO logistic regression, which uses the RF model as a classifier, showed fairly good accuracy on the two gene datasets srbct and leukemia. The GRRF model produced slightly better result on the prostate gene dataset. However, the xRF model produced the best accuracy on most cases of the remaining gene datasets.

Table 5: Test accuracy results (%) of random forest models, GRRF(0.1), varSelRF, and LASSO logistic regression, applied to gene datasets. The average results of 100 repetitions were computed; higher values are better. The number of genes in the strong group and the weak group is used in xRF.

Dataset
xRF
RF
wsRF
GRRF
varSelRF
LASSO

colon
87.65
84.35
84.50
86.45
76.80
82.00
245
317

srbct
97.71
95.90
96.76
97.57
96.50
99.30
606
546

Leukemia
89.25
82.58
84.83
87.25
89.30
92.40
502
200

Lymphoma
99.30
97.15
98.10
99.10
97.80
99.10
1404
275

breast.2.class
78.84
62.72
63.40
71.32
61.40
63.40
194
631

breast.3.class
65.42
56.00
57.19
63.55
58.20
60.00
724
533

nci
74.15
58.85
59.40
63.05
58.20
60.40
247
1345

Brain
81.93
70.79
70.79
74.79
76.90
74.10
1270
1219

Prostate
92.56
88.71
90.79
92.85
91.50
91.20
601
323

Adenocarcinoma
90.88
84.04
84.12
85.52
78.80
81.10
108
669

The detailed results containing the median and the variance values are presented in Figure 7 with box plots. Only the GRRF model was used for this comparison; the LASSO logistic regression and varSelRF method for feature selection were not considered in this experiment because their accuracies are lower than that of the GRRF model, as shown in [17]. We can see that the xRF model achieved the highest average accuracy of prediction on nine datasets out of ten. Its result was significantly different on the prostate gene dataset and the variance was also smaller than those of the other models.

Figure 7: Box plots of test accuracy of the models on the ten gene datasets.

Figure 8 shows the box plots of the () error bound of the RF, wsRF, and xRF models on the ten gene datasets from 100 repetitions. The wsRF model obtained lower error bound rate on five gene datasets out of 10. The xRF model produced a significantly different error bound rate on two gene datasets and obtained the lowest error rate on three datasets. This implies that when the optimal parameters such as and were used in growing trees, the number of genes in the subspace was not small and out-of-bag data was used in prediction, and the results were comparatively favored to the xRF model.

Figure 8: Box plots of () error bound for the models applied to the 10 gene datasets.

5.6. Comparison of Prediction Performance for Various Numbers of Features and Trees

Table 6 shows the average error bound and accuracy test results of 10 repetitions of random forest models on the three large datasets. The xRF model produced the lowest error on the dataset La1s, while the wsRF model showed the lower error bound on other two datasets Fbis and La2s. The RF model demonstrated the worst accuracy of prediction compared to the other models; this model also produced a large error when the small subspace size was used to build trees on the La1s and La2s datasets. The number of features in the and columns on the right of Table 6 was used in the xRF model. We can see that the xRF model achieved the highest accuracy of prediction on all three large datasets.

Table 6: The accuracy of prediction and error bound of the models using a small subspace ; better values are bold.

Dataset
Error bound
Test accuracy (%)

RF
wsRF
xRF
RF
GRRF
wsRF
xRF

Fbis
.2149
.1179
.1209
76.42
76.51
84.14
84.69
201
555

La2s
152.6
.0904
.0780
66.77
67.99
87.26
88.61
353
1136

La1s
40.8
.0892
.1499
77.76
80.49
86.03
87.21
220
1532

Figure 9 shows the plots of the performance curves of the RF models when the number of trees and features increases. The number of trees was increased stepwise by 20 trees from 20 to 200 when the models were applied to the La1s dataset. For the remaining data sets, the number of trees increased stepwise by 50 trees from 50 to 500. The number of random features in a subspace was set to . The number of features, each consisting of a random sum of five inputs, varied from 5 to 100, and for each, 200 trees were combined. The vertical line in each plot indicates the size of a subspace of features . This subspace was suggested by Breiman [1] for the case of low-dimensional datasets. Three feature selection methods, namely, GRRF, varSelRF, and LASSO, were not considered in this experiment. The main reason is that, when the value is large, the computational time of the GRRF and varSelRF models required to deal with large high datasets was too long [17].

(a) Fbis

(b) La2s

(c) La1s

Figure 9: The accuracy of prediction of the three random forests models against the number of trees and features on the three datasets.

It can be seen that the xRF and wsRF models always provided good results and achieved higher prediction accuracies when the subspace was used. However, the xRF model is better than the wsRF model in increasing the prediction accuracy on the three classification datasets. The RF model requires the larger number of features to achieve the higher accuracy of prediction, as shown in the right of Figures 9(a) and 9(b). When the number of trees in a forests was varied, the xRF model produced the best results on the Fbis and La2s datasets. In the La1s dataset where the xRF model did not obtain the best results, as shown in Figure 9(c) (left), the differences from the best results were minor. From the right of Figures 9(a), 9(b), and 9(c), we can observe that the xRF model does not need many features in the selected subspace to achieve the best prediction performance. These empirical results indicate that, for application on high-dimensional data, when the xRF model uses the small subspace, the achieved results can be satisfactory.

However, the RF model using the simple sampling method for feature selection [1] could achieve good prediction performance only if it is provided with a much larger subspace, as shown in the right part of Figures 9(a) and 9(b). Breiman suggested to use a subspace of size in classification problem. With this size, the computational time for building a random forest is still too high, especially for large high datasets. In general, when the xRF model is used with a feature subspace of the same size as the one suggested by Breiman, it demonstrates higher prediction accuracy and shorter computational time than those reported by Breiman. This achievement is considered to be one of the contributions in our work.

6. Conclusions

We have presented a new method for feature subspace selection for building efficient random forest xRF model for classification high-dimensional data. Our main contribution is to make a new approach for unbiased feature sampling, which selects the set of unbiased features for splitting a node when growing trees in the forests. Furthermore, this new unbiased feature selection method also reduces dimensionality using a defined threshold to remove uninformative features (or noise) from the dataset. Experimental results have demonstrated the improvements in increasing of the test accuracy and the AUC measures for classification problems, especially for image and microarray datasets, in comparison with recent proposed random forests models, including RF, GRRF, and wsRF.

For future work, we think it would be desirable to increase the scalability of the proposed random forests algorithm by parallelizing them on the cloud platform to deal with big data, that is, hundreds of millions of samples and features.

Conflict of Interests

The authors declare that there is no conflict of interests regarding the publication of this paper.

Acknowledgments

This research is supported in part by NSFC under Grant no. 61203294 and Hanoi-DOST under the Grant no. 01C-07/01-2012-2. The author Thuy Thi Nguyen is supported by the project “Some Advanced Statistical Learning Techniques for Computer Vision” funded by the National Foundation of Science and Technology Development, Vietnam, under the Grant no. 102.01-2011.17.

References

L. Breiman, “Random forests,” Machine Learning, vol. 450, no. 1, pp. 5–32, 2001.

L. Breiman, J. Friedman, C. J. Stone, and R. A. Olshen, Classification and Regression Trees, CRC Press, Boca Raton, Fla, USA, 1984.

H. Kim and W.-Y. Loh, “Classification trees with unbiased multiway splits,” Journal of the American Statistical Association, vol. 96, no. 454, pp. 589–604, 2001.

A. P. White and W. Z. Liu, “Technical note: bias in information-based measures in decision tree induction,” Machine Learning, vol. 15, no. 3, pp. 321–329, 1994.

T. G. Dietterich, “Experimental comparison of three methods for constructing ensembles of decision trees: bagging, boosting, and randomization,” Machine Learning, vol. 40, no. 2, pp. 139–157, 2000.

Y. Freund and R. E. Schapire, “A desicion-theoretic generalization of on-line learning and an application to boosting,” in Computational Learning Theory, pp. 23–37, Springer, 1995.

T.-T. Nguyen and T. T. Nguyen, “A real time license plate detection system based on boosting learning algorithm,” in Proceedings of the 5th International Congress on Image and Signal Processing (CISP '12), pp. 819–823, IEEE, October 2012.

T. K. Ho, “Random decision forests,” in Proceedings of the 3rd International Conference on Document Analysis and Recognition, vol. 1, pp. 278–282, 1995.

T. K. Ho, “The random subspace method for constructing decision forests,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 20, no. 8, pp. 832–844, 1998.

L. Breiman, “Bagging predictors,” Machine Learning, vol. 24, no. 2, pp. 123–140, 1996.

R. Díaz-Uriarte and S. Alvarez de Andrés, “Gene selection and classification of microarray data using random forest,” BMC Bioinformatics, vol. 7, article 3, 2006.

R. Genuer, J.-M. Poggi, and C. Tuleau-Malot, “Variable selection using random forests,” Pattern Recognition Letters, vol. 31, no. 14, pp. 2225–2236, 2010.

B. Xu, J. Z. Huang, G. Williams, Q. Wang, and Y. Ye, “Classifying very high-dimensional data with random forests built from small subspaces,” International Journal of Data Warehousing and Mining, vol. 8, no. 2, pp. 44–63, 2012.

Y. Ye, Q. Wu, J. Zhexue Huang, M. K. Ng, and X. Li, “Stratified sampling for feature subspace selection in random forests for high dimensional data,” Pattern Recognition, vol. 46, no. 3, pp. 769–787, 2013.

X. Chen, Y. Ye, X. Xu, and J. Z. Huang, “A feature group weighting method for subspace clustering of high-dimensional data,” Pattern Recognition, vol. 45, no. 1, pp. 434–446, 2012.

D. Amaratunga, J. Cabrera, and Y.-S. Lee, “Enriched random forests,” Bioinformatics, vol. 240, no. 18, pp. 2010–2014, 2008.

H. Deng and G. Runger, “Gene selection with guided regularized random forest,” Pattern Recognition, vol. 46, no. 12, pp. 3483–3489, 2013.

C. Strobl, “Statistical sources of variable selection bias in classification trees based on the gini index,” Tech. Rep. SFB 386, 2005, http://epub.ub.uni-muenchen.de/archive/00001789/01/paper_420.pdf.

C. Strobl, A.-L. Boulesteix, and T. Augustin, “Unbiased split selection for classification trees based on the gini index,” Computational Statistics & Data Analysis, vol. 520, no. 1, pp. 483–501, 2007.

C. Strobl, A.-L. Boulesteix, A. Zeileis, and T. Hothorn, “Bias in random forest variable importance measures: illustrations, sources and a solution,” BMC Bioinformatics, vol. 8, article 25, 2007.

C. Strobl, A.-L. Boulesteix, T. Kneib, T. Augustin, and A. Zeileis, “Conditional variable importance for random forests,” BMC Bioinformatics, vol. 9, no. 1, article 307, 2008.

T. Hothorn, K. Hornik, and A. Zeileis, Party: a laboratory for recursive partytioning, r package version 0.9-9999, 2011, http://cran.r-project.org/package=party

F. Wilcoxon, “Individual comparisons by ranking methods,” Biometrics, vol. 10, no. 6, pp. 80–83, 1945.

T.-T. Nguyen, J. Z. Huang, and T. T. Nguyen, “Two-level quantile regression forests for bias correction in range prediction,” Machine Learning, 2014.

T.-T. Nguyen, J. Z. Huang, K. Imran, M. J. Li, and G. Williams, “Extensions to quantile regression forests for very high-dimensional data,” in Advances in Knowledge Discovery and Data Mining, vol. 8444 of Lecture Notes in Computer Science, pp. 247–258, Springer, Berlin, Germany, 2014.

A. S. Georghiades, P. N. Belhumeur, and D. J. Kriegman, “From few to many: illumination cone models for face recognition under variable lighting and pose,” IEEE Transactions on Pattern Analysis and Machine Intelligence, vol. 23, no. 6, pp. 643–660, 2001.

F. S. Samaria and A. C. Harter, “Parameterisation of a stochastic model for human face identification,” in Proceedings of the 2nd IEEE Workshop on Applications of Computer Vision, pp. 138–142, IEEE, December 1994.

M. Turk and A. Pentland, “Eigenfaces for recognition,” Journal of Cognitive Neuroscience, vol. 3, no. 1, pp. 71–86, 1991.

H. Deng, “Guided random forest in the RRF package,” http://arxiv.org/abs/1306.0237

A. Liaw and M. Wiener, “Classification and regression by randomforest,” R News, vol. 20, no. 3, pp. 18–22, 2002.

R. Diaz-Uriarte, “varselrf: variable selection using random forests,” R package version 0.7-1, 2009, http://ligarto.org/rdiaz/Software/Software.html

J. H. Friedman, T. J. Hastie, and R. J. Tibshirani, “glmnet: Lasso and elastic-net regularized generalized linear models,” R package version , pages 1-1, 2010, http://CRAN.R-project.org/package=glmnet