arXiv:2512.22802cs.LGcs.CV2025-12

用强化学习让小模型更快生成,少步数也能媲美大模型。

ReDiF: Reinforced Distillation for Few Step Diffusion

  • 用奖励信号引导学生模型探索更优去噪路径。
  • 仅需10步即可达到主流方法20步的生成质量。
  • 通用框架,适配各类扩散模型,提升推理效率。

知识蒸馏可解决扩散模型采样慢的问题,通过训练小型或少步数的学生模型来逼近高步数教师模型的行为。本文提出一种基于强化学习的蒸馏框架,将蒸馏过程建模为策略优化问题,学生模型通过与教师输出对齐的奖励信号进行训练。该方法动态引导学生探索多种去噪路径,使其能采取更长、更优的步骤直接逼近数据分布的高概率区域,而非依赖逐级精修。框架充分利用扩散模型处理大步长的能力,有效管理生成过程。实验表明,本方法在显著减少推理步数和计算资源的前提下,性能优于现有蒸馏技术。此外,该框架具有模型无关性,适用于任何具备合适奖励函数的扩散模型,为高效扩散学习提供通用优化范式。

原文摘要 · Abstract (English)

Distillation addresses the slow sampling problem in diffusion models by creating models with smaller size or fewer steps that approximate the behavior of high-step teachers. In this work, we propose a reinforcement learning based distillation framework for diffusion models. Instead of relying on fixed reconstruction or consistency losses, we treat the distillation process as a policy optimization problem, where the student is trained using a reward signal derived from alignment with the teacher's outputs. This RL driven approach dynamically guides the student to explore multiple denoising paths, allowing it to take longer, optimized steps toward high-probability regions of the data distribution, rather than relying on incremental refinements. Our framework utilizes the inherent ability of diffusion models to handle larger steps and effectively manage the generative process. Experimental results show that our method achieves superior performance with significantly fewer inference steps and computational resources compared to existing distillation techniques. Additionally, the framework is model agnostic, applicable to any type of diffusion models with suitable reward functions, providing a general optimization paradigm for efficient diffusion learning.

扩散模型知识蒸馏强化学习高效生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。