arXiv:2503.14381stat.MLcs.LG2025-03

通过迭代优化提升高维斜向切分的决策树性能。

Optimizing High-Dimensional Oblique Splits

  • 用迭代精炼法优化高维斜向切分,替代单步优化。
  • 增大稀疏参数可捕捉复杂数据结构,但需更多计算资源。
  • 适合处理复杂数据的决策树集成模型,如随机森林改进版。

研究表明,斜向切分能显著提升决策树性能。本文研究了高维斜向切分的优化问题,提出考虑 $s_0$-稀疏切分的充分不纯度下降(SID)收敛性。随着 $s_0$ 增大,SID 函数类扩大,使模型能捕捉如 $s_0$-维异或函数等复杂数据生成过程,$s_0$ 反映潜在数据函数的未知复杂度。同时,学习这些复杂函数需更大计算开销,凸显统计精度(依赖 $s_0$ 的 SID 函数类大小)与计算成本之间的根本权衡。尤其在难题中,候选切分集可能过大,导致标准集成方法不可行。为此,我们提出渐进树,通过迭代精炼优化斜向切分,并与传统正交切分结合,融入随机森林等集成模型以提升有限样本表现。实验验证表明,该方法在模拟和真实数据上持续优于多种现有斜向树模型。

原文摘要 · Abstract (English)

Evidence suggests that oblique splits can significantly enhance the performance of decision trees. This paper explores the optimization of high-dimensional oblique splits for decision tree construction, establishing the Sufficient Impurity Decrease (SID) convergence that takes into account $s_0$-sparse oblique splits. We demonstrate that the SID function class expands as sparsity parameter $s_0$ increases, enabling the model to capture complex data-generating processes such as the $s_0$-dimensional XOR function. Thus, $s_0$ represents the unknown potential complexity of the underlying data-generating function. Furthermore, we establish that learning these complex functions necessitates greater computational resources. This highlights a fundamental trade-off between statistical accuracy, which is governed by the $s_0$-dependent size of the SID function class, and computational cost. Particularly, for challenging problems, the required candidate oblique split set can become prohibitively large, rendering standard ensemble approaches computationally impractical. To address this, we propose progressive trees that optimize oblique splits through an iterative refinement process rather than a single-step optimization. These splits are integrated alongside traditional orthogonal splits into ensemble models like Random Forests to enhance finite-sample performance. The effectiveness of our approach is validated through simulations and real-data experiments, where it consistently outperforms various existing oblique tree models.

决策树斜向切分迭代优化随机森林

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。