通信配置方法、联邦学习方法、装置、电子设备和介质与流程

本发明涉及联邦学习任务，尤其涉及一种通信配置方法、联邦学习方法、装置、电子设备和介质。

背景技术：

1、联邦学习(federated learning)，是一个机器学习框架，能有效帮助多个参与方在满足用户隐私保护和数据安全的情况下，进行数据使用和机器学习建模。一个联邦学习集群一般由一个中心服务方和若干个联邦参与方组成，在中心服务方的协调下完成各种联邦学习的训练任务。

2、现有的技术方案一般都是先部署一个中心服务方，然后在每个联邦参与方都部署一套系统，通过手动配置与中心服务方的网络连接以及配置各个参与方之间的网络连接，最终组建成一个联邦学习的集群。

3、手动固定的配置为后续联邦学习集群的变更带来了诸多不便，当需要新增一个参与方的时候，需要更改集群中各方的配置，以及更新相应的通信证书，过程繁琐且耗时。

技术实现思路

1、本发明提供一种通信配置方法、联邦学习方法、装置、电子设备和介质，用以解决现有技术中手动固定配置联邦学习任务中各方之间的连接带来诸多不便的缺陷。

2、本发明提供一种通信配置方法，包括：

3、确定联邦学习任务的任务标识，以及参与所述联邦学习任务的各方的定位域名；

4、基于所述任务标识以及所述各方的定位域名，确定所述各方的通信域名；

5、基于所述各方的定位域名，确定所述各方之间的通信证书；

6、将所述各方的通信域名和各方之间的通信证书配置到所述各方，以使所述各方建立通信连接并基于所述通信连接执行所述联邦学习任务。

7、根据本发明提供的通信配置方法，所述各方包括中心服务方和至少两个参与方，所述各方的定位域名确定步骤包括：

8、确定中心服务方的定位域名；

9、所述中心服务方从参与方域名池中进行定位域名分配，得到至少两个参与方的定位域名。

10、根据本发明提供的通信配置方法，所述各方采用域名访问的形式建立通信连接，所述基于所述任务标识以及所述各方的定位域名，确定所述各方的通信域名，包括：

11、在所述中心服务方为被访问方的情况下，基于所述任务标识和所述中心服务方的定位域名，确定所述中心服务方的通信域名；

12、在所述至少两个参与方中任一参与方为被访问方的情况下，基于所述任务标识和所述任一参与方的定位域名，确定所述任一参与方的通信域名。

13、根据本发明提供的通信配置方法，所述基于所述各方的定位域名，确定所述各方之间的通信证书，包括：

14、基于通配符和所述中心服务方的定位域名，确定所述中心服务方与所述至少两个参与方之间的通信证书；

15、基于通配符和所述参与方域名池中的所有定位域名，确定所述至少两个参与方之间的通信证书。

16、根据本发明提供的通信配置方法，所述得到至少两个参与方的定位域名，之后还包括：

17、所述中心服务方将所述至少两个参与方的定位域名，与所述至少两个参与方的地址进行绑定并存储绑定记录，以使所述至少两个参与方基于所述绑定记录进行域名解析。

18、本发明还提供一种联邦学习方法，包括：

19、基于本端与联邦学习任务中其余各方的通信连接，执行所述联邦学习任务；

20、所述通信连接是基于预先配置的各方的通信域名和各方之间的通信证书建立的，所述各方的通信域名是基于所述联邦学习任务的任务标识以及所述各方的定位域名确定的，所述各方之间的通信证书是基于所述各方的定位域名确定的。

21、本发明还提供一种通信配置装置，包括：

22、定位域名确定单元，用于确定联邦学习任务的任务标识，以及参与所述联邦学习任务的各方的定位域名；

23、通信域名确定单元，用于基于所述任务标识以及所述各方的定位域名，确定所述各方的通信域名；

24、通信证书确定单元，用于基于所述各方的定位域名，确定所述各方之间的通信证书；

25、配置单元，用于将所述各方的通信域名和各方之间的通信证书配置到所述各方，以使所述各方建立通信连接并基于所述通信连接执行所述联邦学习任务。

26、本发明还提供一种联邦学习装置，包括：

27、任务执行单元，用于基于本端与联邦学习任务中其余各方的通信连接，执行所述联邦学习任务；

28、所述通信连接是基于预先配置的各方的通信域名和各方之间的通信证书建立的，所述各方的通信域名是基于所述联邦学习任务的任务标识以及所述各方的定位域名确定的，所述各方之间的通信证书是基于所述各方的定位域名确定的。

29、本发明还提供一种电子设备，包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序，所述处理器执行所述程序时实现如上述任一种所述通信配置方法或联邦学习方法。

30、本发明还提供一种非暂态计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现如上述任一种所述通信配置方法或联邦学习方法。

31、本发明还提供一种计算机程序产品，包括计算机程序，所述计算机程序被处理器执行时实现如上述任一种所述通信配置方法或联邦学习方法。

32、本发明提供的通信配置方法、联邦学习方法、装置、电子设备和介质，基于任务标识以及各方的定位域名，确定各方的通信域名，并基于各方的定位域名，确定各方之间的通信证书，使得联邦学习任务中各方根据预先配置好的通信域名和通信证书建立通信连接，并基于通信连接执行该联邦学习任务。能够实现各方自动进行通信连接的配置，在保证通信安全性和可靠性的前提下，极大的简化联邦学习集群的配置流程。

33、此外，本发明支持多个联邦学习任务并行，并且都使用一个端口转发请求，每个参与方只需对外暴露一个端口，可以实现一个端口代理多个请求，从而极大地提升任务运行效率。

技术特征：

1.一种通信配置方法，其特征在于，包括：

2.根据权利要求1所述的通信配置方法，其特征在于，所述各方包括中心服务方和至少两个参与方，所述各方的定位域名确定步骤包括：

3.根据权利要求2所述的通信配置方法，其特征在于，所述各方采用通信域名访问的形式建立通信连接，所述基于所述任务标识以及所述各方的定位域名，确定所述各方的通信域名，包括：

4.根据权利要求2所述的通信配置方法，其特征在于，所述基于所述各方的定位域名，确定所述各方之间的通信证书，包括：

5.根据权利要求2-4中任一项所述的通信配置方法，其特征在于，所述得到所述至少两个参与方的定位域名，之后还包括：

6.一种联邦学习方法，其特征在于，包括：

7.一种通信配置装置，其特征在于，包括：

8.一种联邦学习装置，其特征在于，包括：

9.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至5任一项所述通信配置方法或如权利要求6所述联邦学习方法。

10.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至5任一项所述通信配置方法或如权利要求6所述联邦学习方法。

技术总结
本发明提供一种通信配置方法、联邦学习方法、装置、电子设备和介质，其中通信配置方法包括：确定联邦学习任务的任务标识，以及参与所述联邦学习任务的各方的定位域名；基于所述任务标识以及所述各方的定位域名，确定所述各方的通信域名；基于所述各方的定位域名，确定所述各方之间的通信证书；将所述各方的通信域名和各方之间的通信证书配置到所述各方，以使所述各方建立通信连接并基于所述通信连接执行所述联邦学习任务。本发明提供的通信配置方法、联邦学习方法、装置、电子设备和介质，能够实现各方自动进行通信连接的配置，在保证通信安全性和可靠性的前提下，极大的简化联邦学习集群的配置流程。

技术研发人员：徐生,陈剑波,束柬,王金钖,戈力
受保护的技术使用者：科大讯飞股份有限公司
技术研发日：
技术公布日：2024/1/12