arXiv:2607.13550stat.MLcs.LG2026-07

提出并行梯度提升,高效预测条件分布

Parallel gradient boosting for flexible estimation of conditional distributions

论文配图:Parallel gradient boosting for flexible estimation of conditional distributions
图 1 · 摘自论文原文
  • 用统一下降方向减少每轮训练模型数,仅需一个基模型
  • 多分位数回归下速度比XGBoost快数个数量级,效果相当
  • 适合高维、含混合或缺失变量的条件分布建模

梯度提升在分类与回归任务中表现优异,其扩展至多输出预测问题近年来应用日益广泛。其中,预测整个条件分布(而非单一函数)可视为多输出回归问题,如多分位数回归。传统梯度提升在处理此类问题时计算成本高,因每轮需为每个目标训练一个基模型。现有高效变体通常依赖特定损失函数和基学习器(如决策树)。本文提出一种新算法——并行梯度提升,通过为所有训练样本使用共同下降方向,使每轮仅需一个基模型,显著提升效率。我们建立了算法收敛的充分条件,并以多分位数回归为例展示其实际应用。实验表明,在该设置下,其预测质量接近当前最优的XGBoost等库,但速度提升数个数量级。进一步评估显示,所得条件分布估计器在高维场景及存在混合类型或缺失协变量时,显著优于其他非参数与半参数方法。

原文摘要 · Abstract (English)

Boosting is one of the most successful learning techniques for standard classification and regression tasks. Its extension to multi-output prediction problems has found an increasing number of applications in recent years. Among them is the prediction of entire conditional distributions rather than single functionals, which can often be framed as a multi-output regression problem, for example multiple quantile regression. Addressing such problems with classical implementations of boosting is computationally challenging, because usually one base model is trained for each target at every iteration. More efficient variants of boosting have been proposed to speed up training, but they tend to be tied to specific loss functions and classes of base learners, usually decision trees. In this work, we study a modification of the gradient boosting algorithm, which we call parallel gradient boosting, designed to circumvent all these limitations. The core idea is to use a common descent direction for all training observations. By doing so, only one base model is needed at each iteration, regardless of the number of targets, which allows for considerable performance gains. We establish sufficient conditions for the convergence of the algorithm, whose practical use is introduced via the multiple quantile regression setting. We show that in such a setting, it provides predictions of similar quality to state-of-the-art boosting libraries such as XGBoost, while being faster by several orders of magnitude. Then, we evaluate the properties of the resulting conditional distribution estimator, which is shown empirically to outperform other nonparametric and semiparametric estimators, especially in high-dimensional settings and in the presence of mixed and/or missing covariates.

梯度提升条件分布多输出量化

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