19
$\begingroup$

My Question: Why does random forest consider random subsets of features for splitting at the node level within each tree rather than at the tree level?

Background: This is something of a history question. Tin Kam Ho published this paper on constructing "decision forests" by randomly selecting a subset of features to use for growing each tree in 1998. Several years later, in 2001, Leo Breiman published his seminal Random Forest paper, wherein the feature subset is randomly selected at each node within each tree, not at each tree. While Breiman cited Ho, he did not specifically explain the move from tree-level to node-level random feature selection.

I'm wondering what specifically motivated this development. It seems that selecting the feature subset at the tree level would still accomplish the desired decorrelation of the trees.

My theory: I haven't seen this articulated elsewhere, but it seems like the random subspace method would be less efficient in terms of getting estimates of feature importance. To obtain estimates of variable importance, for each tree, the features are randomly permuted one by one, and the increase in misclassification or increase in error for the out-of-bag observations is recorded. The variables for which the misclassification or error increase resulting from this random permutation is high are those with the greatest importance.

If we use the random subspace method, for each tree, we are only considering $m$ of the $p$ features. It may take several trees to consider all $p$ predictors even once. On the other hand, if we consider a different subset $m_i$ of the $p$ features at each node, we will consider each feature more times after fewer trees, giving us a more robust estimate of the feature importance.

What I've looked at thus far: So far, I have read Breiman's paper and Ho's paper, and done a broad online search for comparisons of the methods without finding a definitive answer. Note that a similar question was asked before. This question goes a bit further by including my speculation/work toward a possible solution. I would be interested in any answers, relevant citations, or simulation studies comparing the two approaches. If none are forthcoming, I plan to run my own simulation comparing the two methods.

$\endgroup$
5
  • 2
    $\begingroup$ I won't cite any references, so let's just call this a comment. If you are trying to understand what variables are useful, it may be the case that a specific variable is critical, but only on a small piece of the data. You could find this out with bagging the variables at the node level. You would never discover this with bagging at the tree level. $\endgroup$
    – meh
    Commented Jul 19, 2018 at 17:01
  • 2
    $\begingroup$ I'm sure that Breiman has a comment related to this in his (imho) seminal paper, 'Statistics- The Two Cultures'. His point there is that sometimes the importance of a variable is masked by another variable. Bagging at the node level will allow one to see the what and when for a variable. $\endgroup$
    – meh
    Commented Jul 19, 2018 at 17:03
  • 1
    $\begingroup$ Thanks for the comments. Getting back to my idea about efficiency: suppose a pair of variables were related and, as you said, the importance of one "masked" the importance of another. If we built an RF predictor with enough trees and used the tree-level feature subsetting, would we not eventually have enough trees with the "masked" feature and without the "masking" feature to get at the importance of the former without the impact of the latter? I do think we are talking about roughly the same idea at least. Thanks! $\endgroup$
    – dlid
    Commented Jul 19, 2018 at 17:08
  • 4
    $\begingroup$ You might, but consider how many more trees you would have to build ! It's also not clear. Variable A might cause splits such that in none of them variable B will shine. It's just clearly intrinsically more robust to sample at the node level. To me, it relates to fundamentally what bootstrapping should be. $\endgroup$
    – meh
    Commented Jul 19, 2018 at 17:21
  • $\begingroup$ FIY - I have answered on the other question. Let me know if it helps, otherwise I can try and go into a deeper analysis with an answer here $\endgroup$
    – Davide ND
    Commented Sep 2, 2021 at 14:39

1 Answer 1

4
$\begingroup$

Suppose we have 10 features f1, f2, ..., f9, f10, then when we take a subset to let's suppose f1, f3, f4, f8 of features at tree level itself, then we construct the whole tree taking these 4 features into consideration.

We calculate the entropy, compare only these 4 features at every node and take that feature that yields maximum entropy. This isn't much use as we are restricting our tree learning to only those 4 features. Contrary to this, when we take some subset of features let's say f1, f8, f9 at first node, we calculate the entropy and compare them among these 3 features and chose the one that gives maximum value. Instead of growing the tree further with same features, we chose another subset of features let's say f4, f7, f2 and make the split based on these features. Suppose f8 was selected at the first node and f2 was selected at the second node. Model is able to learn the relationship between these both which wouldn't be possible if there is some other feature that gives maximum entropy than f2 after f8 has been selected as the root node.

In this way, the model can learn the relationship between different features in a more diversified way. This approach will have a number of features explored in a single tree and thus relations among them is preserved. Hope you got it now :)

$\endgroup$

Not the answer you're looking for? Browse other questions tagged or ask your own question.