基于边端协同的模型训练方法及相关设备

    公开(公告)号:CN119358637A

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

    申请号:CN202411221451.8

    申请日:2024-09-02

    Abstract: 本申请提供一种基于边端协同的模型训练方法及相关设备,所述方法包括:根据移动设备资源参数对待训练模型进行拆分,得到第一模型和第二模型;基于移动设备的本地数据对第二模型进行知识蒸馏得到蒸馏模型;将蒸馏模型的第一参数发送至数据处理器,得到第二参数,将第二参数上传至服务器;基于第二参数和待训练模型的第三参数,计算得到第四参数;基于第四参数,对第一模型进行更新,得到第三模型和第三模型的第五参数;基于第五参数,对蒸馏模型进行训练,得到第四模型的第六参数;当停止训练,将第三模型和第四模型作为训练好的模型。本申请通过结合联邦学习与拆分学习架构,在有限的算力资源以及较高的对数据隐私要求下推动模型的训练。

Patent Agency Ranking