基于数据质量评估的强化联邦学习动态采样方法及设备技术

本申请提供一种基于数据质量评估的强化联邦学习动态采样方法及设备，能够根据初始梯度信息构建初始全局模型，并根据初始全局模型的模型性能确定每个客户端的贡献指数，将贡献指数应用到联邦学习的客户端采样中，可以基于贡献指数评估每个客户端的数据质量。然后根据贡献指数和预设的目标精度确定每个客户端的最优动作价值函数值，因为最优动作价值函数综合考虑了模型性能和模型精度，所以根据最优动作价值函数值对预设数量个客户端进行采样，能够有效地在大量参与联邦学习的客户端中智能化地筛选出高数据质量的客户端，利用具有高数据质量的客户端进行强化联邦学习，可以提高联邦学习得到的全局模型的质量和精度。邦学习得到的全局模型的质量和精度。邦学习得到的全局模型的质量和精度。

【技术实现步骤摘要】
基于数据质量评估的强化联邦学习动态采样方法及设备


[0001]本申请涉及数据处理
，尤其涉及一种基于数据质量评估的强化联邦学习动态采样方法及设备。

技术介绍

[0002]在联邦学习中，参与的客户端数量通常非常庞大且客户端拥有的数据质量复杂多样，因为模型分发和重新上传的带宽受限，在所有参与设备上并行执行模型更新和聚合是不切实际的，所以一般只选取一部分客户端参与联邦学习的训练过程。因此，客户端采样方法对于降低联邦学习的通信开销，提高联邦训练过程中的收敛速度和最终模型精度等至关重要。相关技术中基于客户端上的数据样本数量占整个训练样本的比例进行客户端采样来降低联邦学习的通信开销，但是，简单地将数据量作为评判客户端质量的指标，忽略了数据量大的客户端可能数据质量较低的可能，此时根据数据量选择客户端反而会降低模型质量并影响最终模型精度。

技术实现思路

[0003]有鉴于此，本申请的目的在于提出一种基于数据质量评估的强化联邦学习动态采样方法及设备，用于提高联邦学习得到的模型的质量和精度。
[0004]基于上述目的，本申请的第一方面本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于数据质量评估的强化联邦学习动态采样方法，其特征在于，包括：确定客户端的初始梯度信息；根据所述初始梯度信息构建联邦学习在当前通信回合的初始全局模型；根据所述初始全局模型的模型性能确定每个客户端的贡献指数；根据贡献指数和预设的目标精度，通过深度强化学习确定每个所述客户端的最优动作价值函数值；根据所述最优动作价值函数值对预设数量个客户端进行采样。2.根据权利要求1所述的方法，其特征在于，所述根据所述梯度信息构建联邦学习在当前通信回合的初始全局模型，包括：根据联邦学习的通信轮次数确定历史全局模型；确定每个所述客户端在联邦学习的当前通信轮次的样本量；根据所述初始梯度信息、所述样本量和所述客户端数量确定聚合梯度；根据所述聚合梯度和所述历史全局模型确定联邦学习在当前通信回合的所述初始全局模型。3.根据权利要求1所述的方法，其特征在于，所述模型性能包括标准模型性能和终端模型性能；所述根据所述初始全局模型的模型性能确定每个客户端的贡献指数，包括：根据预设的标准测试集确定所述初始全局模型的所述标准模型性能；根据每个所述客户端的数据集确定所述初始全局模型的所述终端模型性能；根据所述标准模型性能和所述终端模型性能确定每个客户端的贡献指数。4.根据权利要求1所述的方法，其特征在于，所述根据贡献指数和预设的目标精度，通过深度强化学习确定每个所述客户端的最优动作价值函数值，包括：获取联邦学习在所述当前通信回合的上一通信回合的历史模型数据，并根据所述历史模型数据构建深度强化学习的状态空间；根据所述客户端数量确定深度强化学习的动作空间，其中，每个客户端对应所述动作空间内的一个选取动作；根据所述目标精度和所述贡献指数确定所述动作空间中每个所述选取动作对应的即时奖励；根据所述即时奖励和预设的折扣因子确定每个所述选取动作在所述状态空间下的所述最优动作价值函数值。5.根据权利要求4所述的方法...

【专利技术属性】
技术研发人员：梁美玉，赵泽华，杜军平，薛哲，李昂，
申请(专利权)人：北京邮电大学，
类型：发明
国别省市：