To tune or not to tune the number of trees in random forest?

484 citations published

Abstract

The number of trees T in the random forest (RF) algorithm for supervised learning has to be set by the user. It is controversial whether T should simply be set to the largest computationally manageable value or whether a smaller T may in some cases be better. While the principle underlying bagging is that "more trees are better", in practice the classification error rate sometimes reaches a minimum before increasing again for increasing number of trees. The goal of this paper is four-fold: (i) providing theoretical results showing that the expected error rate may be a non-monotonous function of the number of trees and explaining under which circumstances this happens; (ii) providing theoretical results showing that such non-monotonous patterns cannot be observed for other performance measures such as the Brier score and the logarithmic loss (for classification) and the mean squared error (for regression); (iii) illustrating the extent of the problem through an application to a large number (n = 306) of datasets from the public database OpenML; (iv) finally arguing in favor of setting it to a computationally feasible large number, depending on convergence properties of the desired performance measure.

Cite this paper

@article{probst2017,
  title = {To tune or not to tune the number of trees in random forest?},
  author = {Philipp Probst and Anne-Laure Boulesteix},
  journal = {Journal of Machine Learning Research 18 (2018) 1-18},
  year = {2017},
  month = {May},
  doi = {10.48550/arxiv.1705.05654},
  url = {https://doi.org/10.48550/arxiv.1705.05654},
}