一种面向联邦学习的数据分类同步控制方法及装置、介质
摘要:
本申请涉及联邦学习模型训练进度控制技术领域,涉及一种面向联邦学习的数据分类同步控制方法及装置、介质。所述方法包括:云服务器初始化并下发全局模型;手机节点计算本地数据难度分布,云服务器计算本轮全局模型中的数据难度阈值;手机节点根据阈值筛选符合当前难度阶段的数据,进行本地训练;云服务器接收到模型更新,进行聚合,获得新的全局模型;对新的全局模型进行测试得到测试损失,判别测试损失是否满足切换阶段的条件,若满足则进入下一阶段,若不满足继续当前阶段;重复训练过程至训练完成。本发明以全局同步的方式实现自步学习由简单至困难的学习过程,提升了模型精度和收敛速度,具有良好的泛化性。
0/0