arXivDaily arXiv每日学术速递 周一至周五更新
看起来今天可以划水了
arXiv 1906.04887cs.LGcs.CVstat.ML
收藏

使用小型代理数据集加速超参数搜索

Using Small Proxy Datasets to Accelerate Hyperparameter Search

Sam Shleifer, Eric Prokop

摘要:

机器学习工作流程中最大的瓶颈之一是等待模型进行训练。根据可用的计算资源,可能需要几天到几周的时间在一个包含许多类的大型数据集上训练神经网络,例如ImageNet。对于尝试新算法方法的研究人员来说,这是不切实际的耗时和昂贵的。我们的目标是生成更小的“代理数据集”,其中实验运行成本更低,但结果与完整数据集上的实验结果高度相关。我们通过从示例或类中随机采样,仅在最简单或最困难的示例上进行训练,并在“数据蒸馏”生成的合成示例上进行训练来生成这些代理数据集。我们将这些技术与更广泛使用的完整数据集上的训练基线进行比较,以减少历元。对于每一种重复策略,我们估计三个衡量“代理质量”的指标:在整个数据集上的实验结果的方差有多少可以通过代理数据集上的实验结果来解释。Imagenette和Imagewoof(Howard,2019)上的实验表明,在最简单的10%的示例上运行超参数搜索可以解释目标任务实验结果中81%的方差,使用最简单的50%的示例可以解释95%的方差,远远超过在更少时期的所有数据上进行训练,这是一个更广泛使用的基线。这些“简单”的代理比在完整数据集上训练更高的质量,减少了迭代次数(但计算成本相同),并且出乎意料地,比从最难的示例构建的代理质量更高。在不访问训练模型的情况下,研究人员可以通过将子集限制在更少的类中来提高代理质量;基于一半类构建的代理比那些具有相同数量的样本分布在所有类中的代理质量更高。

英文摘要:

One of the biggest bottlenecks in a machine learning workflow is waiting for models to train. Depending on the available computing resources, it can take days to weeks to train a neural network on a large dataset with many classes such as ImageNet. For researchers experimenting with new algorithmic approaches, this is impractically time consuming and costly. We aim to generate smaller "proxy datasets" where experiments are cheaper to run but results are highly correlated with experimental results on the full dataset. We generate these proxy datasets using by randomly sampling from examples or classes, training on only the easiest or hardest examples and training on synthetic examples generated by "data distillation". We compare these techniques to the more widely used baseline of training on the full dataset for fewer epochs. For each proxying strategy, we estimate three measures of "proxy quality": how much of the variance in experimental results on the full dataset can be explained by experimental results on the proxy dataset. Experiments on Imagenette and Imagewoof (Howard, 2019) show that running hyperparameter search on the easiest 10% of examples explains 81% of the variance in experiment results on the target task, and using the easiest 50% of examples can explain 95% of the variance, significantly more than training on all the data for fewer epochs, a more widely used baseline. These "easy" proxies are higher quality than training on the full dataset for a reduced number of epochs (but equivalent computational cost), and, unexpectedly, higher quality than proxies constructed from the hardest examples. Without access to a trained model, researchers can improve proxy quality by restricting the subset to fewer classes; proxies built on half the classes are higher quality than those with an equivalent number of examples spread across all classes.

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

↑