PipeTune: Pipeline Parallelism of Hyper and System Parameters Tuning for Deep Learning Clusters

TL;DR

PipeTune is a framework that optimizes deep learning training by parallelly tuning hyperparameters and system parameters, significantly reducing training time and energy consumption.

Contribution

It introduces a pipelined parallel tuning approach that considers both hyperparameters and system parameters, improving efficiency over existing methods.

Findings

Up to 22.6% reduction in tuning time

1.7x speed-up in training time

Up to 29% energy savings

Abstract

DNN learning jobs are common in today's clusters due to the advances in AI driven services such as machine translation and image recognition. The most critical phase of these jobs for model performance and learning cost is the tuning of hyperparameters. Existing approaches make use of techniques such as early stopping criteria to reduce the tuning impact on learning cost. However, these strategies do not consider the impact that certain hyperparameters and systems parameters have on training time. This paper presents PipeTune, a framework for DNN learning jobs that addresses the trade-offs between these two types of parameters. PipeTune takes advantage of the high parallelism and recurring characteristics of such jobs to minimize the learning cost via a pipelined simultaneous tuning of both hyper and system parameters. Our experimental evaluation using three different types of workloads indicates that PipeTune achieves up to 22.6% reduction and 1.7x speed up on tuning and training time, respectively. PipeTune not only improves performance but also lowers energy consumption up to 29%.