跳到正文
原文
arXiv:cs.LG(机器学习,全量分类)· Chenxuanyin Zou, Jiayang Ren, Qiangqiang Mao, Jing Liu, Marcus Lai, Yankai Cao·· 5 小时前AI 评分31

面向深度分类树的移动视界近似分支归约方法

A Moving-Horizon Approximate Branch-and-Reduce Method for Deep Classification Trees

AI 导读

该论文提出一种移动视界近似分支归约方法,可在含连续特征的大规模数据集上训练近最优深度分类树。方法基于分层根-子树优化框架,根层问题用分支归约求解,子树问题用贪心启发式近似,该近似相当于强化学习中的前瞻 rollout,显著提升深层结构效率,再以低成本移动视界策略迭代精化精度。数值结果显示其测试精度超过现有启发式基线,在数据集规模和树深度上的可扩展性均明显优于全局最优求解器。

正文

View PDF HTML (experimental)

Abstract:Despite the importance for interpretability, decision trees face severe scalability challenges. Existing global optimal methods are often limited by binary feature selection and shallow tree depths, whereas traditional heuristic approaches frequently sacrifice predictive accuracy. To overcome these limitations, this paper proposes a moving-horizon approximate branch-and-reduce method to train near-optimal deep classification trees on large-scale datasets with continuous features. Built on a hierarchical root-subtree optimization framework, the method solves the root-level problem via branch-and-reduce while approximating the induced subtree problem using greedy heuristics. Although the underlying framework is capable of guaranteeing global optimality, the approximation, which functions as a lookahead rollout in a reinforcement learning context, significantly boosts efficiency for deeper structures. A low-cost moving-horizon strategy is then employed to iteratively refine model accuracy. Extensive numerical results demonstrate that our method exceeds the testing accuracy of existing heuristic baselines while offering significantly greater scalability, in terms of both dataset size and tree depth, than global optimal solvers.
Comments: J2C Certification
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Optimization and Control (math.OC)
Cite as: arXiv:2609.38194 [cs.LG]
  (or arXiv:2609.38194v1 [cs.LG] for this version)
  https://doi.org/10.48550/arXiv.2609.38194

arXiv-issued DOI via DataCite

Journal reference: Transactions on Machine Learning Research, August 2026

Submission history

From: Chenxuanyin Zou [view email]
[v1] Fri, 18 Sep 2026 05:50:36 UTC (763 KB)

来源:arXiv:cs.LG(机器学习,全量分类) · arxiv.org