[发明专利]编译方法、装置、电子设备及存储介质

权利要求书

1.一种编译方法，其特征在于，应用于联邦学习的参与方，所述方法包括：

获取源代码，所述源代码中包括用于实现联邦学习的代码段；

对所述源代码进行解析，生成动态计算图，所述动态计算图中包括与所述代码段对应的计算节点，所述动态计算图用于表征所述联邦学习的数据处理逻辑；

利用预设的加密算子，替换所述动态计算图中的对应的计算节点的算子，生成加密计算图；

根据所述加密计算图，生成编译后的代码。

2.根据权利要求1所述的方法，其特征在于，所述利用预设的加密算子，替换所述动态计算图中的对应的计算节点的算子，生成加密计算图，包括：

获取预设的加密算子，所述加密算子包括下述至少一项：用于计算损失函数的加密算子、用于进行加法计算的加密算子、用于进行乘法计算的加密算子、用于进行减法计算的加密算子、用于进行指数计算的加密算子；

对于每一加密算子，在所述动态计算图中查找是否存在对应的计算节点；

若存在对应的计算节点，则利用所述加密算子替换所述计算节点的算子。

3.根据权利要求1或2任一项所述的方法，其特征在于，所述对所述源代码进行解析，生成动态计算图，包括：

对所述源代码进行解析，生成静态计算图，所述静态计算图中包括与所述代码段对应的计算节点，所述静态计算图用于表征源代码对应的数据处理逻辑；

根据所述静态计算图和预设的优化规则，生成动态计算图，所述动态计算图用于表征优化后的数据处理逻辑。

4.根据权利要求3所述的方法，其特征在于，所述根据所述静态计算图和预设的优化规则，生成动态计算图，包括：

根据预设的优化规则，确定所述静态计算图中的待调整节点；

调整所述待调整节点，生成动态计算图。

5.根据权利要求1或2任一项所述的方法，其特征在于，还包括：

确定在联邦学习过程中对数据进行加密时所使用的加密算法，根据加密算法，确定密文加法和/或密文乘法的计算逻辑；

根据所述密文加法和密文乘法的计算逻辑，确定各预设的加密算子的计算逻辑；

相应的，所述根据所述加密计算图，生成编译后的代码，包括：根据所述加密计算图中加密算子对应的计算逻辑，生成编译后的代码。

6.根据权利要求1或2任一项所述的方法，其特征在于，所述源代码中还包括所述代码段对应的执行方标识；

所述对所述源代码进行解析，生成动态计算图，包括：

根据所述执行方标识，生成角色节点，所述角色节点与所述代码段对应的计算节点关联；

所述根据所述加密计算图，生成编译后的代码，包括：

根据所述加密计算图的角色节点，生成对应的角色判断语句，以确定所述联邦学习的参与方是否为所述角色节点关联的计算节点的执行方；

根据所述角色判断语句、所述加密计算图中加密算子对应的计算逻辑，生成编译后的代码。

7.一种编译装置，其特征在于，包括：

获取模块，用于获取源代码，所述源代码中包括用于实现联邦学习的代码段；

解析模块，用于对所述源代码进行解析，生成动态计算图，所述动态计算图中包括与所述代码段对应的计算节点，所述动态计算图用于表征所述联邦学习的数据处理逻辑；

加密模块，用于利用预设的加密算子，替换所述动态计算图中的对应的计算节点的算子，生成加密计算图；

代码生成模块，用于根据所述加密计算图，生成编译后的代码。

8.一种电子设备，其特征在于，所述电子设备包括：存储器、处理器及存储在所述存储器上并可在所述处理器上运行的编译程序，所述编译程序被所述处理器执行时实现如权利要求1至6任一项所述的编译方法的步骤。

9.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有编译程序，所述编译程序被处理器执行时实现如权利要求1至6任一项所述的编译方法的步骤。

10.一种计算机程序产品，包括计算机程序，其特征在于，该计算机程序被处理器执行时实现权利要求1至6任一项所述的方法。