Double random forest

被引:47
作者
Han, Sunwoo [1 ]
Kim, Hyunjoong [2 ]
Lee, Yung-Seop [3 ]
机构
[1] Fred Hutchinson Canc Res Ctr, Vaccine & Infect Dis Div, Seattle, WA 98006 USA
[2] Yonsei Univ, Dept Appl Stat, Seoul 03722, South Korea
[3] Dongguk Univ, Dept Stat, Seoul 04620, South Korea
基金
新加坡国家研究基金会;
关键词
Classification; Ensemble; Random forest; Bootstrap; Decision tree; CLASSIFICATION TREES; ALGORITHMS; ENSEMBLES;
D O I
10.1007/s10994-020-05889-1
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Random forest (RF) is one of the most popular parallel ensemble methods, using decision trees as classifiers. One of the hyper-parameters to choose from for RF fitting is the nodesize, which determines the individual tree size. In this paper, we begin with the observation that for many data sets (34 out of 58), the best RF prediction accuracy is achieved when the trees are grown fully by minimizing the nodesize parameter. This observation leads to the idea that prediction accuracy could be further improved if we find a way to generate even bigger trees than the ones with a minimum nodesize. In other words, the largest tree created with the minimum nodesize parameter may not be sufficiently large for the best performance of RF. To produce bigger trees than those by RF, we propose a new classification ensemble method called double random forest (DRF). The new method uses bootstrap on each node during the tree creation process, instead of just bootstrapping once on the root node as in RF. This method, in turn, provides an ensemble of more diverse trees, allowing for more accurate predictions. Finally, for data where RF does not produce trees of sufficient size, we have successfully demonstrated that DRF provides more accurate predictions than RF.
引用
收藏
页码:1569 / 1586
页数:18
相关论文
共 37 条
[21]  
Kerk C. J., 2003, J STAT ED, V11
[22]   Classification trees with bivariate linear discriminant node models [J].
Kim, H ;
Loh, WY .
JOURNAL OF COMPUTATIONAL AND GRAPHICAL STATISTICS, 2003, 12 (03) :512-530
[23]   Classification trees with unbiased multiway splits [J].
Kim, H ;
Loh, WY .
JOURNAL OF THE AMERICAN STATISTICAL ASSOCIATION, 2001, 96 (454) :589-604
[24]   A weight-adjusted voting algorithm for ensembles of classifiers [J].
Kim, Hyunjoong ;
Kim, Hyeuk ;
Moon, Hojin ;
Ahn, Hongshik .
JOURNAL OF THE KOREAN STATISTICAL SOCIETY, 2011, 40 (04) :437-449
[25]  
Larochelle H, 2012, J MACH LEARN RES, V13, P643
[26]   A comparison of prediction accuracy, complexity, and training time of thirty-three old and new classification algorithms [J].
Lim, TS ;
Loh, WY ;
Shih, YS .
MACHINE LEARNING, 2000, 40 (03) :203-228
[27]  
Lin Y., 2012, J AM STAT ASSOC, P578
[28]   IMPROVING THE PRECISION OF CLASSIFICATION TREES [J].
Loh, Wet-Yin .
ANNALS OF APPLIED STATISTICS, 2009, 3 (04) :1710-1737
[29]   Out-of-bag estimation of the optimal sample size in bagging [J].
Martinez-Munoz, Gonzalo ;
Suarez, Alberto .
PATTERN RECOGNITION, 2010, 43 (01) :143-152
[30]  
Mason L, 2000, ADV NEUR IN, V12, P512