当前位置: 首页 > 专利查询>东南大学专利>正文

基于动态本地训练和梯度压缩策略的联邦学习方法技术

技术编号:43136659 阅读:25 留言:0更新日期:2024-10-29 17:41
本发明专利技术公开了一种基于动态本地训练和梯度压缩策略的联邦学习方法。在模型训练时动态调整本地训练策略中终端与中心服务器之间的通信间隔,同时在每次通信时动态调整梯度压缩的量化比特数和稀疏化程度,通过以上方式降低通信开销。本发明专利技术能够解决联邦学习中通信开销过大的问题,在降低模型训练过程通信开销的同时,最大化模型可用性。相较于传统的通信节约型联邦学习方法,本发明专利技术所提出的方法可以获得更好的性能。

【技术实现步骤摘要】

本专利技术属于联邦学习领域,涉及本地训练和梯度压缩,尤其涉及一种基于动态本地训练和梯度压缩的联邦学习方法。


技术介绍

1、联邦学习已被广泛应用于生活的各个方面,在医疗、环保、制造等领域有许多实际应用。然而,尽管联邦学习中终端数据不需要上传至中心服务器,并因此节省通信开销,但在联邦学习的背景下,参与训练的的终端设备,如智能手机和平板电脑,通常面临着通信资源有限,以及成本较高的网络连接问题,因此无法承担过大的通信开销。此外,目前部署在终端设备上的机器学习应用越来越多地采用复杂的深度神经网络。在这种情况下,每个客户端上传的梯度向量维度较高,这使得通信成为了一个严重的瓶颈问题。因此,需要在联邦学习中采用通信开销优化策略进一步减小通信开销。联邦学习数据传输过程的通信开销优化算法包括缩减通信频率和减小单次数据上传的开销。本地训练的方式可以用来降低通信频率,在本地训练中,模型在终端设备上进行多轮迭代,在达到预设的通信间隔后再上传模型的更新量,用于全局模型的更新。在梯度压缩中,可以通过稀疏化算法和量化算法来减少单次上传的数据量。


技术实现思路<本文档来自技高网...

【技术保护点】

1.一种基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,所述方法包括以下步骤:

2.根据权利要求1所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,通信间隔动态机制子问题P1构建为:

3.根据权利要求2所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,采用拉格朗日法解决约束优化问题P1,具体步骤包括:

4.根据权利要求1所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,步骤2中采用本地训练减少通信频率,其表示如下:

5.根据权利要求1所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特...

【技术特征摘要】

1.一种基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,所述方法包括以下步骤:

2.根据权利要求1所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,通信间隔动态机制子问题p1构建为:

3.根据权利要求2所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,采用拉格朗日法解决约束优化问题p1,具体步骤包括:

4.根据权利要求1所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,步骤2中采用本地训练减少通信频率,其表示如下:

5.根据权利要求1所述的基于动态本地训练和梯度压缩策略的联邦学习方法,其特征在于,步骤2中采用稀疏化算法减少单次上传比特,其中稀疏化采用基于随机选取的稀疏化算法rand-k将d维的梯度g转换为其稀疏表示即将g...

【专利技术属性】
技术研发人员:康维詹宁刘楠
申请(专利权)人:东南大学
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1