arXiv:2602.15277cs.CVcs.AI2026-02

通过探索-利用策略加速大规模数据蒸馏,兼顾效率与精度。

Accelerating Large-Scale Dataset Distillation via Exploration-Exploitation Optimization

  • 先全图初始化保语义,再分探索与利用两阶段优化。
  • ImageNet-1K上比最新方法快18倍,精度更高;ImageNet-21K上提速4.3倍且更准。
  • 适合资源受限场景下高效训练的开发者和研究者。

数据蒸馏将原始数据压缩为紧凑的合成数据集,降低训练时间和存储开销,同时保持模型性能,适用于资源受限环境。尽管近期解耦式蒸馏方法实现了大规模数据蒸馏,但仍存在效率差距:基于优化的方法精度高但计算量大,无优化方法虽快却牺牲精度。为此,我们提出探索-利用蒸馏(E²D),一种简单实用的方法,通过高效流程减少冗余计算。该方法从全图初始化开始,以保持语义完整性和特征多样性;随后采用两阶段优化策略:探索阶段进行均匀更新并识别高损失区域,利用阶段聚焦更新这些区域以加速收敛。我们在大规模基准上评估E²D,于ImageNet-1K上超越当前最优结果且速度快18倍,在ImageNet-21K上显著提升精度的同时仍快4.3倍。结果表明,有针对性地减少冗余更新,而非盲目优化,可有效弥合大规模数据蒸馏中精度与效率的矛盾。代码见https://github.com/ncsu-dk-lab/E2D。

原文摘要 · Abstract (English)

Dataset distillation compresses the original data into compact synthetic datasets, reducing training time and storage while retaining model performance, enabling deployment under limited resources. Although recent decoupling-based distillation methods enable dataset distillation at large scale, they continue to face an efficiency gap: optimization-based decoupling methods achieve higher accuracy but demand intensive computation, whereas optimization-free decoupling methods are efficient but sacrifice accuracy. To overcome this trade-off, we propose Exploration--Exploitation Distillation (E$^2$D), a simple, practical method that minimizes redundant computation through an efficient pipeline that begins with full-image initialization to preserve semantic integrity and feature diversity. It then uses a two-phase optimization strategy: an exploration phase that performs uniform updates and identifies high-loss regions, and an exploitation phase that focuses updates on these regions to accelerate convergence. We evaluate E$^2$D on large-scale benchmarks, surpassing the state-of-the-art on ImageNet-1K while being $18\times$ faster, and on ImageNet-21K, our method substantially improves accuracy while remaining $4.3\times$ faster. These results demonstrate that targeted, redundancy-reducing updates, rather than brute-force optimization, bridge the gap between accuracy and efficiency in large-scale dataset distillation. Code is available at https://github.com/ncsu-dk-lab/E2D.

数据蒸馏效率优化大规模训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。