DPLRS: Distributed Population Learning Rate Schedule

Jia Wei

doi:10.1016/j.future.2022.02.001

DPLRS: Distributed Population Learning Rate Schedule

2022-07-01

1

PRE

AI

DOI：10.1016/j.future.2022.02.001

Original

Original request for help

Save

Abstract

En 中文

Deep neural network models perform very brightly in the field of artificial intelligence, but their success is affected by hyperparameters, and the learning rate schedule is one of the most important hyperparameters, while the search for the learning rate schedule is often time-consuming and compu-tationally resource-intensive. In this paper, we proposed Distributed Population Learning Rate Schedule (DPLRS) based on population joint optimization, which uses distributed data parallel deep neural network training to implement a dynamic learning rate schedule optimization strategy based on the population idea, with almost no loss of test accuracy. DPLRS is able to dynamically refine the learning rate schedule during model training instead of following the usual suboptimal strategy. We conducted experiments on typical AlexNet, VGG16, and ResNet18 using the Tianhe-3 supercomputing prototype. The results illustrate that using DPLRS to dynamically update the learning rate can greatly reduce the searching time of the learning rate schedule and meanwhile, can ensure the close performance with the latest population hyperparameter algorithm. Also, In our experiments, DPLRS lead to 123.85x speedup maximum, which prove the effectiveness and robustness of DPLRS. (c) 2022 Elsevier B.V. All rights reserved.

Keywords:

Deep learning

Distributed training

Hyperparameter search

Data parallel

Population algorithm

DPLRS: Distributed Population Learning Rate Schedule

Abstract

Journal

Organization