基于联邦学习场景的训练参与方的选择方法及装置
摘要:
本发明提供的一种基于联邦学习场景的训练参与方的选择方法及装置,基于数据类别和数据量对数据集进行划分,得到联邦学习场景下每个客户端数据;基于客户端数据对全局模型进行训练,若当前训练轮次小于预设的最大训练轮次,基于线性拟合方法对每个客户端数据进行计算,在客户端中确定当前训练轮次的候选参与方列表;获取当前训练轮次的训练参与方的预设数目;预设数目小于客户端的总数;根据预设数目在当前训练轮次的候选参与方列表中确定当前训练轮次的训练参与方。该方法可以有效降低模型训练过程的通信开销,通过充分利用客户端的信息,选择更优的训练参与方参与本轮次的训练,使模型的训练可以达到更好的效果。
0/0