[發(fā)明專利]一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法

有效

• 申請?zhí)枺?/td>: 202210342809.7
• 申請日: 2022-03-31
• 公開（公告）號: CN114785810B
• 公開（公告）日: 2023-05-16
• 發(fā)明（設(shè)計(jì)）人: 文斌;劉少杰
• 申請（專利權(quán)）人: 海南師范大學(xué)
• 主分類號: H04L67/1095
• 分類號: H04L67/1095;H04L67/30;H04L67/06;H04L9/32;H04L9/06
• 代理公司: 深圳市廣諾專利代理事務(wù)所(普通合伙) 44611
• 代理人: 李亞萍

1.一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，包括如下步驟：

S1：中央服務(wù)端選擇若干參與訓(xùn)練的客戶端，所述中央服務(wù)端生成服務(wù)端信息，所述中央服務(wù)端將所有的所述客戶端排列成完全二叉樹，所述中央服務(wù)端位于根節(jié)點(diǎn)，所述客戶端分別位于子節(jié)點(diǎn)和葉子節(jié)點(diǎn)；

S2：所述中央服務(wù)端將第一全局模型文件拆分成若干種子文件，并計(jì)算所述第一全局模型文件的第一哈希值和不同的所述子文件對應(yīng)的第二哈希值；

S3：所述中央服務(wù)端生成公鑰信息和私鑰信息，將所述公鑰信息分發(fā)給所有的所述客戶端，所述客戶端保存所述公鑰信息；

S4：所述中央服務(wù)端將所述服務(wù)端信息、所述第一哈希值和所述第二哈希值進(jìn)行拼接，形成迭代配置文件，并計(jì)算所述迭代配置文件的第三哈希值，最后使用所述私鑰信息對所述迭代配置文件進(jìn)行簽名，形成簽名結(jié)果，將所述簽名結(jié)果和所述迭代配置文件發(fā)送給所有的所述客戶端，所述客戶端對所述簽名結(jié)果和所述迭代配置文件進(jìn)行對比驗(yàn)證，并生成驗(yàn)證結(jié)果，驗(yàn)證通過后保存到本地；

S5：所述中央服務(wù)端查詢自身的所述服務(wù)端信息，并向所述子節(jié)點(diǎn)處的所述客戶端逐次發(fā)送所有的所述子文件，所述子節(jié)點(diǎn)處的所述客戶端接收所述子文件后，所述子節(jié)點(diǎn)處的所述客戶端計(jì)算所述子文件的第四哈希值，并將所述第四哈希值與所述第三哈希值進(jìn)行對比驗(yàn)證，如果不同，則向中央服務(wù)端發(fā)送錯(cuò)誤反饋，如果相同，則所述子節(jié)點(diǎn)處的所述客戶端查詢自身的所述服務(wù)端信息，并將所述子文件發(fā)送給下一個(gè)所述子節(jié)點(diǎn)處的所述客戶端，同步接收自身的父節(jié)點(diǎn)處的所述客戶端發(fā)送的下一個(gè)所述子文件，所述子節(jié)點(diǎn)處的所述客戶端依次對所述第四哈希值與所述第三哈希值進(jìn)行對比驗(yàn)證，如果不同，則向中央服務(wù)端發(fā)送錯(cuò)誤反饋，如果相同，則將所述子文件發(fā)送給下一個(gè)所述子節(jié)點(diǎn)處的所述客戶端，直至所述葉子節(jié)點(diǎn)處的所述客戶端接收到所述子文件；

S6：所有的所述葉子節(jié)點(diǎn)處的所述客戶端將接收的所述子文件進(jìn)行組合得到當(dāng)前迭代輪次的第二全局模型文件，并計(jì)算所述第二全局模型文件的第五哈希值，將所述第五哈希值與所述第一哈希值進(jìn)行對比，如果相同，則當(dāng)前迭代輪次所述第一全局模型文件同步結(jié)束，如果不同，則重復(fù)S5，直至所有的所述葉子節(jié)點(diǎn)處的所述客戶端完成接收所有的所述子文件。

2.根據(jù)權(quán)利要求1所述的一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，所述S1中，所述服務(wù)端信息包括樹結(jié)構(gòu)體信息和迭代輪次信息，所述樹結(jié)構(gòu)體信息由若干不同的節(jié)點(diǎn)信息組成，所述節(jié)點(diǎn)信息包括客戶端數(shù)據(jù)接收地址、客戶端唯一標(biāo)識、客戶端左子節(jié)點(diǎn)標(biāo)識和客戶端右子節(jié)點(diǎn)標(biāo)識。

3.根據(jù)權(quán)利要求1所述的一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，所述S2中，所述第一哈希值和所述第二哈希值的計(jì)算可采用任何哈希計(jì)算方法。

4.根據(jù)權(quán)利要求1所述的一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，所述S3中，所述中央服務(wù)端僅在訓(xùn)練初始階段以及有新的所述客戶端加入時(shí)分發(fā)所述公鑰信息。

5.根據(jù)權(quán)利要求1所述的一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，所述S4中，可采用任何非對稱加密算法對所述私鑰信息進(jìn)行簽名。

6.根據(jù)權(quán)利要求1所述的一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，所述S4中，所述迭代配置文件為Json格式。

7.根據(jù)權(quán)利要求1所述的一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，所述S4中，所述驗(yàn)證結(jié)果為所述客戶端使用所述公鑰信息對所述簽名結(jié)果進(jìn)行解密，生成第五哈希值，將所述第五哈希值與所述第三哈希值對比，若相同，則驗(yàn)證通過，若不相同，則驗(yàn)證不通過。

8.根據(jù)權(quán)利要求2所述的一種適用于聯(lián)邦學(xué)習(xí)的樹狀廣播數(shù)據(jù)同步方法，其特征在于，所述S5中，所述中央服務(wù)端查詢所述節(jié)點(diǎn)信息，所述子節(jié)點(diǎn)處的所述客戶端查詢自身的所述節(jié)點(diǎn)信息。