arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.15277cs.CVcs.AIcs.LG

通过探索-利用优化加速大规模数据集蒸馏

Accelerating Large-Scale Dataset Distillation via Exploration-Exploitation Optimization

North Carolina State University(北卡罗来纳州立大学);King Abdulaziz University(阿卜杜勒阿齐兹国王大学);Oak Ridge National Laboratory(橡树岭国家实验室)

Muhammad J. Alahmadi, Peng Gao, Feiyi Wang, Dongkuan Xu

AI总结:

本文提出E²D方法,通过探索-利用优化策略在大规模数据集蒸馏中提升效率与准确率,实现18倍加速和显著性能改进。

摘要:

数据集蒸馏将原始数据压缩成紧凑的合成数据集,减少训练时间和存储空间,同时保持模型性能,从而在资源有限的情况下实现部署。尽管最近基于解耦的蒸馏方法能够在大规模上实现数据集蒸馏,但仍面临效率差距:基于优化的解耦方法虽然准确率更高,但计算需求大;而无优化解耦方法虽然高效,但牺牲了准确性。为克服这一权衡,我们提出了探索-利用蒸馏(E²D),一种简单实用的方法,通过高效的管道最小化冗余计算,该管道首先以全图初始化开始,以保留语义完整性与特征多样性。然后采用两阶段优化策略:探索阶段执行均匀更新并识别高损失区域,利用阶段则将更新集中在这些区域以加速收敛。我们在大规模基准上评估了E²D,其在ImageNet-1K上超越了当前最先进的方法,速度提升18倍;在ImageNet-21K上,我们的方法在保持速度提升4.3倍的同时显著提高了准确性。这些结果表明,针对冗余减少的更新,而非暴力优化,能够弥合大规模数据集蒸馏在准确率与效率之间的差距。代码可在https://github.com/ncsu-dk-lab/E2D上获取。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑