Accelerating Large-Scale Dataset Distillation via Exploration-Exploitation Optimization

TL;DR

This paper introduces E$^2$D, an exploration-exploitation optimization method for large-scale dataset distillation that significantly improves efficiency and accuracy, enabling faster training with smaller datasets.

Contribution

The paper proposes a novel two-phase optimization strategy for dataset distillation that balances accuracy and efficiency, outperforming existing methods on large-scale benchmarks.

Findings

E$^2$D achieves state-of-the-art accuracy on ImageNet-1K.

E$^2$D is 18 times faster than previous methods on ImageNet-1K.

E$^2$D improves accuracy and speed on ImageNet-21K.

Abstract

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.