arXiv:2411.04394stat.MLcs.DS2024-11被引 2

揭示贪婪递归划分模型的统计-计算权衡,解释为何在高维稀疏回归中表现差异大。

Statistical-Computational Trade-offs for Recursive Adaptive Partitioning Estimators

  • 用随机过程理论分析贪婪划分算法的机制
  • 不满足MSP时需指数级样本才能低误差,满足时仅需对数级样本
  • 适合关注高维建模计算效率与理论边界的研究者

基于递归自适应划分(如决策树及其集成)的模型因其潜在避免维度灾难的能力,在高维回归中广受欢迎。由于经验风险最小化(ERM)计算不可行,这类模型通常采用贪心算法训练。尽管在许多情况下有效,但实证发现其常陷入局部最优。本文研究在d个二值特征上的稀疏回归函数学习问题,发现当真实回归函数f*不满足Abbe等人(2022)提出的合并阶梯性质(MSP)时,贪心训练需exp(Ω(d))样本才能实现低估计误差;而当f*满足MSP时,仅需O(log d)样本即可达到小误差。这一二分现象与两层神经网络在均场区域下用随机梯度下降(SGD)训练的表现相似,从而建立了SGD训练神经网络与贪心递归划分估计器之间的直接对比。此外,通过ERM训练的递归划分估计器无论f*是否满足MSP,均可在O(log d)样本下实现低误差,揭示了贪心训练的统计-计算权衡。证明基于对贪心递归划分的新随机过程解释及一种可能具有独立兴趣的耦合技术。

原文摘要 · Abstract (English)

Models based on recursive adaptive partitioning such as decision trees and their ensembles are popular for high-dimensional regression as they can potentially avoid the curse of dimensionality. Because empirical risk minimization (ERM) is computationally infeasible, these models are typically trained using greedy algorithms. Although effective in many cases, these algorithms have been empirically observed to get stuck at local optima. We explore this phenomenon in the context of learning sparse regression functions over $d$ binary features, showing that when the true regression function $f^*$ does not satisfy Abbe et al. (2022)'s Merged Staircase Property (MSP), greedy training requires $\exp(Ω(d))$ to achieve low estimation error. Conversely, when $f^*$ does satisfy MSP, greedy training can attain small estimation error with only $O(\log d)$ samples. This dichotomy mirrors that of two-layer neural networks trained with stochastic gradient descent (SGD) in the mean-field regime, thereby establishing a head-to-head comparison between SGD-trained neural networks and greedy recursive partitioning estimators. Furthermore, ERM-trained recursive partitioning estimators achieve low estimation error with $O(\log d)$ samples irrespective of whether $f^*$ satisfies MSP, thereby demonstrating a statistical-computational trade-off for greedy training. Our proofs are based on a novel interpretation of greedy recursive partitioning using stochastic process theory and a coupling technique that may be of independent interest.

机器学习统计学习递归划分计算复杂性

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