xgboost: 分割查找算法:贪婪算法、分桶算法
创始人
2024-05-28 18:33:32
0

1、Basic Exact Greedy Algorithm

树学习的关键问题之一是找到最好的分割,如Eq(7)所示。

贪婪算法:分割查找算法枚举所有特征上的所有可能的分割。精确的贪婪算法如Alg. 1所示。为了高效地完成这一任务,算法必须首先根据特征值对数据进行排序,并按排序顺序访问数据,积累Eq(7)中结构得分的梯度统计量。现有的大多数单个树提升实现都支持精确贪婪算法,如scikit-learn[20]、R的gbm[21]以及XGBoost的单机版本。

Eq(7)在文章:xgboost:算法数学原理_KPer_Yang的博客-CSDN博客
Lsplit=12[(∑i∈ILgi)2∑i∈ILhi+λ+(∑i∈IRgi)2∑i∈IRhi+λ−(∑i∈Igi)2∑i∈Ihi+λ]−γ\mathcal{L}_{split}=\frac{1}{2}\left[\frac{(\sum_{i\in I_L}g_i)^2}{\sum_{i\in I_L}h_i+\lambda}+\frac{(\sum_{i\in I_R}g_i)^2}{\sum_{i\in I_R}h_i+\lambda}-\frac{(\sum_{i\in I}g_i)^2}{\sum_{i\in I}h_i+\lambda}\right]-\gamma Lsplit​=21​[∑i∈IL​​hi​+λ(∑i∈IL​​gi​)2​+∑i∈IR​​hi​+λ(∑i∈IR​​gi​)2​−∑i∈I​hi​+λ(∑i∈I​gi​)2​]−γ
算法的流程如下所示:

在这里插入图片描述

mmm:特征的维度;

sorted(I,byxjk)sorted(I, by \ x_{jk})sorted(I,by xjk​):在特征kkk下的样本特征值进行排序,排序后按照划分点进行特征值划分,计算score。

2、 Approximate Algorithm

精确的贪婪算法非常强大,因为它贪婪地枚举了所有可能的分裂点。然而,当数据太大不能全部放入内存时,全部枚举不能做到。同样的问题也出现在分布式环境中。为了在这两种情况下支持有效的梯度树增强,需要一个近似算法。

**在Alg. 2中,首先根据特征分布的百分位数提出候选分裂点(具体标准将在第3.3节给出)。然后,该算法将连续特征映射到由这些候选点分割的桶中,汇总统计数据,并根据汇总的统计数据在分桶后的数据中找到最佳解决方案。**该算法有两种变体,这取决于给出分桶的时间。在合理的近似水平下,分位数策略可以获得与精确贪婪相同的精度。

  • 全局分桶方法在树构造的初始阶段提出所有候选分割,并在所有级别上使用相同的分割查找分桶。

  • 局部分桶在每次分裂后重新分桶。

在这里插入图片描述

Gkv←=∑j∈{j∣sk,v≥xjk>sk,v−1}gjG_{kv}\leftarrow=\sum_{j\in\{j\mid s_{k,v}\geq\mathbf{x}_{jk}>s_{k,v-1}\}}g_jGkv​←=∑j∈{j∣sk,v​≥xjk​>sk,v−1​}​gj​:按照每个桶计算GGG.

Hkv←=∑j∈{j∣sk,v≥xjk>sk,v−1}hjH_{kv}\leftarrow=\sum_{j\in\{j|s_{k,v}\geq\mathbf{x}_{jk}>s_{k,v-1}\}}h_jHkv​←=∑j∈{j∣sk,v​≥xjk​>sk,v−1​}​hj​:按照每个桶计算HHH.

注:个人理解,分桶其实相当于分裂时就按照每个桶分到左右分支,按照每个桶计算GGG和HHH,而不是一个个样本计算,遍历所有的特征值寻找分裂点。

对比:希格斯10M数据集的AUC收敛性比较。eps参数对应于近似草图的精度。这大致相当于分成1 / eps个桶。相同AUC的情况下,局部分桶需要更少的桶,因为它细化了分裂的候选数据;

在这里插入图片描述

相关内容

热门资讯

自治区财政下达高新技术企业、“... 转自:草原云春潮涌动,助企正当时。申报高新技术企业获得补助,提高了企业应对市场风险、提高发展质量和水...
5.12致敬白衣天使 转自:北京时间 【#5.12致敬白衣天使#】#她是个护士...
5分钟 飙涨近50% 今日早盘,港股三大指数高开高走,恒生科技指数一度涨近3%。截至午间收盘,恒生指数涨0.93%,恒生国...
每周6000+国际航班 外籍游...   民航局消息,今年以来,我国民航国际客运航线网络不断拓展,国际航班量和旅客量进一步增加。  我国通...
​万亿“宁王”,定了! 【导读】宁德时代拟5月20日在港股上市,募资总额将达40亿至50亿美元中国基金报记者 邱德坤5月12...
双向奔“富”:东货西进与西产东... 当西藏牧民巴特尔在海拔4000米的牧场签收到来自山东菏泽的A级芍药时,横亘在西部与东部之间的地理鸿沟...
机构对黄金持续看好,黄金ETF... 截至2025年5月12日 11:30,黄金ETF基金(159937)下跌1.77%,最新报价7.38...
今日75只股长线走稳 站上年线 证券时报·数据宝统计,截至今日上午收盘,上证综指3354.53点,收于年线之上,涨跌幅为0.37%,...
专访丨不确定性加剧背景下巴中深... 新华社巴西利亚5月11日电 专访|不确定性加剧背景下巴中深化合作具有重要意义——访巴西出口投资促进局...
万泽股份跌2.03%,成交额9... 5月12日,万泽股份盘中下跌2.03%,截至13:02,报15.44元/股,成交9444.31万元,...
南方路机涨2.27%,成交额5... 5月12日,南方路机盘中上涨2.27%,截至13:00,报25.63元/股,成交5301.33万元,...
万亿“宁王” 定了! 5月12日,宁德时代披露发行阶段董事会公告及注册招股书,宣布正式启动国际配售簿记,预计5月20日在港...
【西部量化】基金定量研究系列(... (转自:西部证券研究发展中心)本文构建了5个基金特征因子、18个基金净值因子(RBPs),以及11个...
宁德时代动力电池技术获国际航空... 格隆汇5月12日|宁德时代宣布,近日通过AS9100航空航天质量管理体系认证,该认证是由国际航空航天...
融捷股份涨2.01%,成交额7... 5月12日,融捷股份盘中上涨2.01%,截至13:01,报28.87元/股,成交7917.36万元,...
南天信息涨2.08%,成交额2... 5月12日,南天信息盘中上涨2.08%,截至13:01,报19.17元/股,成交2.78亿元,换手率...
粽子里吃出带血创可贴?来伊份:... 5月10日有网友发视频投诉在来伊份买的粽子中吃出疑似带血创可贴“给小孩喂粽子在里面发现了异物(是)创...
让文明成为生活方式,杨浦第二届... 转自:上观新闻日前,上海杨浦第二届文明生活节在长阳创谷举办。市民群众走上城市文明开放麦舞台,观摩文明...
伊外长:可调整铀浓缩水平以与美... 转自:上观新闻据俄罗斯卫星通讯社5月12日报道,伊朗外交部长阿巴斯·阿拉格齐表示,德黑兰不会在铀浓缩...
天蓝水碧迎盛会 转自:廊坊日报 初夏的廊坊绿意盎然。随着2025年廊洽会临近,廊坊精心改善城市环境,提升城市...