公开(公告)号：CN118799698A

公开(公告)日：2024-10-18

申请号：CN202410500530.6

申请日：2024-04-24

Applicant: 吉林大学

Inventor： 常毅 ,  石长隆 ,  郭丹丹 ,  李金梦 ,  赵贺

IPC: G06V10/82 ,  G06V10/94 ,  G06V10/96 ,  G06V20/60 ,  G06N3/098 ,  G06F21/62

Abstract: 本申请实施例提供一种网络模型训练方法、装置、电子设备与存储介质，方法包括：向至少两个客户端发送网络模型的第一全局模型参数，触发各客户端基于接收到的第一全局模型参数以及本地数据对网络模型进行训练，并将训练后的客户端模型参数返回至服务器；对接收到的多个客户端模型参数进行聚合，基于聚合后的第二全局模型参数以及第一全局模型参数，确定全局梯度值；基于第一全局模型参数、全局梯度值、预设超参数以及全局学习率，确定第一收缩因子，第一收缩因子为客户端模型参数对应的收缩因子；基于第一收缩因子以及第二全局模型参数对网络模型进行第一权重收缩，得到第一权重收缩后的网络模型。