Learnability-Guided Diffusion for Dataset Distillation

TL;DR

This paper introduces a learnability-guided diffusion method for dataset distillation that incrementally constructs synthetic datasets, reducing redundancy and improving performance on image classification benchmarks.

Contribution

It proposes a novel curriculum-based approach using learnability scores and diffusion models to generate more effective, less redundant synthetic datasets for training machine learning models.

Findings

Reduces dataset redundancy by 39.1%.

Achieves state-of-the-art results on ImageNet-1K with 60.1%.

Promotes specialization across training stages.

Abstract

Training machine learning models on massive datasets is expensive and time-consuming. Dataset distillation addresses this by creating a small synthetic dataset that achieves the same performance as the full dataset. Recent methods use diffusion models to generate distilled data, either by promoting diversity or matching training gradients. However, existing approaches produce redundant training signals, where samples convey overlapping information. Empirically, disjoint subsets of distilled datasets capture 80-90% overlapping signals. This redundancy stems from optimizing visual diversity or average training dynamics without accounting for similarity across samples, leading to datasets where multiple samples share similar information rather than complementary knowledge. We propose learnability-driven dataset distillation, which constructs synthetic datasets incrementally through successive stages. Starting from a small set, we train a model and generate new samples guided by learnability scores that identify what the current model can learn from, creating an adaptive curriculum. We introduce Learnability-Guided Diffusion (LGD), which balances training utility for the current model with validity under a reference model to generate curriculum-aligned samples. Our approach reduces redundancy by 39.1%, promotes specialization across training stages, and achieves state-of-the-art results on ImageNet-1K (60.1%), ImageNette (87.2%), and ImageWoof (72.9%). Our code is available on our project page https://jachansantiago.github.io/learnability-guided-distillation/.