arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2004.02401cs.LGcs.CLstat.ML
收藏

循环学习率在神经机器翻译中的应用

Applying Cyclical Learning Rate to Neural Machine Translation

Artificial Intelligence Application Research Center, Huawei Technologies(华为技术有限公司人工智能应用研究中心)

Choon Meng Lee, Jianfeng Liu, Wei Peng

摘要:

在训练深度学习网络时,优化器和相关的学习率通常在没有经过太多考虑或进行最小调整的情况下使用,尽管它对于确保快速收敛到损失函数的良好质量最小值至关重要,该最小值也可以在测试数据集上很好地推广。从计算机视觉相关卷积网络和数据集的循环学习率策略的成功应用中汲取灵感,我们探索了如何将循环学习率应用于神经机器翻译的训练基于transformer的神经网络。从我们精心设计的实验,我们表明,优化器的选择和相关的周期性学习率政策可以有显着的影响性能。此外,我们建立了将循环学习率应用于神经机器翻译任务的指导方针。因此,通过我们的工作,我们希望提高人们对选择正确的优化器和相应的学习率政策的重要性的认识,同时鼓励进一步研究易于使用的学习率政策。

英文摘要:

In training deep learning networks, the optimizer and related learning rate are often used without much thought or with minimal tuning, even though it is crucial in ensuring a fast convergence to a good quality minimum of the loss function that can also generalize well on the test dataset. Drawing inspiration from the successful application of cyclical learning rate policy for computer vision related convolutional networks and datasets, we explore how cyclical learning rate can be applied to train transformer-based neural networks for neural machine translation. From our carefully designed experiments, we show that the choice of optimizers and the associated cyclical learning rate policy can have a significant impact on the performance. In addition, we establish guidelines when applying cyclical learning rates to neural machine translation tasks. Thus with our work, we hope to raise awareness of the importance of selecting the right optimizers and the accompanying learning rate policy, at the same time, encourage further research into easy-to-use learning rate policies.

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

↑