一種GBDT模型訓(xùn)練中的梯度聚合方法、裝置及電子設(shè)備

基本信息

申請?zhí)?/td> CN202210041304.7 申請日 -
公開（公告）號 CN114065950A 公開（公告）日 2022-02-18
申請公布號 CN114065950A 申請公布日 2022-02-18
分類號 G06N20/10（2019.01）I 分類 計算；推算；計數(shù);
發(fā)明人 陳智隆;郝天一;陳琨;王國賽;王西利 申請（專利權(quán)）人 華控清交信息科技（北京）有限公司
代理機構(gòu) 北京市廣友專利事務(wù)所有限責(zé)任公司 代理人 李欣
地址 100084北京市海淀區(qū)中關(guān)村東路1號院3號樓10層1009-1
法律狀態(tài) -

摘要

摘要 本申請公開了一種GBDT模型訓(xùn)練中的梯度聚合方法、裝置及電子設(shè)備，應(yīng)用于訓(xùn)練系統(tǒng)中的密文計算節(jié)點，接收主動方數(shù)據(jù)節(jié)點密文方式發(fā)送的多個樣本的整數(shù)型梯度，整數(shù)型梯度為主動方數(shù)據(jù)節(jié)點將多個樣本的浮點數(shù)類型的梯度轉(zhuǎn)換為整數(shù)類型得到的；基于接收的各被動方數(shù)據(jù)節(jié)點發(fā)送的所持有特征的分箱結(jié)果，針對每個特征的分箱結(jié)果的每個分箱，將該分箱包括的樣本的整數(shù)型梯度進行聚合，得到針對該種特征的整數(shù)型梯度聚合結(jié)果；向主動方數(shù)據(jù)節(jié)點密文方式發(fā)送每種特征的整數(shù)型梯度聚合結(jié)果，使得主動方數(shù)據(jù)節(jié)點將整數(shù)型梯度聚合結(jié)果轉(zhuǎn)換為浮點數(shù)類型的梯度聚合結(jié)果。采用本方案，提高了GBDT模型訓(xùn)練的效率。