一種優(yōu)化樹模型訓(xùn)練的方法及縱向聯(lián)邦學(xué)習(xí)系統(tǒng)

基本信息

申請?zhí)?/td> CN202110777115.1 申請日 -
公開（公告）號 CN113537333A 公開（公告）日 2021-10-22
申請公布號 CN113537333A 申請公布日 2021-10-22
分類號 G06K9/62;G06N20/20;G06F21/60 分類 計算；推算；計數(shù);
發(fā)明人 黃一珉;王灣灣;何浩;姚明 申請（專利權(quán)）人 深圳市洞見智慧科技有限公司
代理機構(gòu) 北京柏杉松知識產(chǎn)權(quán)代理事務(wù)所（普通合伙） 代理人 丁蕓;馬敬
地址 518118 廣東省深圳市福田區(qū)福田街道崗廈社區(qū)彩田路3069號星河世紀(jì)A棟3603B8
法律狀態(tài) -

摘要

摘要 本發(fā)明實施例提供了一種優(yōu)化樹模型訓(xùn)練的方法及縱向聯(lián)邦學(xué)習(xí)系統(tǒng)，該方法包括：數(shù)據(jù)提供方先基于分桶得到的各分桶中第一用戶樣本數(shù)據(jù)的數(shù)量，確定用戶樣本數(shù)據(jù)的數(shù)量最多的分桶；然后基于數(shù)據(jù)需求方發(fā)送的同態(tài)加密的梯度信息和除用戶樣本數(shù)據(jù)的數(shù)量最多的分桶外的各分桶的第一用戶樣本數(shù)據(jù)建立加密的第一梯度直方圖；數(shù)據(jù)需求方基于第二用戶樣本數(shù)據(jù)的梯度總和以及解密后的第一梯度直方圖，確定的用戶樣本數(shù)據(jù)的數(shù)量最多的分桶所對應(yīng)的梯度信息，最后基于解密后的第一梯度直方圖和用戶樣本數(shù)據(jù)的數(shù)量最多的分桶所對應(yīng)的梯度信息，確定樹模型的最優(yōu)分裂點，可以減少加密計算過程和解密計算過程所帶來的時間開銷，進(jìn)而可以提高樹模型的訓(xùn)練效率。