基于扩散驱动数据回放的联邦类持续学习图像分类方法

    公开(公告)号:CN118710996B

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

    申请号:CN202411191850.4

    申请日:2024-08-28

    Abstract: 本发明公开了一种基于扩散驱动数据回放的联邦类持续学习图像分类方法,所述方法包括:初始化阶段,服务器初始化预训练扩散模型和图像分类模型并发送给客户端;类别逆向编码阶段,客户端对新任务类别进行逆向编码,得到类别嵌入向量;图像生成阶段,利用类别嵌入向量生成图像;图像分类模型训练阶段,各个客户端用生成数据和合成数据训练图像分类模型。对于联邦类持续学习图像分类,其关键难点在于灾难性遗忘,即模型在学习新任务时遗忘旧任务的知识,为解决这一难点,通过预训练扩散模型对各个类别数据进行逆向工程,得到类别嵌入向量,在学习新任务时利用类别嵌入向量对历史任务数据进行生成回放,有效缓解灾难性遗忘。

    基于扩散驱动数据回放的联邦类持续学习图像分类方法

    公开(公告)号:CN118710996A

    公开(公告)日:2024-09-27

    申请号:CN202411191850.4

    申请日:2024-08-28

    Abstract: 本发明公开了一种基于扩散驱动数据回放的联邦类持续学习图像分类方法,所述方法包括:初始化阶段,服务器初始化预训练扩散模型和图像分类模型并发送给客户端;类别逆向编码阶段,客户端对新任务类别进行逆向编码,得到类别嵌入向量;图像生成阶段,利用类别嵌入向量生成图像;图像分类模型训练阶段,各个客户端用生成数据和合成数据训练图像分类模型。对于联邦类持续学习图像分类,其关键难点在于灾难性遗忘,即模型在学习新任务时遗忘旧任务的知识,为解决这一难点,通过预训练扩散模型对各个类别数据进行逆向工程,得到类别嵌入向量,在学习新任务时利用类别嵌入向量对历史任务数据进行生成回放,有效缓解灾难性遗忘。

Patent Agency Ranking