Reinforcing Few-step Generators via Reward-Tilted Distribution Matching
通过奖励倾斜分布匹配增强少步生成器
机构 * Tencent Hunyuan(腾讯文英) ; Hong Kong University of Science and Technology(香港科技大学) ; Westlake University(西湖大学)
专题命中 效率与蒸馏 :image generation(abstract);text-to-image(abstract);diffusion(abstract);分类 cs.CV
AI总结 提出奖励倾斜分布匹配蒸馏(RTDMD)两阶段框架,结合分布匹配蒸馏与奖励引导强化学习,在仅4步推理下实现文本到图像生成的最新性能。
Comments Code and models are available at https://github.com/Harahan/RTDMD