arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.08086cs.LGcs.AI

数据集蒸馏遇上可证明子集选择

Dataset Distillation Meets Provable Subset Selection

DataHeroes;MIT(麻省理工学院);The University of Haifa(海法大学)

Murad Tukan, Alaa Maalouf, Margarita Osadchy

AI总结:

本文针对数据集蒸馏的初始化与训练采样问题,提出可证明的重要实例子集选择方法,并将其融入现有蒸馏流程以提升性能。

摘要:

近年来,深度学习取得了巨大发展,在各个领域产生了最先进的结果。然而,训练此类模型需要海量数据,从而增加了计算时间和成本。为解决这一问题,研究者提出数据集蒸馏,将大型训练数据集压缩为一个更小的合成数据集,同时保留其性能——这通常通过以下步骤完成:(1)均匀初始化一个合成集;(2)通过从完整数据中均匀采样实例,依据预定义损失迭代更新/学习该集合。本文改进了数据集蒸馏的两个阶段:(1)我们提出一种可证明的、基于采样的方法,通过识别数据中的重要点并移除冗余点来初始化蒸馏集;(2)我们进一步将数据子集选择思想与数据集蒸馏相结合,在训练过程中于“重要的”采样点上训练蒸馏集,而不是随机采样下一个批次。为此,我们基于实例相对于两种不同损失函数的相对贡献定义重要性概念:一种用于初始化阶段(针对核岭回归的核拟合函数,以及针对任何其他蒸馏方法的基于$K$-means的损失函数),另一种是用于训练阶段的相对交叉熵损失(或任何其他预定义损失)函数。最后,我们给出实验结果,展示我们的方法如何接入现有数据集蒸馏技术并提升其性能。

英文摘要:

Deep learning has grown tremendously over recent years, yielding state-of-the-art results in various fields. However, training such models requires huge amounts of data, increasing the computational time and cost. To address this, dataset distillation was proposed to compress a large training dataset into a smaller synthetic one that retains its performance -- this is usually done by (1) uniformly initializing a synthetic set and (2) iteratively updating/learning this set according to a predefined loss by uniformly sampling instances from the full data. In this paper, we improve both phases of dataset distillation: (1) we present a provable, sampling-based approach for initializing the distilled set by identifying important and removing redundant points in the data, and (2) we further merge the idea of data subset selection with dataset distillation, by training the distilled set on ``important'' sampled points during the training procedure instead of randomly sampling the next batch. To do so, we define the notion of importance based on the relative contribution of instances with respect to two different loss functions, i.e., one for the initialization phase (a kernel fitting function for kernel ridge regression and $K$-means based loss function for any other distillation method), and the relative cross-entropy loss (or any other predefined loss) function for the training phase. Finally, we provide experimental results showing how our method can latch on to existing dataset distillation techniques and improve their performance.

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

↑