[发明专利]一种联邦学习方法、装置及电子设备

权利要求书

1.一种联邦学习方法，应用于服务端，其特征在于，包括：

向各客户端发送元模型以及初始模型参数，以使各客户端基于所述元模型、初始模型参数、客户端本地的训练数据以及目标补丁模型进行第一轮模型训练；

从各所述客户端中确定出用于进行模型再训练的、若干目标客户端，并采集各目标客户端当前第n轮训练获得的第一模型的第一模型参数，所述n为正整数；

基于各目标客户端的第一模型参数，采用预定的计算方式计算获得用于进行n+1轮模型训练的第二模型参数；

将所述第二模型参数发送给各所述目标客户端，以使各所述目标客户端基于已接收的元模型、第二模型参数、客户端本地的训练数据以及目标补丁模型进行模型再训练获得当前第二模型；

判断各客户端训练获得的当前第二模型是否均符合训练条件，在各客户端训练获得的当前第二模型不符合训练条件时，重新确定出用于进行模型再训练的、若干目标客户端；在各客户端训练获得第二模型符合训练条件时停止训练；

所述方法还包括：基于模型训练任务的任务类型、服务器数据与客户端数据的数据差异度以及服务器元模型的结构复杂度，确定映射补丁模型、残差补丁模型、内部补丁模型中的任意一种为所述目标补丁模型，具体包括：

在所述任务类型为监控任务或定位任务时，确定所述映射补丁模型为所述目标补丁模型；

在所述服务器数据与客户端数据的数据差异度大于预定差异度阈值时，确定所述残差补丁模型为所述目标补丁模型；

在服务器元模型的结构复杂度大于预定复杂度时，确定所述内部补丁模型为所述目标补丁模型；

其中，所述映射补丁模型包括：映射网络以及激活层；

所述残差补丁模型包括：残差连接层；

所述内部补丁模型包括：卷积层以及激活层。

2.如权利要求1所述的方法，其特征在于，在向各客户端发送元模型以及初始模型参数之前，所述方法还包括：

接收各客户端发送的各数据类型以及与各数据类型对应的各训练数据的数据标识；

基于各客户端发送的数据类型以及数据标识的数量，计算获得各数据类型之间数据标识总量的数据占比；

在进行每一轮模型训练之前，基于所述数据占比以及各客户端所包含的目标数据类型，从各目标数据类型对应的若干数据标识中确定若干目标数据标识，并将各所述目标数据标识发送给对应的客户端，以为各客户端重新分配用于进行模型训练的训练数据。

3.如权利要求2所述的方法，其特征在于，所述基于所述数据占比以及各客户端所包含的目标数据类型，从各目标数据类型对应的若干数据标识中确定若干目标数据标识，具体包括：

基于所述数据占比以及各客户端所包含的目标数据类型，分别确定与各客户端对应的目标数据占比；

基于各客户端的目标数据占比，从对应客户端所发送的各数据类型的数据标识中确定出若干目标数据标识。

4.如权利要求1所述的方法，其特征在于，所述基于各目标客户端的第一模型参数，采用预定的计算方式计算获得用于进行n+1轮模型训练的第二模型参数，具体包括：

基于各目标客户端第n轮训练获得的第一模型参数以及各目标客户端第n-1轮训练获得的历史模型参数，确定各目标客户端对应的梯度参数；

基于各目标客户端的梯度参数，采用预定的计算公式计算获得目标梯度参数；

基于所述目标梯度参数确定用于进行n+1轮模型训练的第二模型参数。