Random Planted Forest

With Enno Mammen (Heidelberg University)

Random Planted Forest: a directly interpretable tree ensemble

We introduce a novel interpretable and tree-based algorithm for prediction in a regression setting in which each tree in a classical random forest is replaced by a family of planted trees that grow simultaneously.
The motivation for our algorithm is to estimate the unknown regression function from a functional ANOVA decomposition perspective, where each tree corresponds to a function within that decomposition.
Therefore, planted trees are limited in the number of interaction terms. The maximal order of approximation in the ANOVA decomposition can be specified or left unlimited. If a first order approximation is chosen, the result is an additive model. In the other extreme case, if the order of approximation is not limited, the resulting model places no restrictions on the form of the regression function. In a simulation study we find encouraging prediction and visualisation properties of our random planted forest method.
We also develop theory for an idealised version of random planted forests. In particular, for an additive model we show that the idealised version achieves asymptotically optimal one-dimensional convergence rates of order $n^{-2/5}$ up to a logarithmic factor. The talk reports on joint work with Munir Hiabu (Copenhagen) and Joseph Theo Meyer (Heidelberg).

Add to your calendar or Include in your list