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

SwiftLearn:一种使用重要性采样的深度学习模型数据高效训练方法

SwiftLearn: A Data-Efficient Training Method of Deep Learning Models using Importance Sampling

Huawei Technologies(华为技术有限公司)

Habib Hajimolahoseini, Omar Mohamed Awad, Walid Ahmed, Austin Wen, Saina Asani, Mohammad Hassanpour, Farnoosh Javadi, Mehdi Ahmadi, Foozhan Ataiefard, Kangling Liu, Yang Liu

AI总结:

本文提出SwiftLearn,一种基于重要性采样的数据高效训练方法,通过在warm-up阶段选取重要样本子集加速深度学习模型训练,在保持性能的同时实现显著速度提升。

摘要:

本文提出了SwiftLearn,一种数据高效的方法,通过在训练的warm-up阶段选取数据样本子集来加速深度学习模型的训练。该子集基于warm-up阶段在整个数据集上测得的重要性标准进行选择,旨在在剩余训练中使用更少的样本保持模型性能。我们提出的重要性度量可以在训练期间偶尔更新,以确保所有数据样本在表现出更高重要性时有机会返回训练循环。模型架构保持不变,但由于数据样本数量控制着训练期间的前向和反向传播次数,我们可以通过减少每个epoch中使用的训练样本数量来缩短训练时间。在多种CV和NLP模型的预训练和微调阶段的实验结果表明,该方法能在保持模型性能的同时实现显著的训练加速。具体而言,在GLUE基准上进行BERT微调的实验显示,可以丢弃近90%的数据,实现3.36倍的端到端平均加速,同时平均准确率下降保持在0.92%以下。

英文摘要:

In this paper, we present SwiftLearn, a data-efficient approach to accelerate training of deep learning models using a subset of data samples selected during the warm-up stages of training. This subset is selected based on an importance criteria measured over the entire dataset during warm-up stages, aiming to preserve the model performance with fewer examples during the rest of training. The importance measure we propose could be updated during training every once in a while, to make sure that all of the data samples have a chance to return to the training loop if they show a higher importance. The model architecture is unchanged but since the number of data samples controls the number of forward and backward passes during training, we can reduce the training time by reducing the number of training samples used in each epoch of training. Experimental results on a variety of CV and NLP models during both pretraining and finetuning show that the model performance could be preserved while achieving a significant speed-up during training. More specifically, BERT finetuning on GLUE benchmark shows that almost 90% of the data can be dropped achieving an end-to-end average speedup of 3.36x while keeping the average accuracy drop less than 0.92%.

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

↑