Conditional variable importance for random forests

被引:2210
作者
Strobl, Carolin [1 ]
Boulesteix, Anne-Laure [2 ]
Kneib, Thomas [1 ]
Augustin, Thomas [1 ]
Zeileis, Achim [3 ]
机构
[1] Univ Munich, Dept Stat, D-80539 Munich, Germany
[2] Sylvia Lawry Ctr Multiple Sclerosis Res, D-81677 Munich, Germany
[3] Vienna Univ Econ & Business Adm, Dept Math Stat, A-1090 Vienna, Austria
关键词
D O I
10.1186/1471-2105-9-307
中图分类号
Q5 [生物化学];
学科分类号
071010 ; 081704 ;
摘要
Background: Random forests are becoming increasingly popular in many scientific fields because they can cope with "small n large p" problems, complex interactions and even highly correlated predictor variables. Their variable importance measures have recently been suggested as screening tools for, e. g., gene expression studies. However, these variable importance measures show a bias towards correlated predictor variables. Results: We identify two mechanisms responsible for this finding: (i) A preference for the selection of correlated predictors in the tree building process and (ii) an additional advantage for correlated predictor variables induced by the unconditional permutation scheme that is employed in the computation of the variable importance measure. Based on these considerations we develop a new, conditional permutation scheme for the computation of the variable importance measure. Conclusion: The resulting conditional variable importance reflects the true impact of each predictor variable more reliably than the original marginal approach.
引用
收藏
页数:11
相关论文
共 36 条
[1]  
[Anonymous], 2005, Permutation, Parametric and Bootstrap Tests of Hypotheses
[2]   Empirical characterization of random forest variable importance measures [J].
Archer, Kelfie J. ;
Kirnes, Ryan V. .
COMPUTATIONAL STATISTICS & DATA ANALYSIS, 2008, 52 (04) :2249-2260
[3]   An empirical comparison of voting classification algorithms: Bagging, boosting, and variants [J].
Bauer, E ;
Kohavi, R .
MACHINE LEARNING, 1999, 36 (1-2) :105-139
[4]   SmcHD1, containing a structural-maintenance-of-chromosomes hinge domain, has a critical role in X inactivation [J].
Blewitt, Marnie E. ;
Gendrel, Anne-Valerie ;
Pang, Zhenyi ;
Sparrow, Duncan B. ;
Whitelaw, Nadia ;
Craig, Jeffrey M. ;
Apedaile, Anwyn ;
Hilton, Douglas J. ;
Dunwoodie, Sally L. ;
Brockdorff, Neil ;
Kay, Graham F. ;
Whitelaw, Emma .
NATURE GENETICS, 2008, 40 (05) :663-669
[5]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[6]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[7]   Statistical modeling: The two cultures [J].
Breiman, L .
STATISTICAL SCIENCE, 2001, 16 (03) :199-215
[8]  
Breiman L, 1998, ANN STAT, V26, P801
[9]  
BREIMAN L, RANDOM FORESTS CLASS
[10]  
Bühlmann P, 2002, ANN STAT, V30, P927