We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

stat.ME

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Statistics > Methodology

Title: Linear Aggregation in Tree-based Estimators

Abstract: Regression trees and their ensemble methods are popular methods for nonparametric regression: they combine strong predictive performance with interpretable estimators. To improve their utility for locally smooth response surfaces, we study regression trees and random forests with linear aggregation functions. We introduce a new algorithm that finds the best axis-aligned split to fit linear aggregation functions on the corresponding nodes, and we offer a quasilinear time implementation. We demonstrate the algorithm's favorable performance on real-world benchmarks and in an extensive simulation study, and we demonstrate its improved interpretability using a large get-out-the-vote experiment. We provide an open-source software package that implements several tree-based estimators with linear aggregation functions.
Subjects: Methodology (stat.ME); Applications (stat.AP); Computation (stat.CO)
Cite as: arXiv:1906.06463 [stat.ME]
  (or arXiv:1906.06463v6 [stat.ME] for this version)

Submission history

From: Theo Saarinen [view email]
[v1] Sat, 15 Jun 2019 04:25:55 GMT (572kb,D)
[v2] Tue, 18 Jun 2019 20:00:42 GMT (572kb,D)
[v3] Mon, 11 Jan 2021 23:17:20 GMT (3803kb,D)
[v4] Wed, 13 Jan 2021 20:45:03 GMT (3812kb,D)
[v5] Sat, 23 Jan 2021 02:18:06 GMT (3813kb,D)
[v6] Thu, 9 Sep 2021 18:40:38 GMT (4500kb,D)

Link back to: arXiv, form interface, contact.