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