一种联邦学习模型优化方法、设备、介质及程序产品

    公开(公告)号:CN118036768A

    公开(公告)日:2024-05-14

    申请号:CN202410148779.5

    申请日:2024-02-01

    Abstract: 本发明公开了一种联邦学习模型优化方法、设备、介质及程序产品,所述方法包括:获取边缘端采集的多属性数据,并计算边缘端的数据量因子、数据质量因子以及数据分布特征因子,从而得到边缘端的可信度;选取可信边缘端进行全局模型的参数更新;向可信边缘端发送第一全局参数,以使可信边缘端进行本地模型的训练,并得到本地模型参数;根据本地模型的平均误差以及预设的全局模型学习率,计算得到动态学习率;根据第一全局参数、动态学习率、可信边缘端的可信度以及本地模型参数,计算得到第二全局参数,实现全局模型的参数更新。本发明实施例解决了数据异构所导致的全局模型性能低下的问题,实现了全局参数更新速度和算法稳定度的平衡。

Patent Agency Ranking