提升数据蒸馏多样性,用分阶段训练让合成图像更丰富且更快。
DELT: A Simple Diversity-driven EarlyLate Training for Dataset Distillation
- 将每类样本分成小任务,局部优化生成不同阶段的分布图像
- 在多个数据集上平均性能领先2~5%,同类图像多样性提升超5%
- 适合需要高效、高质量合成数据的科研与工业场景
数据蒸馏近年发展出两大方向:传统批次间匹配适用于小规模数据集,采用模型与合成样本的双层优化方法,如FRePo、RCIG、RaT-BPTT,以及分布匹配、梯度匹配和权重轨迹匹配等;而批次到全局匹配则属于解耦方法,更适合大规模数据集,代表性工作包括SRe$^2$L、G-VBSM、WMDD和CDA。但该方法面临各类别合成样本多样性不足的问题,因样本独立优化且共享同一全局监督信号。本文提出一种简单有效的多样性驱动早期-晚期训练(DELT)方案,在批次到全局匹配框架下提升合成图像多样性并降低计算开销。通过将预设的IPC样本划分为更小子任务,利用局部优化将每个子集蒸馏为不同阶段的分布,从而减少统一优化带来的同质化。这些子任务生成的图像在整体任务中展现出良好泛化能力。在CIFAR、Tiny-ImageNet、ImageNet-1K及其子集上进行大量实验,结果表明,DELT在不同数据集和IPC设置下平均性能优于此前最先进方法2~5%,同类图像多样性提升超过5%,合成时间最多减少39.3%,显著提升训练效率。代码已公开于:https://github.com/VILA-Lab/DELT。
原文摘要 · Abstract (English)
Recent advances in dataset distillation have led to solutions in two main directions. The conventional batch-to-batch matching mechanism is ideal for small-scale datasets and includes bi-level optimization methods on models and syntheses, such as FRePo, RCIG, and RaT-BPTT, as well as other methods like distribution matching, gradient matching, and weight trajectory matching. Conversely, batch-to-global matching typifies decoupled methods, which are particularly advantageous for large-scale datasets. This approach has garnered substantial interest within the community, as seen in SRe$^2$L, G-VBSM, WMDD, and CDA. A primary challenge with the second approach is the lack of diversity among syntheses within each class since samples are optimized independently and the same global supervision signals are reused across different synthetic images. In this study, we propose a new Diversity-driven EarlyLate Training (DELT) scheme to enhance the diversity of images in batch-to-global matching with less computation. Our approach is conceptually simple yet effective, it partitions predefined IPC samples into smaller subtasks and employs local optimizations to distill each subset into distributions from distinct phases, reducing the uniformity induced by the unified optimization process. These distilled images from the subtasks demonstrate effective generalization when applied to the entire task. We conduct extensive experiments on CIFAR, Tiny-ImageNet, ImageNet-1K, and its sub-datasets. Our approach outperforms the previous state-of-the-art by 2$\sim$5% on average across different datasets and IPCs (images per class), increasing diversity per class by more than 5% while reducing synthesis time by up to 39.3% for enhancing the training efficiency. Code is available at: https://github.com/VILA-Lab/DELT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。