揭示了XGBoost实际学习的函数类型及其理论优势。
What Functions Does XGBoost Learn?
- 构建无限维函数类,刻画XGBoost隐含学习的函数空间。
- 证明其估计器收敛率接近最优,避免维度灾难。
- 首次严谨解析XGBoost与变差理论的关系,适合算法研究者。
本文为XGBoost隐式学习的函数类建立了严格的理论基础,弥合了其经验成功与理论理解之间的差距。我们引入了一个无限维函数类 $\mathcal{F}^{d, s}_{\infty-\text{ST}}$,扩展了有界深度回归树的有限集成,并定义了一个复杂度度量 $V^{d, s}_{\infty-\text{XGB}}(\cdot)$,推广了XGBoost中使用的$L^1$正则化惩罚。我们证明,任何XGBoost目标函数的优化器也是在 $\mathcal{F}^{d, s}_{\infty-\text{ST}}$ 上带有惩罚 $V^{d, s}_{\infty-\text{XGB}}(\cdot)$ 的等价正则化回归问题的优化器,表明XGBoost实际上是在追求更广的函数类。我们还基于光滑性,用Hardy--Krause变差对 $\mathcal{F}^{d, s}_{\infty-\text{ST}}$ 和 $V^{d, s}_{\infty-\text{XGB}}(\cdot)$ 进行解释。证明在 $\{f \in \mathcal{F}^{d, s}_{\infty-\text{ST}}: V^{d, s}_{\infty-\text{XGB}}(f) \le V\}$ 上的最小二乘估计器达到几乎极小极大最优收敛率 $n^{-2/3} (\log n)^{4(\min(s, d) - 1)/3}$,从而避免了维度灾难。结果首次严格刻画了支撑XGBoost的函数空间,阐明其与经典变差概念的联系,并指出一个关键开放问题:XGBoost算法本身是否在此类上实现极小极大最优性。
原文摘要 · Abstract (English)
This paper establishes a rigorous theoretical foundation for the function class implicitly learned by XGBoost, bridging the gap between its empirical success and our theoretical understanding. We introduce an infinite-dimensional function class $\mathcal{F}^{d, s}_{\infty-\text{ST}}$ that extends finite ensembles of bounded-depth regression trees, together with a complexity measure $V^{d, s}_{\infty-\text{XGB}}(\cdot)$ that generalizes the $L^1$ regularization penalty used in XGBoost. We show that every optimizer of the XGBoost objective is also an optimizer of an equivalent penalized regression problem over $\mathcal{F}^{d, s}_{\infty-\text{ST}}$ with penalty $V^{d, s}_{\infty-\text{XGB}}(\cdot)$, providing an interpretation of XGBoost as implicitly targeting a broader function class. We also develop a smoothness-based interpretation of $\mathcal{F}^{d, s}_{\infty-\text{ST}}$ and $V^{d, s}_{\infty-\text{XGB}}(\cdot)$ in terms of Hardy--Krause variation. We prove that the least squares estimator over $\{f \in \mathcal{F}^{d, s}_{\infty-\text{ST}}: V^{d, s}_{\infty-\text{XGB}}(f) \le V\}$ achieves a nearly minimax-optimal rate of convergence $n^{-2/3} (\log n)^{4(\min(s, d) - 1)/3}$, thereby avoiding the curse of dimensionality. Our results provide the first rigorous characterization of the function space underlying XGBoost, clarify its connection to classical notions of variation, and identify an important open problem: whether the XGBoost algorithm itself achieves minimax optimality over this class.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。