本发明涉及无线通信,尤其涉及一种数据传输方法、装置、计算机可读存储介质和通信设备。
背景技术:
1、随着数据隐私保护和通信资源受限的原因,基于联邦学习技术的分布式模型训练得到越来越多的关注。联邦学习技术允许客户端(client)使用本地数据进行模型训练,将本地模型参数(简称本地模型)上传给中心节点汇聚后形成全局模型,该过程经过多次迭代以满足模型的收敛性和精度要求。
2、移动通信系统中,当基站作为汇聚中心节点,终端作为client节点参与联邦学习训练时,如果终端移动到小区边缘,发生基站间的小区切换后,将导致该终端不再与原来的基站(模型汇聚中心节点)进行交互,可能出现本地模型无法到达汇聚节点的问题,进而影响全局模型的汇聚过程和精度。
技术实现思路
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、第四方面,本发明实施例还提供一种数据传输装置,包括第一通讯单元;
34、所述第一通讯单元,用于在确定第一终端切换至第二网络设备、且所述第一终端正在执行第一训练任务的情况下,向所述第二网络设备发送用于请求切换的第一消息;所述第一消息中至少包括第一信息,所述第一信息用于请求所述第二网络设备转发第一模型数据;
35、所述第一通讯单元,还用于接收所述第二网络设备发送的第二消息,所述第二消息至少包括第一指示信息,所述第一指示信息用于指示所述第二网络设备支持或不支持转发所述第一模型数据;
36、其中,所述第一终端为参与第一网络设备的模型聚合任务的多个终端中的任一终端,所述第一训练任务为所述第一网络设备基于所述模型聚合任务向所述第一终端分配的、针对本地模型训练的任务,所述第一模型数据为所述第一终端执行所述第一训练任务获得的数据。
37、第五方面,本发明实施例还提供一种数据传输装置,包括第二通讯单元和第二处理单元;其中,
38、所述第二通讯单元,用于接收第一网络设备发送的第一消息,所述第一消息用于请求第一终端由第一网络设备切换至第二网络设备;所述第一消息中至少包括第一信息,所述第一信息用于请求转发第一模型数据;
39、所述第二处理单元,用于确定支持或不支持转发所述第一模型数据;
40、所述第二通讯单元,还用于向所述第一网络设备发送第二消息,所述第二消息至少包括第一指示信息,所述第一指示信息用于指示支持或不支持转发所述第一模型数据;
41、其中,所述第一终端为参与所述第一网络设备的模型聚合任务的多个终端中的任一终端,所述第一模型数据为所述第一终端执行第一训练任务获得的数据,所述第一训练任务为所述第一网络设备基于所述模型聚合任务向所述第一终端分配的、针对本地模型训练的任务。
42、第六方面,本发明实施例还提供一种数据传输装置,包括第三通讯单元;
43、所述第三通讯单元,用于接收第一网络设备发送的第三消息,所述第三消息用于指示接入第二网络设备后、将继续执行第一训练任务获得的第一模型数据发送至所述第二网络设备;所述第三消息中至少包括所述第二网络设备的标识信息;
44、所述第三通讯单元,还用于向所述第二网络设备发送所述第一模型数据;
45、其中,所述第三消息由所述第一网络设备向所述第二网络设备发送第一消息、且确认所述第二网络设备支持转发所述第一模型数据后发送,所述第一消息用于请求第一终端由所述第一网络设备切换至所述第二网络设备;所述第一消息中至少包括第一信息,所述第一信息用于请求所述第二网络设备转发所述第一模型数据;所述第一终端为参与所述第一网络设备的模型聚合任务的多个终端中的任一终端,所述第一训练任务为所述第一网络设备基于所述模型聚合任务向所述第一终端分配的、针对本地模型训练的任务。
46、第七方面,本发明实施例还提供一种计算机可读存储介质,其上存储有计算机程序,该程序被处理器执行时实现前述第一方面、第二方面或第三方面所述方法的步骤。
47、第八方面,本发明实施例还提供一种通信设备,包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,所述处理器执行所述程序时实现前述第一方面、第二方面或第三方面所述方法的步骤。
48、本发明实施例提供一种数据传输方法、装置、计算机可读存储介质和通信设备,第一网络设备在确定第一终端切换至第二网络设备、且第一终端正在执行第一训练任务的情况下,向第二网络设备发送用于请求切换的第一消息,其中至少包括用于请求第二网络设备转发第一模型数据的第一信息;接收第二网络设备发送的第二消息,其中至少包括用于指示第二网络设备支持或不支持转发第一模型数据的第一指示信息;终端进行小区切换时,可在网络设备之间进行模型数据转发的能力交互,并可在切换后的网络设备支持模型数据转发的情况下,实现终端切换后本地模型的正常上传,保障了模型汇聚中心节点对全局模型的聚合过程以及聚合后的全局模型精度。
1.一种数据传输方法,其特征在于,所述方法应用于第一网络设备中;所述方法包括:
2.根据权利要求1所述的方法,其特征在于,所述第一信息至少包括所述第一网络设备的标识信息、所述第一训练任务的标识信息、所述模型聚合任务对应的全局模型的标识信息、所述第一模型数据的数据量和所述第一模型数据的转发时延需求;其中,所述数据量和所述转发时延需求用于所述第二网络设备确定支持或不支持转发所述第一模型数据。
3.根据权利要求1所述的方法,其特征在于,在所述第一指示信息用于指示所述第二网络设备支持转发所述第一模型数据的情况下,所述第二消息中包括所述第二网络设备的标识信息;所述方法还包括:
4.根据权利要求3所述的方法,其特征在于,所述方法还包括:
5.根据权利要求3所述的方法,其特征在于,所述第一信息还包括用于指示所述第二网络设备进行模型聚合的第二指示信息、模型聚合的迭代次数和模型聚合的算法信息中的至少一项信息;和/或,
6.根据权利要求5所述的方法,其特征在于,在所述第三指示信息用于指示所述第二网络设备支持模型聚合的情况下,所述第三消息中还包括模型聚合的迭代次数和所述第一模型数据的精度需求信息。
7.根据权利要求6所述的方法,其特征在于,所述方法还包括:
8.根据权利要求4所述的方法,其特征在于,所述方法还包括:
9.根据权利要求1所述的方法,其特征在于,在所述第一指示信息用于指示所述第二网络设备不支持转发所述第一模型数据的情况下;所述方法还包括:
10.根据权利要求1所述的方法,其特征在于,所述方法还包括:
11.一种数据传输方法,其特征在于,所述方法应用于第二网络设备中;所述方法包括:
12.根据权利要求11所述的方法,其特征在于,所述第一信息至少包括所述第一网络设备的标识信息、所述第一训练任务的标识信息、所述模型聚合任务对应的全局模型的标识信息、所述第一模型数据的数据量和所述第一模型数据的转发时延需求。
13.根据权利要求12所述的方法,其特征在于,所述确定支持或不支持转发所述第一模型数据,包括:
14.根据权利要求11所述的方法,其特征在于,在确定支持转发所述第一模型数据的情况下,所述第二消息中还包括所述第二网络设备的标识信息。
15.根据权利要求14所述的方法,其特征在于,所述方法还包括:
16.根据权利要求14所述的方法,其特征在于,所述第一信息还包括用于指示进行模型聚合的第二指示信息、模型聚合的迭代次数和模型聚合的算法信息中的至少一项信息;所述方法还包括:
17.根据权利要求16所述的方法,其特征在于,所述方法还包括:
18.根据权利要求15所述的方法,其特征在于,所述方法还包括:
19.一种数据传输方法,其特征在于,所述方法应用于第一终端;所述方法包括:
20.根据权利要求19所述的方法,其特征在于,所述第三消息中还包括所述第二网络设备进行模型聚合的迭代次数和所述第一模型数据的精度需求信息。
21.根据权利要求19所述的方法,其特征在于,所述方法还包括:
22.一种数据传输装置,其特征在于,所述装置包括第一通讯单元;
23.一种数据传输装置,其特征在于,所述装置包括第二通讯单元和第二处理单元;其中,
24.一种数据传输装置,其特征在于,所述装置包括第三通讯单元;
25.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,该程序被处理器执行时实现权利要求1至10任一项所述方法的步骤;或者,
26.一种通信设备,包括存储器、处理器及存储在存储器上并可在处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现权利要求1至10任一项所述方法的步骤;或者,所述处理器执行所述程序时实现权利要求11至18任一项所述方法的步骤;或者,所述处理器执行所述程序时实现权利要求19至21任一项所述方法的步骤。