mlsanityData to 7 Oct 2026

Data Warmup: Complexity-Aware Curricula for Efficient Diffusion Training

arXiv 2604.07397 ↗First posted 8 Apr 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

A key inefficiency in diffusion training occurs when a randomly initialized network, lacking visual priors, encounters gradients from the full complexity spectrum--most of which it lacks the capacity to resolve. We propose Data Warmup, a curriculum strategy that schedules training images from simple to complex without modifying the model or loss. Each image is scored offline by a semantic-aware complexity metric combining foreground dominance (how much of the image salient objects occupy) and foreground typicality (how closely the salient content matches learned visual prototypes). A temperature-controlled sampler then prioritizes low-complexity images early and anneals toward uniform sampling. On ImageNet 256x256 with SiT backbones (S/2 to XL/2), Data Warmup improves IS by up to 6.11 and FID by up to 3.41, reaching baseline quality tens of thousands of iterations earlier. Reversing the curriculum (exposing hard images first) degrades performance below the uniform baseline, confirming that the simple-to-complex ordering itself drives the gains. The method combines with orthogonal accelerators such as REPA and requires only ~10 minutes of one-time preprocessing with zero per-iteration overhead.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 28 of its 39 references that are in the map. First posted on arXiv 8 Apr 2026.

Builds on

28 of 39 references.

References · 28

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
  3. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  4. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
  5. 2021Diffusion Models Beat GANs on Image Synthesiscited by 8,185
  6. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
  7. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
  8. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
  9. 2016Improved Techniques for Training GANscited by 2,938
  10. 2015Prioritized Experience Replaycited by 1,070
  11. 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 925
  12. 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformerscited by 707
16 more
  1. 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Thinkcited by 666
  2. 2020Curriculum Learning for Reinforcement Learning Domains: A Framework and Surveycited by 350
  3. 2018Not All Samples Are Created Equal: Deep Learning with Importance Samplingcited by 323
  4. 2021GRAD-MATCH: Gradient Matching based Data Subset Selection for Efficient Deep Model Trainingcited by 269
  5. 2021Generating Images with Sparse Representationscited by 262
  6. 2017Automated Curriculum Learning for Neural Networkscited by 251
  7. 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 233
  8. 2019Competence-based Curriculum Learning for Neural Machine Translationcited by 201
  9. 2023Fast Training of Diffusion Models with Masked Transformerscited by 172
  10. 2015Online Batch Selection for Faster Training of Neural Networkscited by 121
  11. 2024FasterDiT: Towards Faster Diffusion Transformers Training without Architecture Modificationcited by 64
  12. 2024FiTv2: Scalable and Improved Flexible Vision Transformer for Diffusion Modelcited by 15
  13. 2025SARA: Structural and Adversarial Representation Alignment for Training-efficient Diffusion Modelscited by 11
  14. 2025Efficient Generative Model Training via Embedded Representation Warmupcited by 5
  15. 2024From Prototypes to General Distributions: An Efficient Curriculum for Masked Image Modelingcited by 4
  16. 2025Translation from Wearable PPG to 12-Lead ECGcited by 1

Built on by

Not cited yet.