On the overestimation of random forest's out-of-bag error

被引:153
作者
Janitza, Silke [1 ]
Hornung, Roman [1 ]
机构
[1] Univ Munich, Inst Med Informat Proc Biometry & Epidemiol, Munich, Germany
来源
PLOS ONE | 2018年 / 13卷 / 08期
关键词
PREDICTION; CLASSIFICATION; TUMOR; DISCOVERY; PATTERNS; IMPACT; CANCER;
D O I
10.1371/journal.pone.0201904
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
The ensemble method random forests has become a popular classification tool in bioinformatics and related fields. The out-of-bag error is an error estimation technique often used to evaluate the accuracy of a random forest and to select appropriate values for tuning parameters, such as the number of candidate predictors that are randomly drawn for a split, referred to as mtry. However, for binary classification problems with metric predictors it has been shown that the out-of-bag error can overestimate the true prediction error depending on the choices of random forests parameters. Based on simulated and real data this paper aims to identify settings for which this overestimation is likely. It is, moreover, questionable whether the out-of-bag error can be used in classification tasks for selecting tuning parameters like mtry, because the overestimation is seen to depend on the parameter mtry. The simulation-based and real-data based studies with metric predictor variables performed in this paper show that the overestimation is largest in balanced settings and in settings with few observations, a large number of predictor variables, small correlations between predictors and weak effects. There was hardly any impact of the overestimation on tuning parameter selection. However, although the prediction performance of random forests was not substantially affected when using the out-of-bag error for tuning parameter selection in the present studies, one cannot be sure that this applies to all future data. For settings with metric predictor variables it is therefore strongly recommended to use stratified subsampling with sampling fractions that are proportional to the class sizes for both tuning parameter selection and error estimation in random forests. This yielded less biased estimates of the true prediction error. In unbalanced settings, in which there is a strong interest in predicting observations from the smaller classes well, sampling the same number of observations from each class is a promising alternative.
引用
收藏
页数:31
相关论文
共 46 条
  • [1] Broad patterns of gene expression revealed by clustering analysis of tumor and normal colon tissues probed by oligonucleotide arrays
    Alon, U
    Barkai, N
    Notterman, DA
    Gish, K
    Ybarra, S
    Mack, D
    Levine, AJ
    [J]. PROCEEDINGS OF THE NATIONAL ACADEMY OF SCIENCES OF THE UNITED STATES OF AMERICA, 1999, 96 (12) : 6745 - 6750
  • [2] [Anonymous], ARXIV180209596
  • [3] [Anonymous], ARXIV08113619
  • [4] Correcting the Optimal Resampling-Based Error Rate by Estimating the Error Rate of Wrapper Algorithms
    Bernau, Christoph
    Augustin, Thomas
    Boulesteix, Anne-Laure
    [J]. BIOMETRICS, 2013, 69 (03) : 693 - 702
  • [5] Ten Simple Rules for Reducing Overoptimistic Reporting in Methodological Computational Research
    Boulesteix, Anne-Laure
    [J]. PLOS COMPUTATIONAL BIOLOGY, 2015, 11 (04)
  • [6] Overview of random forest methodology and practical guidance with emphasis on computational biology and bioinformatics
    Boulesteix, Anne-Laure
    Janitza, Silke
    Kruppa, Jochen
    Koenig, Inke R.
    [J]. WILEY INTERDISCIPLINARY REVIEWS-DATA MINING AND KNOWLEDGE DISCOVERY, 2012, 2 (06) : 493 - 507
  • [7] Random forest Gini importance favours SNPs with large minor allele frequency: impact, sources and recommendations
    Boulesteix, Anne-Laure
    Bender, Andreas
    Bermejo, Justo Lorenzo
    Strobl, Carolin
    [J]. BRIEFINGS IN BIOINFORMATICS, 2012, 13 (03) : 292 - 304
  • [8] Random forests
    Breiman, L
    [J]. MACHINE LEARNING, 2001, 45 (01) : 5 - 32
  • [9] Random forests
    Breiman, L
    [J]. MACHINE LEARNING, 2001, 45 (01) : 5 - 32
  • [10] Breiman L., 1996, OUT OF BAG ESTIMATIO