← 返回时间线

paper

ROAD: Modern Adversarial Training for Simple One-Step Distillation at Scale

arXiv ↗
ID
2610.39539
分类
—
首次捕获
2026-10-30
状态
unread
作者
Xiaoou Li, Shan You, Tao Huang
信号
SI 63

信号历史

  • 2026-10-30Scholar Inbox · 相关分 63
暂无信号数据

摘要

One-step distillation turns pretrained diffusion models into fast generators, but strong results often require substantial training machinery and compute. Pure adversarial distillation offers a simpler route, yet has struggled at modern text-to-image scale. To address this limitation, we propose ROAD (Representation-grounded One-step Adversarial Distillation), a one-stage framework that distills text-to-image models with one adversarial objective and a lightweight representation-space critic. First, we replace the classical convolutional head with a Lipschitz-controlled Transformer head with token-level text conditioning, which matches the semantic tokens of modern visual encoders. Second, we observed a fixed random feature mixing leads to long-horizon degradation. We attribute it to generator-critic co-adaptation around one realization-specific view, and periodically re-draw the mixing to keep the randomization effective. Third, for our dense and vector-valued logits, we introduce Jacobian R1 which penalizes the full input-logit Jacobian, overcoming pooled R1's limitation of neglecting most of logit sensitivities. A single adversarial objective then suffices: in one step, ROAD reaches 11.03 FID on SDXL and 13.73 on SD3.5-M on COCO-10K, surpassing the multi-step teachers at a fraction of distribution matching's training compute.

我的笔记

还没有笔记。

在 GitHub 上写笔记 ↗(新建 content/notes/2610.39539.md,PR 合并后本页自动更新)