about
Beyond Data and Model Parallelism for Deep Neural Networks (Stanford) (arxiv.org)
3 points by Katydid on Jul 31, 2018 | hide | past | pdf | discuss on HN

In plain words: Instead of splitting training by data or by model layers, this system searches four ways to divide the work, using a quick simulator to pick the fastest split. It trained networks up to 3.8 times faster than the best setups, even counting search time.

Abstract · Beyond Data and Model Parallelism for Deep Neural Networks

The computational requirements for training deep neural networks (DNNs) have grown to the point that it is now standard practice to parallelize training. Existing deep learning systems commonly use data or model parallelism, but unfortunately, these strategies often result in suboptimal parallelization performance. In this paper, we define a more comprehensive search space of parallelization strategies for DNNs called SOAP, which includes strategies to parallelize a DNN in the Sample, Operation, Attribute, and Parameter dimensions. We also propose FlexFlow, a deep learning framework that uses guided randomized search of the SOAP space to find a fast parallelization strategy for a specific parallel machine. To accelerate this search, FlexFlow introduces a novel execution simulator that can accurately predict a parallelization strategy's performance and is three orders of magnitude faster than prior approaches that have to execute each strategy. We evaluate FlexFlow with six real-world DNN benchmarks on two GPU clusters and show that FlexFlow can increase training throughput by up to 3.8x over state-of-the-art approaches, even when including its search time, and also improves scalability.

Zhihao Jia, Matei Zaharia, Alex Aiken
arXiv:1807.05358 · cs.DC · submitted Jul 14, 2018
abstract · pdf · html

add comment on HN