arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2512.08306cs.LG
收藏

雅可比对齐随机森林

Jacobian Aligned Random Forests

Department of Applied Mathematics and Statistics(应用数学与统计学系)

Sarwesh Rauniyar

AI总结:

雅可比对齐随机森林通过特征空间旋转提升轴对齐森林的性能,同时保持简单性和鲁棒性。

摘要:

轴对齐决策树速度快且稳定,但在具有旋转或交互依赖决策边界的数据显示集上表现不佳,其中信息性分割需要特征的线性组合而非单特征阈值。Oblique forests通过每个节点超平面分割来解决这个问题,但增加了计算成本和实现复杂性。我们提出了一种简单替代方法:JARF,雅可比对齐随机森林。具体来说,我们首先拟合一个轴对齐森林来估计类别概率或回归输出,计算这些预测相对于每个特征的有限差分梯度,将它们聚合为一个期望雅可比外积,这扩展了期望梯度外积(EGOP),并将其作为所有输入的单一全局线性预条件器。这个监督预条件器对特征空间应用单个全局旋转,然后将变换后的数据返回到标准轴对齐森林中,保留现成的训练流程,同时捕捉到斜边界和特征交互,这些交互通常需要许多轴对齐分割来近似。相同的构造适用于任何提供梯度的模型,尽管我们在这项工作中专注于随机森林和梯度提升树。在表格分类和回归基准上,这种预处理一致地提高了轴对齐森林的性能,并且经常与Oblique基线匹配或超过,同时提高了训练时间。我们的实验结果和理论分析共同表明,监督预处理可以恢复Oblique森林的大部分准确性,同时保留轴对齐树的简单性和鲁棒性。

英文摘要:

Axis-aligned decision trees are fast and stable but struggle on datasets with rotated or interaction-dependent decision boundaries, where informative splits require linear combinations of features rather than single-feature thresholds. Oblique forests address this with per-node hyperplane splits, but at added computational cost and implementation complexity. We propose a simple alternative: JARF, Jacobian-Aligned Random Forests. Concretely, we first fit an axis-aligned forest to estimate class probabilities or regression outputs, compute finite-difference gradients of these predictions with respect to each feature, aggregate them into an expected Jacobian outer product that generalizes the expected gradient outer product (EGOP), and use it as a single global linear preconditioner for all inputs. This supervised preconditioner applies a single global rotation of the feature space, then hands the transformed data back to a standard axis-aligned forest, preserving off-the-shelf training pipelines while capturing oblique boundaries and feature interactions that would otherwise require many axis-aligned splits to approximate. The same construction applies to any model that provides gradients, though we focus on random forests and gradient-boosted trees in this work. On tabular classification and regression benchmarks, this preconditioning consistently improves axis-aligned forests and often matches or surpasses oblique baselines while improving training time. Our experimental results and theoretical analysis together indicate that supervised preconditioning can recover much of the accuracy of oblique forests while retaining the simplicity and robustness of axis-aligned trees.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑