学习率与batch_size对模型性能的影响

目录

学习率

Batch_size(批量大小)

学习率与batch_size在权重更新中的关系

学习率与batch_size对模型性能的影响(转载:原文)


学习率

学习率(lr)表示每次更新权重参数的尺度(步长),学习率与batch_size对模型性能的影响

Batch_size(批量大小)

batch_size有一个计算公式,即 学习率与batch_size对模型性能的影响,  或者 学习率与batch_size对模型性能的影响, N为训练数据集大小, batch为需要多少次迭代才能在一个epoch中训练完整个数据集。batch_size参数决定了完整训练数据集1个epoch需要多少个batch。

学习率与batch_size在权重更新中的关系

  • 学习率(lr)直观可以看出lr越大,权重更新的跨度越大,模型参数调整变化越快。

  • batch_size对模型的影响,在于模型每次更新时,计算梯度是计算整个Batch的平均梯度,即权重更新公式中的学习率与batch_size对模型性能的影响, 整合就是学习率与batch_size对模型性能的影响。即lr与batch_size共同影响模型更新。 

学习率与batch_size对模型性能的影响(转载:原文

  • 大的batchsize减少训练时间,提高稳定性。 这是肯定的,同样的epoch数目,大的batchsize需要的batch数目减少了,所以可以减少训练时间。另一方面,大的batch size梯度的计算更加稳定,因为模型训练曲线会更加平滑。在微调的时候,大的batch size可能会取得更好的结果。
  • 大的batchsize泛化能力下降。 在一定范围内,增加batchsize有助于收敛的稳定性,但是随着batchsize的增加,模型的性能会下降,如下图,来自于文[1]。学习率与batch_size对模型性能的影响

这是研究者们普遍观测到的规律,虽然可以通过一些技术缓解。这个导致性能下降的batch size在上图就是8000左右。

研究[2]表明大的batchsize收敛到sharp miminum,而小的batchsize收敛到flat mimimum,后者具有更好的泛化能力。两者的区别就在于变化的趋势,一个快一个慢,如下图,造成这个现象的主要原因是大小的batchsize带来的噪声有助于逃离sharp mininum。

学习率与batch_size对模型性能的影响

Hoffer[3]等人的研究表明,大的batchsize性能下降是因为训练时间不够长,本质上并不是batchsize的问题,在同样的epochs下的参数更新变少了,因此需要更长的迭代次数。

 

[1] Goyal P, Dollar P, Girshick R B, et al. Accurate, Large Minibatch SGD: Training ImageNet in 1 Hour.[J]. arXiv: Computer Vision and Pattern Recognition, 2017.

[2] Keskar N S, Mudigere D, Nocedal J, et al. On large-batch training for deep learning: Generalization gap and sharp minima[J]. arXiv preprint arXiv:1609.04836, 2016.

[3] Hoffer E, Hubara I, Soudry D. Train longer, generalize better: closing the generalization gap in large batch training of neural networks[C]//Advances in Neural Information Processing Systems. 2017: 1731-1741.