发明公开
- 专利标题: 基于联邦学习场景的训练参与方的选择方法及装置
-
申请号: CN202111506144.0申请日: 2021-12-10
-
公开(公告)号: CN114492829A公开(公告)日: 2022-05-13
- 发明人: 王晓 , 陈薏竹 , 田永林 , 王飞跃
- 申请人: 中国科学院自动化研究所
- 申请人地址: 北京市海淀区中关村东路95号
- 专利权人: 中国科学院自动化研究所
- 当前专利权人: 中国科学院自动化研究所
- 当前专利权人地址: 北京市海淀区中关村东路95号
- 主分类号: G06N20/00
- IPC分类号: G06N20/00 ; G06F16/22 ; G06F16/2455
摘要:
本发明提供的一种基于联邦学习场景的训练参与方的选择方法及装置,基于数据类别和数据量对数据集进行划分,得到联邦学习场景下每个客户端数据;基于客户端数据对全局模型进行训练,若当前训练轮次小于预设的最大训练轮次,基于线性拟合方法对每个客户端数据进行计算,在客户端中确定当前训练轮次的候选参与方列表;获取当前训练轮次的训练参与方的预设数目;预设数目小于客户端的总数;根据预设数目在当前训练轮次的候选参与方列表中确定当前训练轮次的训练参与方。该方法可以有效降低模型训练过程的通信开销,通过充分利用客户端的信息,选择更优的训练参与方参与本轮次的训练,使模型的训练可以达到更好的效果。