基于SoftGBDT的跨特征聯(lián)邦學習方法、預測方法

基本信息

申請?zhí)?/td> CN202110806104.1 申請日 -
公開（公告）號 CN113435537A 公開（公告）日 2021-09-24
申請公布號 CN113435537A 申請公布日 2021-09-24
分類號 G06K9/62（2006.01）I;G06N20/00（2019.01）I 分類 計算；推算；計數(shù);
發(fā)明人 周一競;孟丹;李宏宇;李曉林 申請（專利權(quán)）人 同盾科技有限公司
代理機構(gòu) 北京律智知識產(chǎn)權(quán)代理有限公司 代理人 王輝;闞梓瑄
地址 311121浙江省杭州市余杭區(qū)五常街道文一西路998號18幢704室
法律狀態(tài) -

摘要

摘要 本公開是關(guān)于一種基于Soft GBDT的跨特征聯(lián)邦學習方法、數(shù)據(jù)預測方法及裝置，涉及機器學習技術(shù)領(lǐng)域，該基于Soft GBDT的跨特征聯(lián)邦學習方法包括：利用Soft GBDT計算第一特征數(shù)據(jù)在Soft GBDT中的第一內(nèi)節(jié)點的第一線性回歸部分；計算第一線性回歸部分以及第二線性回歸部分在Soft GBDT中的葉子節(jié)點的輸出值，并計算Soft GBDT中所包括的當前軟決策樹的局部損失函數(shù)；根據(jù)局部損失函數(shù)計算Soft GBDT的全局損失函數(shù)，并根據(jù)全局損失函數(shù)計算第一內(nèi)節(jié)點的第一梯度，以對Soft GBDT的第一內(nèi)節(jié)點的參數(shù)進行更新。本公開有利于加快模型訓練速度，提高模型訓練效率。