TPDM – 西湖大学联合北大等高校推出的时间预测扩散模型

TPDM（Time Prediction Diffusion Model）是西湖大学MAPLE实验室、南方科技大学、北京大学及西湖大学高等研究院先进技术研究所联合推出的图像生成模型，能自适应地调整去噪时间表，优化图像质量和生成效率。模型用即插即用的时间预测模块（TPM），在每个去噪步骤中根据当前隐空间特征预测下一个噪声水平。TPDM基于强化学习进行训练，目标是最大化考虑去噪步数的图像质量奖励，实现与人类偏好一致的高质量图像生成。在Stable Diffusion 3 Medium架构下，TPDM展现出用更少去噪步骤获得更好性能的能力，显著提升图像生成的性能和效率。

官网：https://arxiv.org/pdf/2412.01243

立即打开官网