Tree-based Methods

Complete List in Reverse Chronological Order

1. A. Agarwal, A. M.  Kenny, Y. S. Tan, T. M. Tang, B. Yu (2023). MDI+: a flexible random forest-based feature importance framework.  https://arxiv.org/abs/2307.01932 (PCS related)

2. Y. S. Tan, C. Singh, K. Nasseri, A. Agarwal,  J. Duncan, O. Ronen, M. Epland, A. Kornblith, B. Yu (2022).  Fast interpretable greedy-tree sums (FIGS).  https://arxiv.org/abs/2201.11931 (imodels 🔎: a python package for fitting interpretable models contains code for FIGS). 

3. M. Behr, Y. Wang, X. Li, B. Yu (2022). Provable Boolean Interaction Recovery from Tree Ensemble obtained via Random Forests.  PNAS,  https://arxiv.org/abs/2102.11800 (theory for a tractable version of iRF, PCS-related)

4. A. Agarwal, Y. S. Tan, O. Ronen, C. Singh, B. Yu (2022). Hierarchical shrinkage: improving accuracy and interpretability of tree-based methods. Proc. ICML https://arxiv.org/abs/2202.00858 ( imodels 🔎: a python package for fitting interpretable models contains code for hierarchical shrinkage (HS))

5. Y. Tan, A. Agarwal, and B. Yu (2021). A cautionary tale on fitting decision trees to data from additive models: generalization lower bounds. Proc. AISTATS, https://arxiv.org/abs/2110.09626

6. M. Behr*, K. Kumbier*, A. Cordova-Palomera, M. Aguirre, E. Ashley, A. Butte, R. Arnaout, J. B. Brown, J.  Preist*, B. Yu* (2020). Learning epistatic polygenic phenotypes with Boolean interactions  https://www.biorxiv.org/content/10.1101/2020.11.24.396846v1 (code) (PCS inference case study)

7. K. Kumbier, S. Sumanta, J. B. Brown, S. Celniker, and B. Yu* (2018) Refining interaction search through signed iterative Random Forests. https://arxiv.org/abs/1810.0728 (an enhanced version of iRF, PCS related)

8. S. Basu, K. Kumbier, J. B. Brown*, and B. Yu* (2018) iterative Random Forests to discover predictive and stable high-order interactions PNAS, 115 (8), 1943-1948. (code) (PCS related)