Breadth-First Pipeline Parallelism

• Main Author: Lamy-Poirier, Joel
• Format: Journal Article
• Language: English
• Published: 10.11.2022
• Subjects: Computer Science - Artificial Intelligence; Computer Science - Computation and Language; Computer Science - Distributed, Parallel, and Cluster Computing; Computer Science - Learning

We introduce Breadth-First Pipeline Parallelism, a novel training schedule which optimizes the combination of pipeline and data parallelism. Breadth-First Pipeline Parallelism lowers training time, cost and memory usage by combining a high GPU utilization with a small batch size per GPU, and by maki...