arXiv:2507.05914cs.LG2025-07被引 6

用极少量数据训练高质量扩散模型,速度提升百倍以上。

Accelerating Diffusion Model Training under Minimal Budgets: A Condensation-Based Perspective

  • 从海量数据中精选关键样本,再注入丰富语义信息增强训练信号。
  • 仅用0.8%的ImageNet数据,40k步即达FID 4.3,速度比原版快233倍。
  • 适合资源有限却想快速训练高性能生成模型的研究者。

扩散模型在众多生成任务中表现卓越,但从头训练极为耗资源,通常需数百万张图片和大量GPU天数。受数据瓶颈启发,本文提出一种数据浓缩视角:给定大规模训练集,目标是构建一个极小的浓缩数据集,在极低数据与计算预算下仍能支持强扩散模型训练。为此,我们提出扩散数据浓缩框架D2C,包含选择与附加两阶段:第一阶段结合扩散难度评分与间隔采样,从原始数据中选出紧凑且信息量高的子集;第二阶段在此基础上,通过注入丰富的语义与视觉表征来强化条件信号。据我们所知,D2C是首个系统研究扩散模型数据浓缩的框架,而以往方法主要针对判别性架构。在多种数据预算(0.8%-8% ImageNet)、模型架构与图像分辨率下的实验表明,D2C显著加速训练并保持高生成质量。在ImageNet 256x256上,使用SiT-XL/2模型,仅0.8%数据、40k步即可达到FID 4.3,相比原版SiT-XL/2和SiT-XL/2 + REPA分别快233倍与100倍。

原文摘要 · Abstract (English)

Diffusion models have achieved remarkable performance on a wide range of generative tasks, yet training them from scratch is notoriously resource-intensive, typically requiring millions of training images and many GPU days. Motivated by a data-centric view of this bottleneck, we adopt a condensation-based perspective: given a large training set, the goal is to construct a much smaller condensed dataset that still supports training strong diffusion models under minimal data and compute budgets. To operationalize this perspective, we introduce Diffusion Dataset Condensation (D2C), a two-phase framework comprising Select and Attach. In the Select phase, a diffusion difficulty score combined with interval sampling is used to identify a compact, informative training subset from the original data. Building on this subset, the Attach phase further strengthens the conditional signals by augmenting each selected image with rich semantic and visual representations. To our knowledge, D2C is the first framework that systematically investigates dataset condensation for diffusion models, whereas prior condensation methods have mainly targeted discriminative architectures. Extensive experiments across data budgets (0.8%-8% of ImageNet), model architectures, and image resolutions demonstrate that D2C dramatically accelerates diffusion model training while preserving high generative quality. On ImageNet 256x256 with SiT-XL/2, D2C attains an FID of 4.3 in just 40k steps using only 0.8% of the training images, corresponding to about 233x and 100x faster training than vanilla SiT-XL/2 and SiT-XL/2 + REPA, respectively.

扩散模型数据浓缩高效训练生成模型

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