[发明专利]基于TEE的联邦推荐方法、装置、设备及介质

说明书

本发明实施例公开了一种基于TEE的联邦推荐方法、装置、设备及介质，该方法包括：基于云边端框架对任意两个边缘设备执行DH密钥交换协议并将生成的对称加密密钥存储于TEE的内存中；接收边缘设备发送的推送请求；在TEE中将存储于每个边缘设备的本地硬盘中的本地模型进行解密；在TEE中将每个边缘设备的本地数据输入至对应的解密后的本地模型中并采用对应的对称加密密钥将生成的中间结果进行加密后发送至推送请求对应的边缘设备中进行解密和推理计算以进行推送服务。本申请还涉及区块链技术，推送请求存储于区块链中。本发明基于数字签名技术，通过在TEE中执行DH协议以及云边端框架中数据的加密和处理，杜绝了在进行联邦推理过程中数据被篡改的可能性。

技术领域

本发明涉及联邦推理技术领域，尤其涉及一种基于TEE的联邦推荐方法、装置、设备及介质。

背景技术

随着边缘计算的发展，云边端协同一体化已经成为人工智能领域的新趋势，智能边缘设备如智能网关，边缘小型云服务器等已广泛的进行了应用。边缘设备通过运行一系列人工智能相关应用提供数据处理，推理等服务应用于生活中，但是边缘端设备往往处于开放的网络环境和不安全的物理环境中，极易通过网络设备上安装嗅探器来截取数据报文，如截获云端向边缘设备推送的深度学习模型，或通过内存总线窥探、内存篡改等方式直接攻击内存中的数据，不仅导致用户隐私泄露，而且无法保证在进行联邦推理过程中数据不被篡改。

发明内容

本发明实施例提供了一种基于TEE的联邦推荐方法、装置、设备及介质，避免了用户隐私泄露，同时杜绝了在进行联邦推理过程中数据被篡改的问题。

第一方面，本发明实施例提供了一种基于TEE的联邦推荐方法，其包括：

基于云边端框架，在TEE中对任意两个边缘设备执行DH密钥交换协议，并将生成的对称加密密钥存储于所述TEE的内存中；

接收所述云边端框架中边缘设备发送的推送请求；

在所述TEE中将存储于每个所述边缘设备的本地硬盘中的本地模型进行解密；

在所述TEE中将每个所述边缘设备的本地数据输入至对应的解密后的本地模型中，得到每个所述边缘设备的中间结果；

在所述TEE中将所述云边端框架中其余边缘设备的中间结果，采用对应的对称加密密钥进行加密，并将加密后的中间结果发送至所述推送请求的边缘设备中；

在所述TEE中进行解密和推理计算，并根据生成的推理结果向所述推送请求的边缘设备进行推送服务。

第二方面，本发明实施例提供了一种基于TEE的联邦推荐装置，其包括：

第一密钥生成单元，用于基于云边端框架，在TEE中对任意两个边缘设备执行DH密钥交换协议，并将生成的对称加密密钥存储于所述TEE的内存中；

接收单元，用于接收所述云边端框架中边缘设备发送的推送请求；

第一解密单元，用于在所述TEE中将存储于每个所述边缘设备的本地硬盘中的本地模型进行解密；

输入单元，用于在所述TEE中将每个所述边缘设备的本地数据输入至对应的解密后的本地模型中，得到每个所述边缘设备的中间结果；

第一发送单元，用于在所述TEE中将所述云边端框架中其余边缘设备的中间结果，采用对应的对称加密密钥进行加密，并将加密后的中间结果发送至所述推送请求的边缘设备中；

推送单元，用于在所述TEE中进行解密和推理计算，并根据生成的推理结果向所述推送请求的边缘设备进行推送服务。

第三方面，本发明实施例又提供了一种计算机设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现如上述第一方面所述的基于TEE的联邦推荐方法。