一种GBDT模型训练中的梯度聚合方法、装置及电子设备
基本信息
| 申请号 | CN202210041304.7 | 申请日 | - |
| 公开（公告）号 | CN114065950A | 公开（公告）日 | 2022-02-18 |
| 申请公布号 | CN114065950A | 申请公布日 | 2022-02-18 |
| 分类号 | G06N20/10（2019.01）I | 分类 | 计算；推算；计数; |
| 发明人 | 陈智隆;郝天一;陈琨;王国赛;王西利 | 申请（专利权）人 | 华控清交信息科技（北京）有限公司 |
| 代理机构 | 北京市广友专利事务所有限责任公司 | 代理人 | 李欣 |
| 地址 | 100084北京市海淀区中关村东路1号院3号楼10层1009-1 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本申请公开了一种GBDT模型训练中的梯度聚合方法、装置及电子设备，应用于训练系统中的密文计算节点，接收主动方数据节点密文方式发送的多个样本的整数型梯度，整数型梯度为主动方数据节点将多个样本的浮点数类型的梯度转换为整数类型得到的；基于接收的各被动方数据节点发送的所持有特征的分箱结果，针对每个特征的分箱结果的每个分箱，将该分箱包括的样本的整数型梯度进行聚合，得到针对该种特征的整数型梯度聚合结果；向主动方数据节点密文方式发送每种特征的整数型梯度聚合结果，使得主动方数据节点将整数型梯度聚合结果转换为浮点数类型的梯度聚合结果。采用本方案，提高了GBDT模型训练的效率。 |
