mlsanityData to 7 Oct 2026

A Tabular Schedule Abstraction for Communication-Aware Evaluation of Pipeline-Parallel LLM Training

arXiv 2605.24006 ↗First posted 19 May 2026cs.DC cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Pipeline parallelism is a key technique for distributed training of large language models because it reduces per-device parameter and activation memory. However, comparing pipeline schedules is difficult: analytical models expose structural quantities such as bubble ratios, while end-to-end hardware experiments are costly and system-specific. In this work, we introduce a tabular schedule abstraction and a unified multi-abstraction methodology that connects formula-based reasoning, idealized schedule tables, and communication-aware execution simulation. Using this framework, we compare GPipe, 1F1B, Chimera, and Hanayo in its restricted regime across multiple modeled system configurations. Our results show that schedule rankings are not abstraction-invariant: communication can negate structural advantages suggested by bubble analysis alone. Under the assumptions considered here, GPipe and 1F1B are runtime-equivalent, but 1F1B achieves a lower activation-memory peak. Chimera is advantageous mainly at low microbatch counts and in communication-favorable regimes, while Hanayo is effective in its intended restricted operating point but remains sensitive to network bottlenecks. We further study an asymmetric Chimera-style placement, which does not reduce the global peak memory requirement but reveals limited runtime gains in shallow pipelines. Overall, pipeline schedule quality is meaningful only in the context of the modeled execution environment.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 15 of its 22 references that are in the map. First posted on arXiv 19 May 2026.

Builds on

15 of 22 references.

References · 15

  1. 2020Language Models are Few-Shot Learnerscited by 32,199
  2. 2020Scaling Laws for Neural Language Modelscited by 6,411
  3. 2016Gaussian Error Linear Units (GELUs)cited by 3,431
  4. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,829
  5. 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,278
  6. 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 621
  7. 2018Beyond Data and Model Parallelism for Deep Neural Networkscited by 177
  8. 2020Memory-Efficient Pipeline-Parallel DNN Trainingcited by 133
  9. 2021Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelinescited by 68
  10. 2023ASTRA-sim2.0: Modeling Hierarchical Networks and Disaggregated Systems for Large-model Training at Scalecited by 47
  11. 2023Hanayo: Harnessing Wave-like Pipeline Parallelism for Enhanced Large Model Training Efficiencycited by 20
  12. 2020A Simple Model for Portable and Fast Prediction of Execution Time and Power Consumption of GPU Kernelscited by 7
3 more
  1. 2023Compressing the Backward Pass of Large-Scale Neural Architectures by Structured Activation Pruningcited by 3
  2. 2024Less Memory Means smaller GPUs: Backpropagation with Compressed Activationscited by 1
  3. 2025Scalable and Efficient Intra- and Inter-node Interconnection Networks for Post-Exascale Supercomputers and Data centerscited by 1

Built on by

Not cited yet.