Method · Diffusion-based synthesis
Dataset Condensation with Color Compensation
Huyu Wu, Duo Su, Junjie Hou, Guang Li
TMLR 2025 · first public 2025-08-02 · arXiv 2508.01139
In one paragraph
After a calibrated real-image selection, fine-tunes a pretrained latent diffusion model on the condensed set to enhance the color diversity of the selected images rather than generating new ones from scratch, arguing color is both an information carrier and a semantic unit that prior condensation methods overlook; FID/IS results show the enhanced images avoid model collapse, and DC3 outperforms state-of-the-art methods across benchmarks.
Where it sits
- Diffusion-based synthesis (Generative priors)
- Setting: Image classification
Design choices
| Prior / networks used | diffusion |
| Optimization regime | generator-fine-tuning |
Abstract (verbatim from arXiv)
Dataset condensation always faces a constitutive trade-off: balancing performance and fidelity under extreme compression. Existing methods struggle with two bottlenecks: image-level selection methods (Coreset Selection, Dataset Quantization) suffer from inefficiency condensation, while pixel-level optimization (Dataset Distillation) introduces semantic distortion due to over-parameterization. With empirical observations, we find that a critical problem in dataset condensation is the oversight of color's dual role as an information carrier and a basic semantic representation unit. We argue that improving the colorfulness of condensed images is beneficial for representation learning. Motivated by this, we propose DC3: a Dataset Condensation framework with Color Compensation. After a calibrated selection strategy, DC3 utilizes the latent diffusion model to enhance the color diversity of an image rather than creating a brand-new one. Extensive experiments demonstrate the superior performance and generalization of DC3 that outperforms SOTA methods across multiple benchmarks. To the best of our knowledge, besides focusing on downstream tasks, DC3 is the first research to fine-tune pre-trained diffusion models with condensed datasets. The Frechet Inception Distance (FID) and Inception Score (IS) results prove that training networks with our high-quality datasets is feasible without model collapse or other degradation issues. Code and generated data are available at https://github.com/528why/Dataset-Condensation-with-Color-Compensation.
BibTeX (generated; prefer the venue's official entry)
@article{wu2025dataset,
title = {Dataset Condensation with Color Compensation},
author = {Huyu Wu and Duo Su and Junjie Hou and Guang Li},
journal = {TMLR 2025},
year = {2025}
}Nearby in Diffusion-based synthesis
Learnability-guided diffusion — Learnability-Guided Diffusion for Dataset Distillation
Jeffrey A. Chan-Santiago, Mubarak Shah · CVPR 2026notablepaper ↗
ManifoldGD — ManifoldGD: Training-Free Hierarchical Manifold Guidance for Diffusion-Based Dataset Distillation
Ayush Roy, Wei-Yang Alex Lee, Rudrasis Chakraborty et al. · CVPR 2026notablepaper ↗code ↗