Can a Single Tree Outperform an Entire Forest?

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Mao, Qiangqiang, Cao, Yankai
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912133761990656
author Mao, Qiangqiang
Cao, Yankai
author_facet Mao, Qiangqiang
Cao, Yankai
contents The prevailing mindset is that a single decision tree underperforms classic random forests in testing accuracy, despite its advantages in interpretability and lightweight structure. This study challenges such a mindset by significantly improving the testing accuracy of an oblique regression tree through our gradient-based entire tree optimization framework, making its performance comparable to the classic random forest. Our approach reformulates tree training as a differentiable unconstrained optimization task, employing a scaled sigmoid approximation strategy. To ameliorate numerical instability, we propose an algorithmic scheme that solves a sequence of increasingly accurate approximations. Additionally, a subtree polish strategy is implemented to reduce approximation errors accumulated across the tree. Extensive experiments on 16 datasets demonstrate that our optimized tree outperforms the classic random forest by an average of $2.03\%$ improvements in testing accuracy.
format Preprint
id arxiv_https___arxiv_org_abs_2411_17003
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Can a Single Tree Outperform an Entire Forest?
Mao, Qiangqiang
Cao, Yankai
Machine Learning
Artificial Intelligence
The prevailing mindset is that a single decision tree underperforms classic random forests in testing accuracy, despite its advantages in interpretability and lightweight structure. This study challenges such a mindset by significantly improving the testing accuracy of an oblique regression tree through our gradient-based entire tree optimization framework, making its performance comparable to the classic random forest. Our approach reformulates tree training as a differentiable unconstrained optimization task, employing a scaled sigmoid approximation strategy. To ameliorate numerical instability, we propose an algorithmic scheme that solves a sequence of increasingly accurate approximations. Additionally, a subtree polish strategy is implemented to reduce approximation errors accumulated across the tree. Extensive experiments on 16 datasets demonstrate that our optimized tree outperforms the classic random forest by an average of $2.03\%$ improvements in testing accuracy.
title Can a Single Tree Outperform an Entire Forest?
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2411.17003