用可学习性指导生成数据,让小样本集更高效。
Learnability-Guided Diffusion for Dataset Distillation
- 按模型学习能力逐步生成新数据,形成自适应训练课程
- 在ImageNet-1K上达60.1%准确率,冗余降低39.1%
- 适合需要高效数据压缩的模型训练场景
大规模数据训练机器学习模型成本高、耗时长。数据蒸馏通过生成小型合成数据集来解决此问题,使其性能接近原始数据集。现有方法使用扩散模型生成数据,或强调多样性,或匹配训练梯度,但常导致冗余——不同样本传递相似信息。实验表明,蒸馏数据集中80%-90%的信号存在重叠。这源于仅优化视觉多样性或平均训练动态,未考虑样本间相似性,导致多份数据共享相同知识。本文提出可学习性引导的数据蒸馏方法,通过分阶段增量构建合成数据集:从初始小集合开始,训练模型后,根据可学习性分数生成新样本,识别当前模型能从中学习的内容,形成自适应课程。我们提出学习性引导扩散(LGD),在当前模型训练效用与参考模型有效性之间取得平衡,生成符合课程的样本。该方法将冗余减少39.1%,促进各阶段数据专一化,在ImageNet-1K(60.1%)、ImageNette(87.2%)和ImageWoof(72.9%)上达到当前最优性能。代码已公开于https://jachansantiago.github.io/learnability-guided-distillation/。
原文摘要 · Abstract (English)
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/.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。