用双批次+循环渐进训练,提升大模型分布式训练的效率和精度。
Hybrid Dual-Batch and Cyclic Progressive Learning for Efficient Distributed Training
- 同时使用大、小两种批量大小,兼顾速度与泛化能力。
- 在CIFAR-100上准确率提升3.3%,训练时间减少10.1%。
- 适合需要高效训练大规模模型的研究者和工程师。
分布式机器学习对在大规模数据集上训练深层模型至关重要。现有研究多依赖更多硬件资源和强大计算单元来加速训练,常采用大批次以提高速度,但可能导致泛化能力下降。为此,我们提出双批次学习方案,在参数服务器框架下,利用硬件支持的最大批次提升效率,同时引入小批次增强模型泛化。通过并行使用两种批次,该方法在几乎不增加训练时间的前提下提升准确率。此外,为缓解双批次带来的时延开销,我们提出循环渐进学习策略:训练中逐步从低分辨率图像向高分辨率递增,有效缩短训练时间。将两者结合后,我们的混合方法显著提升了模型泛化能力和训练效率。在ResNet-18上的实验表明,相较于传统方法,本方案在CIFAR-100上准确率提升3.3%,训练时间减少10.1%;在ImageNet上训练时间更减少34.8%。
原文摘要 · Abstract (English)
Distributed machine learning is critical for training deep learning models on large datasets with numerous parameters. Current research primarily focuses on leveraging additional hardware resources and powerful computing units to accelerate the training process. As a result, larger batch sizes are often employed to speed up training. However, training with large batch sizes can lead to lower accuracy due to poor generalization. To address this issue, we propose the dual-batch learning scheme, a distributed training method built on the parameter server framework. This approach maximizes training efficiency by utilizing the largest batch size that the hardware can support while incorporating a smaller batch size to enhance model generalization. By using two different batch sizes simultaneously, this method improves accuracy with minimal additional training time. Additionally, to mitigate the time overhead caused by dual-batch learning, we propose the cyclic progressive learning scheme. This technique repeatedly and gradually increases image resolution from low to high during training, thereby reducing training time. By combining cyclic progressive learning with dual-batch learning, our hybrid approach improves both model generalization and training efficiency. Experimental results with ResNet-18 demonstrate that, compared to conventional training methods, our approach improves accuracy by 3.3% while reducing training time by 10.1% on CIFAR-100, and further achieves a 34.8% reduction in training time on ImageNet.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。