[發明專利]基于TEE的聯邦推薦方法、裝置、設備及介質

有效

• 申請號: 202110609712.3
• 申請日: 2021-06-01
• 公開（公告）號: CN113259100B
• 公開（公告）日: 2022-10-28
• 發明（設計）人: 李正揚;王健宗;黃章成;司世景
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: H04L9/08
• 分類號: H04L9/08;H04L9/40;H04L67/1097;H04L67/55;G06N20/20
• 代理公司: 深圳市精英專利事務所 44242
• 代理人: 涂年影

1.一種基于TEE的聯邦推薦方法，其特征在于，包括以下步驟：

基于云邊端框架，在可信執行環境TEE中對任意兩個邊緣設備執行DH密鑰交換協議，并將生成的對稱加密密鑰存儲于所述TEE的內存中；

接收所述云邊端框架中邊緣設備發送的推送請求；

在所述TEE中將存儲于每個所述邊緣設備的本地硬盤中的本地模型進行解密；

在所述TEE中將每個所述邊緣設備的本地數據輸入至對應的解密后的本地模型中，得到每個所述邊緣設備的中間結果；

在所述TEE中將所述云邊端框架中其余邊緣設備的中間結果，采用對應的對稱加密密鑰進行加密，并將加密后的中間結果發送至所述推送請求的邊緣設備中；

在所述TEE中進行解密和推理計算，并根據生成的推理結果向所述推送請求的邊緣設備進行推送服務；

其中，所述其余邊緣設備是指云邊端框架中除了當前發送推薦請求的邊緣設備之外的所有其它邊緣設備。

2.根據權利要求1所述的基于TEE的聯邦推薦方法，其特征在于，所述基于云邊端框架，在TEE中對任意兩個邊緣設備執行DH密鑰交換協議，并將生成的對稱加密密鑰存儲于所述TEE的內存中之后，還包括：

在所述TEE中執行云服務器與每個所述邊緣設備的DH密鑰交換協議，并將生成的對稱加密密鑰存儲于所述TEE的內存中。

3.根據權利要求2所述的基于TEE的聯邦推薦方法，其特征在于，所述在所述TEE中執行云服務器與每個所述邊緣設備的DH密鑰交換協議并將生成的對稱加密密鑰存儲于所述TEE的內存中之后，還包括：

將每個所述本地模型的模型參數在所述TEE中采用對應的對稱加密密鑰進行加密，得到加密后的模型參數；

根據所述加密后的模型參數在所述TEE中更新每個所述本地模型。

4.根據權利要求3所述的基于TEE的聯邦推薦方法，其特征在于，所述根據所述加密后的模型參數在所述TEE中更新每個所述本地模型，包括：

將所述加密后的模型參數進行二進制序列化；

采用HTTPS協議將二進制序列化后的模型參數發送至對應的邊緣設備中，并在所述TEE中進行本地模型的更新。

5.根據權利要求4所述的基于TEE的聯邦推薦方法，其特征在于，所述在所述TEE中進行本地模型的更新，包括：

在所述TEE中對所述二進制序列化后的模型參數依次進行反序列化和解密；

根據解密后的模型參數在所述TEE中對所述本地模型進行更新。

6.根據權利要求3所述的基于TEE的聯邦推薦方法，其特征在于，所述根據所述加密后的模型參數在所述TEE中更新每個所述本地模型之后，還包括：

基于云服務器與邊緣設備執行DH密鑰交換協議后生成的對稱加密密鑰，將更新后的本地模型進行加密處理并計算加密處理后的本地模型的MD5值；

將所述MD5值存儲于所述TEE的內存中，并將所述加密處理后的本地模型存儲于對應的邊緣設備的本地硬盤中。

7.根據權利要求1所述的基于TEE的聯邦推薦方法，其特征在于，所述在所述TEE中將存儲于每個所述邊緣設備的本地硬盤中的本地模型進行解密，包括：

將存儲于每個所述邊緣設備的本地硬盤中的本地模型讀入至富執行環境REE中；

將讀入至所述REE中的本地模型加載至所述TEE中進行MD5校驗；

根據所述MD5校驗的結果對每個所述本地模型進行解密。