一种面向数据不平衡分布的节能联邦学习方法

    公开(公告)号:CN118133984A

    公开(公告)日:2024-06-04

    申请号:CN202410235759.1

    申请日:2024-03-01

    Abstract: 本发明涉及联邦学习技术领域,具体涉及一种面向数据不平衡分布的节能联邦学习方法,首先构建企业信用评估场景系统模型,针对多客户端的数据不平衡分布问题,通过为每个客户端构造一个局部代理函数来保证全局损失的平稳下降;其次考虑同步通信的方式,将单轮联邦学习的最大时延作为统一的时延约束。最后在此约束下以最小化系统的能耗为目标,提出一种联合客户端CPU频率和带宽分配的联邦学习方法。本发明提出的联邦学习方法有效缓解了数据不平衡分布对模型训练的影响,可以在更少的迭代次数内实现全局模型的收敛,同时通过数值分析推导出最优的CPU频率,在此基础上采用一种迭代的L‑BGFS‑B算法求解最优的带宽分配,最终可以有效降低系统能耗。

Patent Agency Ranking