DOLLAR: Few-Step Video Generation via Distillation and Latent Reward Optimization¶
会议: ICCV 2025
arXiv: 2412.15689
代码: 无
领域: 视频生成 / 扩散模型加速
关键词: video generation, distillation, consistency distillation, latent reward, few-step generation
一句话总结¶
结合变分分数蒸馏(VSD)和一致性蒸馏实现少步视频生成,同时提出潜空间奖励模型微调方法进一步优化特定质量维度,4步student模型在VBench上达82.57分超越teacher模型和Gen-3/Kling等商业基线,1步蒸馏实现278.6倍采样加速。
研究背景与动机¶
领域现状:扩散概率模型在视频生成领域取得了显著进展,能够生成高质量的文本到视频内容。然而,这些模型通常需要50步以上的迭代采样才能生成满意的视频,计算开销极大。
现有痛点:直接减少采样步数往往会导致视频质量退化或生成多样性下降。现有的图像领域蒸馏方法(如SANA-Sprint可实现1-4步图像生成)在视频领域面临额外挑战——视频需要在时间维度上维持一致性,简单减少步数极易产生时间闪烁和质量下降。此外,现有视频蒸馏方法通常仅支持特定步数(如仅能4步采样),缺乏步数灵活性。
核心矛盾:如何在大幅减少采样步数的同时保持视频的质量和多样性?如何通过奖励信号进一步优化蒸馏后模型的特定质量指标?
本文目标:提出DOLLAR框架,通过混合蒸馏策略实现灵活步数的高质量视频生成,并引入潜空间奖励微调进一步提升特定质量维度。
切入角度:将VSD(保多样性)和一致性蒸馏(保质量与步数灵活性)互补结合,同时在潜空间而非像素空间进行奖励优化以降低显存开销。
核心 idea:混合蒸馏+潜空间奖励微调=少步高质量视频生成。
方法详解¶
整体框架¶
DOLLAR采用两阶段训练策略:(1) 混合蒸馏阶段——结合变分分数蒸馏(VSD)和一致性蒸馏(CD),将teacher模型的50步采样能力压缩至student模型的1-4步;(2) 潜空间奖励优化阶段——使用latent reward model对蒸馏后的student模型进行微调,针对性地提升特定质量指标。
关键设计¶
-
变分分数蒸馏(VSD):
- 功能:通过分布匹配将teacher模型的多步采样分布对齐到student模型的少步分布
- 核心思路:VSD最小化teacher和student输出分布之间的KL散度。设teacher模型为 \(\epsilon_\phi\),student模型为 \(\epsilon_\theta\),VSD优化目标为 \(\mathcal{L}_{\text{VSD}} = \mathbb{E}_{t,\epsilon}\left[\|\epsilon_\theta(x_t, t) - \epsilon_\phi(x_t, t)\|^2\right]\),其中 \(x_t\) 为加噪后的视频latent
- 设计动机:纯一致性蒸馏可能导致模式坍缩(mode collapse),VSD通过分布匹配保证生成多样性
-
一致性蒸馏(CD):
- 功能:确保student模型在不同步数下输出一致的结果
- 核心思路:对同一噪声输入,要求student在任意中间时间步的预测结果一致,即 \(f_\theta(x_t, t) \approx f_\theta(x_{t'}, t')\),其中 \(x_t\) 和 \(x_{t'}\) 位于同一PF-ODE轨迹上
- 设计动机:使student不受限于固定步数,可在1-4步之间灵活选择,避免step-specific训练
-
混合蒸馏策略:
- 功能:将VSD和CD以加权方式联合训练
- 核心思路:总损失为 \(\mathcal{L} = \lambda_{\text{VSD}} \mathcal{L}_{\text{VSD}} + \lambda_{\text{CD}} \mathcal{L}_{\text{CD}}\)
- 设计动机:VSD保多样性,CD保质量和步数灵活性,两者互补
-
潜空间奖励模型微调(Latent Reward Fine-tuning):
- 功能:利用潜空间奖励模型对蒸馏后的student进一步微调,提升指定质量维度
- 核心思路:不要求reward model可微,而是在latent空间中操作。通过在潜空间直接计算奖励信号,避免将视频解码到像素空间,大幅降低GPU显存需求。可针对任意奖励指标(美学质量、文本对齐、时间一致性等)进行优化
- 设计动机:传统奖励微调需要在像素空间计算梯度,对于128帧视频显存开销不可接受;潜空间操作同时解决显存和可微性两个问题
损失函数 / 训练策略¶
- 第一阶段:VSD loss + Consistency loss 联合训练,权重在训练过程中动态调整
- 第二阶段:Latent reward fine-tuning,使用策略梯度方法优化reward指标
- 训练数据:使用teacher模型的50步DDIM采样结果作为目标分布
- 视频规格:128帧@12FPS(约10秒),远超之前大多数方法(2-4秒)
实验关键数据¶
主实验¶
| 方法 | 步数 | VBench Score | 加速比 |
|---|---|---|---|
| Teacher (50-step DDIM) | 50 | ~81 | 1× |
| DOLLAR (4-step) | 4 | 82.57 | 12.5× |
| DOLLAR (1-step) | 1 | ~79 | 278.6× |
| Gen-3 | - | <82.57 | - |
| T2V-Turbo | 4 | <82.57 | - |
| Kling | - | <82.57 | - |
消融实验¶
| 配置 | VBench Score | 说明 |
|---|---|---|
| 仅VSD | ~80 | 多样性好但质量不足 |
| 仅CD | ~79 | 质量好但多样性差 |
| VSD + CD | ~81.5 | 互补提升 |
| VSD + CD + Latent Reward | 82.57 | 奖励微调进一步突破 |
关键发现¶
- 4步student模型在VBench上超越50步teacher模型,蒸馏+奖励微调可超越原始模型
- 1步蒸馏实现278.6倍加速,接近实时生成
- 人类评估进一步验证4步student优于50步teacher
- 潜空间奖励微调使蒸馏后模型在特定维度(如时间一致性)上显著提升
亮点与洞察¶
- 蒸馏+奖励=超越teacher:蒸馏后的student不仅更快,还能通过奖励微调在质量上超越teacher,打破"蒸馏必然损失质量"的常见认知
- 潜空间而非像素空间:在潜空间进行奖励计算大幅降低显存需求,使长视频(10秒128帧)的奖励微调变得可行
- 步数灵活性:一致性蒸馏赋予的步数灵活性(1-4步均可用)为不同应用场景提供了质量-速度的灵活权衡
- 10秒长视频验证:在128帧@12FPS的10秒视频上验证,远超之前方法仅在短视频上的实验
局限与展望¶
- 论文未公开代码,可复现性有待验证
- 潜空间奖励模型的具体设计细节和训练过程描述不够充分
- 仅在VBench上验证,缺少更多元化的视频质量评估基准
- 未讨论更长视频(如30秒以上)或更高分辨率场景的适用性
- 奖励微调依赖奖励模型质量,存在over-optimization风险
相关工作与启发¶
- Consistency Models:Song et al. 提出的一致性模型是本文CD组件的理论基础
- ProlificDreamer/VSD:Wang et al. 提出的VSD被本文拓展到视频蒸馏
- SANA-Sprint:图像领域的1-4步生成先行工作
- T2V-Turbo:之前的视频蒸馏工作,本文性能超越
- 启发:潜空间奖励微调的思路可推广到其他生成模型(3D生成、音频合成等)
评分¶
- 新颖性: ⭐⭐⭐⭐ VSD+CD混合蒸馏和潜空间奖励微调的结合在视频生成领域较新
- 实验充分度: ⭐⭐⭐ VBench结果有说服力,但缺消融细节和多基准对比
- 写作质量: ⭐⭐⭐⭐ 结构清晰,动机明确
- 价值: ⭐⭐⭐⭐ 接近实时的视频生成具有重要应用价值
title: >- [论文解读] DOLLAR: Few-Step Video Generation via Distillation and Latent Reward Optimization description: >- [ICCV 2025][视频理解][video generation] 结合变分分数蒸馏(VSD)和一致性蒸馏实现few-step视频生成,同时提出潜空间奖励模型微调方法进一步优化生成质量,4步生成的10秒视频(128帧@12FPS)在VBench上达82.57分超越teacher模型和Gen-3/Kling等基线,1步蒸馏实现278.6倍加速。 tags: - ICCV 2025 - 视频理解 - video generation - distillation - consistency distillation - latent reward - few-step - VBench