Provable Boolean interaction recovery from tree ensemble obtained via random forests.
Provable Boolean interaction recovery from tree ensemble obtained via random forests.
复制标题
DOI:
10.1073/pnas.2118636119
复制
发表时间:
2022-05-31
影响因子:
11.1
通讯作者:
中科院分区:
文献类型:
--
作者:
Random Forests (RFs) are among the most successful machine-learning algorithms in terms of prediction accuracy. In many domain problems, however, the primary goal is not prediction, but to understand the data-generation process—in particular, finding important features and feature interactions. There exists strong empirical evidence that RF-based methods—in particular, iterative RF (iRF)—are very successful in terms of detecting feature interactions. In this work, we propose a biologically motivated, Boolean interaction model. Using this model, we complement the existing empirical evidence with theoretical evidence for the ability of iRF-type methods to select desirable interactions. Our theoretical analysis also yields deeper insights into the general interaction selection mechanism of decision-tree algorithms and the importance of feature subsampling. Random Forests (RFs) are at the cutting edge of supervised machine learning in terms of prediction performance, especially in genomics. Iterative RFs (iRFs) use a tree ensemble from iteratively modified RFs to obtain predictive and stable nonlinear or Boolean interactions of features. They have shown great promise for Boolean biological interaction discovery that is central to advancing functional genomics and precision medicine. However, theoretical studies into how tree-based methods discover Boolean feature interactions are missing. Inspired by the thresholding behavior in many biological processes, we first introduce a discontinuous nonlinear regression model, called the “Locally Spiky Sparse” (LSS) model. Specifically, the LSS model assumes that the regression function is a linear combination of piecewise constant Boolean interaction terms. Given an RF tree ensemble, we define a quantity called “Depth-Weighted Prevalence” (DWP) for a set of signed features . Intuitively speaking, DWP() measures how frequently features in appear together in an RF tree ensemble. We prove that, with high probability, DWP() attains a universal upper bound that does not involve any model coefficients, if and only if corresponds to a union of Boolean interactions under the LSS model. Consequentially, we show that a theoretically tractable version of the iRF procedure, called LSSFind, yields consistent interaction discovery under the LSS model as the sample size goes to infinity. Finally, simulation results show that LSSFind recovers the interactions under the LSS model, even when some assumptions are violated.
登录
查看更多内容
影响因子:
4.4
作者:
Chen, Xi;Ishwaran, Hemant
通讯作者:
Ishwaran, Hemant
影响因子:
5.4
作者:
Levine, Erel;Hwa, Terence
通讯作者:
Hwa, Terence
影响因子:
1.6
作者:
Janitza, Silke;Celik, Ender;Boulesteix, Anne-Laure
通讯作者:
Boulesteix, Anne-Laure
影响因子:
13.8
作者:
Ferrell, JE
通讯作者:
Ferrell, JE
DOI:
10.1073/pnas.0500670102
发表时间:
2005-03-22
影响因子:
11.1
作者:
Kobiler, O;Rokney, A;Oppenheim, AB
通讯作者:
Oppenheim, AB