一种基于掩码的图像生成模型训练效率提升方法

    公开(公告)号:CN117876520A

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

    申请号:CN202410055134.7

    申请日:2024-01-15

    Applicant: 浙江大学

    Abstract: 本发明公开了一种基于掩码的图像生成模型训练效率提升方法,属于图像生成技术领域。通过获取大量训练图像,确定图像数据的掩码方式和掩码率,得到数据集;在数据集中随时采样N张图像并分别进行随机遮盖;通过添加噪声和去噪训练在遮盖后的剩余图像上训练扩散模型,在完整图像上进一步训练扩散模型,直至模型收敛;最后基于收敛后的模型进行采样生成图片。本发明能够在不引入额外训练开销,保持低复杂度的情况下,直接在图像空间提升扩散模型训练效率,且提升了模型生成图像的质量。

Patent Agency Ranking