Smaller, more accurate regression forests using tree alternating optimization
Arman Zharmagambetov, Miguel Á. Carreira-Perpiñán
Abstract
Regression forests, based on ensemble approaches such as bagging or boosting, have long been recognized as the leading off-the-shelf method for regression. However, forests rely on a greedy top-down procedure such as CART to learn each tree. We extend a recent algorithm for learning classification trees, Tree Alternating Optimization (TAO), to the regression case, and use it with bagging to construct regression forests of oblique trees, having hyperplane splits at the decision nodes. In a wide range of datasets, we show that the resulting forests exceed the accuracy of state-of-the-art algorithms such as random forests, AdaBoost or gradient boosting, often considerably, while yielding forests that have usually fewer and shallower trees and hence fewer parameters and faster inference overall. This result has an immense practical impact and advocates for the power of optimization in ensemble learning.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 96f981fe-9b09-463c-9a48-fded70a41d02Cited by top-tier papers13
- Counterfactual Explanations for Oblique Decision Trees: Exact, Efficient AlgorithmsMiguel Á. Carreira-Perpiñán, Suryabhan Singh HadaAAAI 2021 · 39 citations
- Feature Learning for Interpretable, Performant Decision TreesJack H. Good, Torin Kovach, Kyle Miller, Artur DubrawskiNeurIPS 2023 · 16 citations
- Pushing the Envelope of Gradient Boosting Forests via Globally-Optimized Oblique TreesMagzhan Gabidolla, Miguel Á. Carreira-PerpiñánCVPR 2022 · 13 citations
- Very Fast, Approximate Counterfactual Explanations for Decision ForestsMiguel Á. Carreira-Perpiñán, Suryabhan Singh HadaAAAI 2023 · 7 citations
- Tree in Tree: from Decision Trees to Decision GraphsBingzhao Zhu, Mahsa ShoaranNeurIPS 2021 · 5 citations
Related papers
- A faster training algorithm for regression trees with linear leaves, and an analysis of its complexityKuat Gazizov, Miguel Á. Carreira-PerpiñánNeurIPS 2025
- Towards Better Decision Forests: Forest Alternating OptimizationMiguel Á. Carreira-Perpiñán, Magzhan Gabidolla, Arman ZharmagambetovCVPR 2023
- Symbolic Regression Enhanced Decision Trees for Classification TasksKei Sen Fong, Mehul MotaniAAAI 2024 · 12 citations
- Softmax Tree: An Accurate, Fast Classifier When the Number of Classes Is LargeArman Zharmagambetov, Magzhan Gabidolla, Miguel Á. Carreira-PerpiñánEMNLP 2021 · 4 citations
- Jacobian Aligned Random ForestsSarwesh RauniyarICLR 2026
