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

Download:

Current browse context:

stat.ML

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 > Machine Learning

Title: Generating Compact Tree Ensembles via Annealing

Abstract: Tree ensembles are flexible predictive models that can capture relevant variables and to some extent their interactions in a compact and interpretable manner. Most algorithms for obtaining tree ensembles are based on versions of boosting or Random Forest. Previous work showed that boosting algorithms exhibit a cyclic behavior of selecting the same tree again and again due to the way the loss is optimized. At the same time, Random Forest is not based on loss optimization and obtains a more complex and less interpretable model. In this paper we present a novel method for obtaining compact tree ensembles by growing a large pool of trees in parallel with many independent boosting threads and then selecting a small subset and updating their leaf weights by loss optimization. We allow for the trees in the initial pool to have different depths which further helps with generalization. Experiments on real datasets show that the obtained model has usually a smaller loss than boosting, which is also reflected in a lower misclassification error on the test set.
Comments: Comparison with Random Forest included in the results section
Subjects: Machine Learning (stat.ML); Machine Learning (cs.LG)
Cite as: arXiv:1709.05545 [stat.ML]
  (or arXiv:1709.05545v4 [stat.ML] for this version)

Submission history

From: Yangzi Guo [view email]
[v1] Sat, 16 Sep 2017 18:26:18 GMT (1081kb,D)
[v2] Tue, 17 Oct 2017 03:03:27 GMT (1082kb,D)
[v3] Mon, 5 Feb 2018 18:27:57 GMT (1083kb,D)
[v4] Thu, 20 Feb 2020 03:25:08 GMT (6530kb,D)

Link back to: arXiv, form interface, contact.