about
SRIFTY: Swift and Thrifty Distributed Training on the Cloud (rev.3 2022) (arxiv.org)
1 point by eternalban on Feb 12, 2023 | hide | past | pdf | discuss on HN

In plain words: Srifty times a training job on the cloud, then uses a model built from those timings to pick the cheapest, fastest mix of machines that fits the user's limits. It predicted step time within 8% and beat other pickers on speed and cost.

Abstract · Srifty: Swift and Thrifty Distributed Training on the Cloud

Finding the best VM configuration is key to achieve lower cost and higher throughput, two primary concerns in cloud-based distributed neural network (NN) training today. Optimal VM selection that meets user constraints requires efficiently navigating a large search space while controlling for the performance variance associated with sharing cloud instances and networks. In this work, we characterize this variance in the context of distributed NN training and present results of a comprehensive throughput and cost-efficiency study we conducted across a wide array of instances to prune for the optimal VM search space. Using insights from these studies, we built Srifty, a system that combines runtime profiling with learned performance models to accurately predict training performance and find the best VM choice that satisfies user constraints, potentially leveraging both heterogeneous setups and spot instances. We integrated Srifty with PyTorch and evaluated it on Amazon EC2. We conducted a large-scale generalization study of Srifty across more than 2K training setups on EC2. Our results show that Srifty achieves an iteration latency prediction error of 8%, and its VM instance recommendations offer significant throughput gain and cost reduction while satisfying user constraints compared to existing solutions in complex, real-world scenarios.

Liang Luo, Peter West, Arvind Krishnamurthy, Luis Ceze
arXiv:2011.14243 · cs.DC · submitted Nov 29, 2020 · updated Jul 2, 2022
abstract · pdf · html

add comment on HN