用小样本预训练代替全量数据,显著降低计算成本。
Efficient Neural Network Training via Subset Pretraining
- 用子集最小值近似全集最小值,加速训练
- 在MNIST/CIFAR上达到与全量训练相当效果
- 适合资源有限的模型快速迭代场景
神经网络训练中常使用小批量计算梯度,但其理论基础在常见学习率调度下不成立,且难以结合高效二阶优化。本文提出新假设:训练集的损失最小值可用其子集的最小值良好逼近。通过在MNIST、CIFAR-10和CIFAR-100图像分类任务上验证,即使小样本也能代表整体,且计算耗时可降至十分之一以下,所得结果等效于传统训练。关键条件是模型参数量与样本数的过定比超过1。
原文摘要 · Abstract (English)
In training neural networks, it is common practice to use partial gradients computed over batches, mostly very small subsets of the training set. This approach is motivated by the argument that such a partial gradient is close to the true one, with precision growing only with the square root of the batch size. A theoretical justification is with the help of stochastic approximation theory. However, the conditions for the validity of this theory are not satisfied in the usual learning rate schedules. Batch processing is also difficult to combine with efficient second-order optimization methods. This proposal is based on another hypothesis: the loss minimum of the training set can be expected to be well-approximated by the minima of its subsets. Such subset minima can be computed in a fraction of the time necessary for optimizing over the whole training set. This hypothesis has been tested with the help of the MNIST, CIFAR-10, and CIFAR-100 image classification benchmarks, optionally extended by training data augmentation. The experiments have confirmed that results equivalent to conventional training can be reached. In summary, even small subsets are representative if the overdetermination ratio for the given model parameter set sufficiently exceeds unity. The computing expense can be reduced to a tenth or less.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。