---
title: Diffusion Dataset Condensation (D2C)
url: https://www.emergentmind.com/topics/diffusion-dataset-condensation-d2c
type: topic
---

# Diffusion Dataset Condensation (D2C)

Searching arXiv for the cited papers to ground the article in the current literature.
Diffusion Dataset Condensation (D2C) denotes the construction of a substantially smaller training set that still supports effective diffusion-model training from scratch under constrained optimization budgets. In current usage, the term refers both to a general problem setting for generative modeling and, more narrowly, to the two-phase D\textsuperscript{2}C framework introduced for class-conditional image diffusion training. Across recent work, the central departure from classical discriminative condensation is consistent: diffusion models must preserve the structure of the data distribution itself—its support, local neighborhoods, mode coverage, semantics, and realism—rather than merely class-separating cues. This has led to several distinct formulations, including diffusion-difficulty-based selection with attached conditioning priors, geometry-aware real-subset selection via partial optimal transport, diffusion-generated candidate pools followed by subset extraction, and coreset compression followed by on-the-fly diffusion expansion [2507.05914][2606.05883][2505.06670][2412.04668].

## 1. Problem setting and formal scope

The common problem setting starts from a large labeled dataset and asks for a much smaller representation that remains useful for downstream diffusion training. In the notation of D\textsuperscript{2}C, the full dataset is
\[
\mathcal{D} = (\hat{\mathbf{X}}, \hat{\mathbf{Y}}) = \{(\hat{\mathbf{x}}_i, \hat{y}_i)\}_{i=1}^{|\mathcal{D}|},
\]
and the condensed dataset is
\[
\mathcal{D}^{\mathcal{S}} = (\mathbf{X}, \mathbf{Y}) = \{(\mathbf{x}_j, y_j)\}_{j=1}^{|\mathcal{D}^{\mathcal{S}}|}, \qquad |\mathcal{D}^{\mathcal{S}}| \ll |\mathcal{D}|.
\]
For class-conditional generation, the same paper writes
\[
\mathcal{D} = \bigcup_{y=1}^{C} \mathcal{D}_y,\qquad \mathcal{D}_y = \{x_i\}_{i=1}^{|\mathcal{D}_y|},
\]
and evaluates condensation by image-generation quality metrics such as gFID, sFID, IS, Precision, and Recall rather than classification accuracy [2507.05914].

A closely related formulation appears in geometry-aware real subset selection for diffusion training. Given a labeled dataset
\[
T=\bigcup_{c=1}^{C}T_c,
\]
the goal is to select a much smaller real subset
\[
S=\bigcup_{c=1}^{C}S_c,\qquad S_c\subset T_c,\qquad |S_c|\ll |T_c|,
\]
typically with an equal number of selected samples per class. The downstream assumption is explicit: a diffusion model is then trained from scratch on \(S\) under a limited training budget, and the subset should support generative performance close to that obtained from the full data [2606.05883].

The underlying diffusion objective in D\textsuperscript{2}C is the standard noise-prediction loss
\[
\mathcal{L}_{\text{diff}} = \mathbb{E}_{\mathbf{x}_0\sim q_0(\mathbf{x}),\, \epsilon\sim\mathcal{N}(0,\mathbf{I}),\, t\sim \mathcal{U}[0,1]} \left[ \| \epsilon - \epsilon_\theta(\mathbf{x}_t, t, \mathbf{c}) \|_2^2 \right],
\]
with forward noising
\[
q_t(\mathbf{x}_t \mid \mathbf{x}_0) = \mathcal{N}(\mathbf{x}_t;\alpha_t \mathbf{x}_0,\sigma_t^2 \mathbf{I}), \qquad \mathbf{x}_t = \alpha_t \mathbf{x}_0 + \sigma_t \epsilon.
\]
This formalization makes the central constraint plain: the condensed representation must remain a usable surrogate for a likelihood-based generative objective, not only for class prediction [2507.05914].

## 2. Methodological paradigms

Recent work uses markedly different condensed representations and roles for diffusion. The principal variants can be organized as follows.

| Method | Condensed representation | Principal mechanism |
|---|---|---|
| D\textsuperscript{2}C [2507.05914] | Real-image subset with attached semantic and visual features | Select and Attach |
| Geometry-aware subset selection [2606.05883] | Compact subset of real images | One-sided partial OT with statistics and confidence regularization |
| Video condensation with diffusion [2505.06670] | Representative subset of diffusion-generated synthetic videos | Generate synthetic pool, then select with VST-UNet or TAC-DT |
| Diffusion-augmented coreset expansion [2412.04668] | Real informative patches, seeds, and teacher soft labels | Patch coreset plus on-the-fly LDM expansion |

D\textsuperscript{2}C, introduced as “Diffusion Dataset Condensation: Training Your Diffusion Model Faster with Less Data,” is explicit that its condensed dataset is not synthetic pixels optimized end-to-end. It is better described as an augmented subset of real samples plus attached side information. The framework has two phases: Select, which chooses a compact and diverse subset using a diffusion-difficulty score and interval sampling, and Attach, which enriches the chosen samples with semantic representations from a pretrained text encoder and visual representations from a pretrained vision encoder [2507.05914].

The geometry-aware method of 2026 pushes further toward real-subset selection and argues that the condensation objective itself must be redesigned around the needs of diffusion likelihood training. Its target object is a compact subset of real images whose feature-space arrangement preserves “distributional support geometry”: local neighborhoods, mode coverage, and support allocation. The paper treats previous ranking-based methods, including the prior diffusion-specific selector \(D^2C\) of Huang et al., as inadequate because a scalar ranking collapses a high-dimensional, multimodal distribution [2606.05883].

The video formulation differs again. There, diffusion is used to generate a large synthetic candidate pool \(\mathcal{S}_g=\mathcal{F}_\theta(\mathcal{T})\), with \(N_c=100\) generated synthetic videos per class, after which condensation is handled mainly as subset extraction from \(\mathcal{S}_g\). This is a generate-then-select paradigm rather than direct optimization of final exemplars [2505.06670].

The coreset-expansion line is adjacent rather than identical to direct D2C. It first stores informative low-resolution real patches and then uses a pretrained latent diffusion model to expand, super-resolve, and diversify them dynamically during student training. The stored distilled representation is procedural rather than a fixed set of final images [2412.04668].

## 3. Selection objectives and geometry preservation

The Select phase of D\textsuperscript{2}C is organized around a diffusion difficulty score. Using a pretrained class-conditional diffusion model, the method ranks each sample by a class-conditional likelihood proxy derived from diffusion loss:
\[
s_\text{diff}(\mathbf{x}) =  - p_\theta(\mathbf{x}|\mathbf{c}) = -\log\left(\exp\left(-\mathbb{E}_{\epsilon\sim\mathcal{N}(0,\mathbf{I}),t\sim\mathcal{U}[0,1]} \left[ \left\| \epsilon - \epsilon_\theta\left( \alpha_{t} \mathbf{x} + \sigma_{t} \epsilon, \, t, \, \mathbf{c} \right) \right\|^2_2 \right]\right)\right).
\]
Operationally, the score is effectively based on per-sample denoising MSE under a pretrained diffusion model. Samples are sorted class-wise and selected by interval sampling,
\[
\mathcal{D}_{\text{IS}} = \bigcup_{y=1}^{C} \left\{ x^{(i)} \in \mathcal{D}_y \;\middle|\; i \in \{0, k, 2k, \dots \} \right\},
\]
with \(k=96\) for the 10K subset, \(k=16\) for the 50K subset, and \(k=10\) for the 100K subset. The stated rationale is a trade-off between learnability and diversity: very easy samples aid early convergence but under-cover the distribution, whereas very hard samples are cluttered or noisy and can destabilize training [2507.05914].

The geometry-aware alternative replaces scalar ranking with a distribution-alignment objective in representation space. Working class-wise, with selected embeddings \(S_c^{(e)}=\{x_i\}_{i=1}^m\) and full-data embeddings \(T_c^{(e)}=\{y_j\}_{j=1}^n\), it defines the pairwise cost
\[
C_{ij} = \|x_i-y_j\|_2^2.
\]
The paper recalls balanced optimal transport,
\[
\min_{\pi \ge 0} \langle C,\pi\rangle \quad \text{s.t.} \quad \pi 1_n = \mu,\; \pi^\top 1_m = \nu,
\]
but argues that this is ill-suited under severe compression because balanced OT forces all target mass to be matched. Its replacement is a one-sided partial optimal transport problem,
\[
\min_{\pi \ge 0} \langle C,\pi\rangle \quad \text{s.t.} \quad \pi 1_n = \mu,\; \pi^\top 1_m \le \kappa \bar{\nu},
\]
with
\[
\mu = \frac{1}{m}1_m,\qquad \bar{\nu} = \frac{1}{n}1_n,
\]
so that, when \(\kappa>1\), some target mass can remain unmatched [2606.05883].

The same method implements this through a dummy-source reformulation and entropy-regularized Sinkhorn iterations, then optimizes the one-sided transport loss
\[
L_{OT} = \langle C,T_{\text{real}}\rangle,
\]
augmented by feature-statistics regularization
\[
L_{\text{sta}} = \|\mu_S - \mu_T\|_2^2 + \|\sigma_S^2 - \sigma_T^2\|_2^2
\]
and semantic consistency / confidence regularization
\[
L_{\text{conf}} = \frac{1}{m}\sum_{i=1}^{m} -\log p(c|x_i).
\]
The final class-wise objective is
\[
L = L_{OT} + \alpha L_{\text{sta}} + \beta L_{\text{conf}}.
\]
This objective is optimized over discrete subset members by a two-stage procedure: greedy construction using marginal gains, followed by swap-based refinement. The paper’s interpretation is that one-sided POT concentrates transport on the dense, geometrically stable support while avoiding forced matching to low-density tails, thereby preserving the neighborhood and support geometry needed for score matching [2606.05883].

Taken together, these two lines expose a central conceptual divide within D2C. The original D\textsuperscript{2}C framework treats diffusion loss as a useful scalar proxy for sample usefulness. The geometry-aware critique treats that scalarization as fundamentally insufficient because distinct modes can share similar scores while nearby manifold points can be separated by ranking. This suggests that “difficulty” and “geometry” are not interchangeable design primitives in diffusion condensation.

## 4. Conditioning enrichment, generation, and expansion

The Attach phase of D\textsuperscript{2}C enriches each selected image with semantic and visual priors. For class \(c\), a prompt \(P(c)\) is encoded by a pretrained T5 encoder to obtain
\[
t_c, t_{\mathrm{mask}} = f_{\text{text}}(P(c)).
\]
These text features are fused with a learnable class embedding \(e_c\) through
\[
\tilde{t}_c = \text{Conv1d}(t_c \times t_\textrm{mask}),\qquad y_\text{text} = \text{MLP}(\tilde{t}_c) + \tilde{t}_c + e_c.
\]
Instance-level visual priors are extracted with DINOv2-B,
\[
y_\text{vis} = f_\text{vis}(x) \in \mathbb{R}^{N \times d_\text{text}},
\]
and truncated to the first \(h\) tokens. The diffusion model is then trained with semantic conditioning and an auxiliary projection loss
\[
\mathcal{L}_{\text{proj}} = - \frac{1}{h} \sum_{i=1}^h \left\langle \frac{\phi(h_i)}{\|\phi(h_i)\|}, \frac{v_i}{\|v_i\|} \right\rangle,
\]
leading to
\[
\mathcal{L}_{\text{total}} = \mathcal{L}_{\text{diff}} + \lambda\,\mathbb{E}[\mathcal{L}_{\text{proj}}],
\]
with default \(\lambda=0.5\). In implementation, the framework uses a pretrained DiT-XL/2 for scoring, a T5 encoder for text features, DINOv2-B for visual features, and a Stable Diffusion VAE for latent encoding and decoding [2507.05914].

In video condensation, diffusion plays a different role. A pretrained class-conditional Latte model is fine-tuned on all real video datasets, using randomly sampled 16-frame clips. After training, it generates \(N_c=100\) synthetic videos per class to form a reusable candidate pool. Subset selection is then performed either by VST-UNet, a 4D U-Net operating on latent videos, or by TAC-DT, which embeds videos with pretrained VideoMAE, reduces dimensionality by PCA, applies BIRCH hierarchical clustering, and selects representative videos from clusters. The distilled output contains exactly VPC videos per class, with evaluations at VPC \(=1,5,10\) [2505.06670].

The coreset-expansion approach uses diffusion not for subset scoring but for procedural reconstruction and augmentation. For each informative patch \(x_i^*\), it first upsamples to \(\hat{x}_i^*=\mathrm{INTERP}(x_i^*)\), encodes into latent space \(\hat{z}_i^*=\mathrm{Encode}(\hat{x}_i^*)\), adds partial noise at level \(\rho=t'/T\), and then performs a few reverse-diffusion steps with SDXL-Turbo. The resulting decoded image \(\bar{x}_i^*=\mathrm{Decode}(\bar{z}_i^*)\) is a high-resolution training view derived from the real patch anchor. The stored representation is effectively
\[
\{(\text{patch},\text{seed},\text{soft label})\},
\]
and the full-resolution images are regenerated on the fly during student training. The same work also introduces latent-space mixup,
\[
\hat{z}_{\text{interp}} = \gamma \hat{z}_1^* + (1-\gamma)\hat{z}_2^*,
\]
to further diversify diffusion-anchored augmentation [2412.04668].

These mechanisms clarify that the object being “condensed” varies substantially across the literature. In one case it is a real-image subset with side information; in another it is a subset of diffusion-generated videos; in another it is a low-resolution coreset plus procedural generation metadata. The shared goal is compressed training utility for downstream models, but the representation class is method-dependent.

## 5. Empirical performance and characteristic regimes

On ImageNet-1K, D\textsuperscript{2}C reports three retained-data budgets—10K, 50K, and 100K images, corresponding to 0.8%, 4.0%, and 8.0% of the training set—and evaluates on DiT-L/2, SiT-L/2, and SiT-XL/2 at 256×256 and 512×512. Its headline result is on SiT-XL/2 with 10K images: gFID-50K \(=4.23\) at 40K iterations, without CFG, which the paper states is over 100× acceleration over REPA and 233× over vanilla SiT. At 50K images with CFG \(=1.5\), it reports FID \(2.78\) at 180K steps. On 256×256 ImageNet with 10K images and 100K iterations, the framework reports DiT-L/2 FID \(4.20\) and SiT-L/2 FID \(3.98\), while the discriminative condensation baseline SRe\textsuperscript{2}L records \(104.2\) and \(82.3\), respectively. At 512×512 with 10K images and 100K iterations, D\textsuperscript{2}C reports DiT-L/2 gFID \(14.8\), sFID \(6.9\), IS \(109.2\), Precision \(0.63\), and Recall \(0.52\) [2507.05914].

The geometry-aware method evaluates the same ImageNet-1K setting but argues that the D\textsuperscript{2}C ranking objective preserves the wrong object for diffusion training. On ImageNet 256×256 with DiT-L/2 and 100K iterations, its FID-50K is \(3.43\) at the 10K budget, compared with \(4.20\) for \(D^2C\); at 50K and 100K budgets it reports \(11.01\) and \(17.09\), compared with \(14.81\) and \(22.55\) for \(D^2C\). At 512×512 with DiT-L/2 and a 10K budget, it reports FID \(6.17\), IS \(451.0\), Precision \(0.81\), and Recall \(0.67\), compared with \(14.8\), \(109.2\), \(0.63\), and \(0.52\) for \(D^2C\). On SiT-L/2 at 512×512, it reports FID \(5.8\), IS \(461.6\), Precision \(0.81\), and Recall \(0.66\), compared with \(9.1\), \(261.7\), \(0.72\), and \(0.34\). Its ablations further show that removing \(L_{OT}\), \(L_{\text{sta}}\), or \(L_{\text{conf}}\), or replacing partial OT with balanced OT, degrades performance; the full model reaches FID \(3.43\) at 10K images, compared with \(3.54\) for balanced OT and \(4.62\) without \(L_{\text{sta}}\). Runtime on ImageNet with 10K selected images is reported as 5.5 hours on 1 RTX 3090, versus 30.4 hours for DQ and 41.9 hours for \(D^2C\); with 8 GPUs it is 96 minutes, versus 238 minutes and 314 minutes [2606.05883].

The video literature reports a different evaluation regime because downstream training targets action recognition rather than diffusion generation quality. On MiniUCF at VPC \(=5\), VST-UNet achieves \(41.81\) and TAC-DT \(41.47\), compared with \(31.2\) for the best prior FRePo+VD; the paper describes the largest gain as \(10.61\%\). On HMDB51 at VPC \(=5\), VST-UNet gives \(18.05\) and TAC-DT \(19.62\). On Kinetics400 and SSv2, the main table reports top-5 improvements at both VPC \(=1\) and \(5\). An ablation on MiniUCF shows that CE only yields \(31.77\), CE+Div \(40.18\), CE+Rep \(39.80\), and CE+Div+Rep \(41.81\), supporting the diversity and representativeness design [2505.06670].

The diffusion-augmented coreset-expansion approach evaluates standard dataset distillation benchmarks by downstream classification accuracy. On ImageNet-1k it reports \(13.9\) at IPC \(=1\) versus \(6.6 \pm 0.2\) for RDED, and \(52.1\) at IPC \(=10\) versus \(42.0 \pm 0.1\). On ImageNette it reports \(51.4\) at IPC \(=1\), \(73.6\) at IPC \(=10\), and \(87.6\) at IPC \(=50\), compared with \(35.8 \pm 1.0\), \(61.4 \pm 0.4\), and \(80.4 \pm 0.4\) for RDED. Its ablations further show that “Only text cond.” is weaker than anchored super-resolution, and that “Superres + Aug + Mixup” is strongest, reaching \(25.0\) on ImageWoof and \(51.4\) on ImageNette at IPC \(=1\) [2412.04668].

A consistent empirical pattern runs across these works. The strongest gains typically occur under severe compression and limited training budgets. This suggests that D2C is especially consequential in the small-data, early-iteration regime, where sample selection quality, semantic enrichment, and geometry preservation materially alter the optimization trajectory.

## 6. Limitations, misconceptions, and open directions

A common misconception is that diffusion dataset condensation is synonymous with direct optimization of a tiny synthetic image set. The current literature does not support that equivalence. D\textsuperscript{2}C selects real images and attaches side information; the geometry-aware method selects only real images; the video method selects representative synthetic videos from a diffusion-generated pool; and the coreset-expansion method stores real patches plus procedural metadata rather than final images [2507.05914][2606.05883][2505.06670][2412.04668].

A second misconception is that generic discriminative condensation transfers directly to diffusion training. The evidence cited by D\textsuperscript{2}C and by the geometry-aware critique points the other way. The former reports severe degradation when adapting SRe\textsuperscript{2}L to diffusion training, while the latter argues that ranking-based or prototype-based selectors fail because diffusion learning is sensitive to distortions of data support, relative relationships, and local structure rather than only to sample importance or class separation [2507.05914][2606.05883].

Current formulations also carry substantial assumptions. The geometry-aware method depends on a pretrained feature space whose geometry is assumed to correlate with the diffusion-relevant manifold, uses class-wise balanced selection, and includes a confidence regularizer that may inherit classifier biases; it is evaluated only on ImageNet-scale class-conditional generation. D\textsuperscript{2}C depends on a pretrained diffusion scorer, a pretrained text encoder, a pretrained vision encoder, and a pretrained VAE, and its appendix states that final performance upon convergence does not yet fully match baseline benchmarks in all cases. The video method relies on access to and fine-tuning of Latte, reports no formal generative quality metrics such as FVD, and leaves some details of VST-UNet under-specified. The coreset-expansion approach does not introduce a new diffusion-specific condensation loss; diffusion serves as a pretrained procedural generator, and runtime still grows with the number of patches due to repeated diffusion calls [2606.05883][2507.05914][2505.06670][2412.04668].

Several broader implications follow from these constraints. One is that D2C is presently most mature for labeled, class-conditional settings with strong pretrained backbones and encoders. Another is that “what must be preserved” remains an open design question. D\textsuperscript{2}C emphasizes learnability balanced by interval-sampled difficulty, whereas geometry-aware selection emphasizes distributional support geometry, neighborhood structure, and mode coverage. A plausible implication is that future work will need to reconcile these views rather than treat them as mutually exclusive. The same literature also points toward expansion beyond ImageNet-scale image generation: D\textsuperscript{2}C explicitly mentions possible extension to 3D and video diffusion, while the video work already demonstrates that diffusion-based condensation can be reformulated around synthetic-pool selection in temporal domains [2507.05914][2505.06670].

In its current form, Diffusion Dataset Condensation is best understood not as a single algorithm but as a research program that asks how much of a diffusion training set can be discarded, compressed, enriched, or procedurally regenerated without breaking the generative objective. The most technically developed answers so far revolve around three recurrent principles: preserve diffusion-relevant geometry, preserve enough semantic and visual conditioning to compensate for subsampling, and exploit generative priors only when they remain anchored to a faithful representation of the original data distribution.

Source: https://www.emergentmind.com/topics/diffusion-dataset-condensation-d2c