一种基于客户端样本熵感知和联邦迁移学习的网络模型训练方法

    公开(公告)号:CN119250130A

    公开(公告)日:2025-01-03

    申请号:CN202411192991.8

    申请日:2024-08-28

    Applicant: 上海大学

    Abstract: 本发明涉及计算机深度学习技术领域,公开了一种基于客户端样本熵感知和联邦迁移学习的网络模型训练方法,该方法是将待训练的网络模型分别搭载在中央服务器和多个源客户端中,采用训练样本对搭载在源客户端中的本地模型进行训练,得到训练后本地模型;搭载在中央服务器中的全局模型根据接收的本地网络模型参数更新全局模型网络参数;然后将更新后的全局模型下载到各个源客户端中,采用验证样本集对下载至每个源客户端中的更新后的全局模型进行任务验证,并根据验证损失对全局模型网络参数进行优化,得到更新全局模型;将更新全局模型下载到每个源客户端,采用验证样本集对下载至每个源客户端的更新全局模型进行任务验证,任务验证完成后,根据每个源客户端中所述更新训练后本地模型、更新全局模型的任务验证性能来更新所述更新训练后本地模型的网络参数;重复上述步骤直至全局模型达到更新预设值或预设性能,完成全局模型训练。

Patent Agency Ranking