无需数据即可实现高质量流模型蒸馏,显著加速生成过程。
Flow Map Distillation Without Data
- 从教师模型的先验分布采样,避免数据不匹配风险。
- 仅用1步采样即达ImageNet 256x256上FID 1.45的顶尖性能。
- 适合追求高效、鲁棒生成模型的科研与工程人员。
当前最优的流模型虽生成质量高,但采样速度慢且需迭代。为加速采样,可将流映射从预训练教师模型中蒸馏出来,传统方法依赖外部数据集采样,存在教师-数据不匹配的根本风险,因静态数据集可能无法完整或准确反映教师模型的生成能力。本文质疑这种对数据的依赖是否必要,提出一种完全无数据的替代方案:仅从教师模型保证遵循的先验分布采样,彻底规避不匹配风险。为此,我们设计了一个系统性框架,能预测教师模型的采样路径,并主动校正自身累积误差以保证高保真度。实验表明,该方法超越所有基于数据的蒸馏方法,在使用SiT-XL/2+REPA作为教师模型时,仅用1次采样即在ImageNet 256x256和512x512上分别达到FID 1.45和1.49,显著刷新纪录。本工作为加速生成模型提供了更稳健的新范式,推动无数据流映射蒸馏的广泛应用。
原文摘要 · Abstract (English)
State-of-the-art flow models achieve remarkable quality but require slow, iterative sampling. To accelerate this, flow maps can be distilled from pre-trained teachers, a procedure that conventionally requires sampling from an external dataset. We argue that this data-dependency introduces a fundamental risk of Teacher-Data Mismatch, as a static dataset may provide an incomplete or even misaligned representation of the teacher's full generative capabilities. This leads us to question whether this reliance on data is truly necessary for successful flow map distillation. In this work, we explore a data-free alternative that samples only from the prior distribution, a distribution the teacher is guaranteed to follow by construction, thereby circumventing the mismatch risk entirely. To demonstrate the practical viability of this philosophy, we introduce a principled framework that learns to predict the teacher's sampling path while actively correcting for its own compounding errors to ensure high fidelity. Our approach surpasses all data-based counterparts and establishes a new state-of-the-art by a significant margin. Specifically, distilling from SiT-XL/2+REPA, our method reaches an impressive FID of 1.45 on ImageNet 256x256, and 1.49 on ImageNet 512x512, both with only 1 sampling step. We hope our work establishes a more robust paradigm for accelerating generative models and motivates the broader adoption of flow map distillation without data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。