SlimPack: Fine-Grained Asymmetric Packing for Balanced and Efficient Variable-Length LLM Training
Not on the map. Why?
Abstract (arXiv)
The efficient distributed training of Large Language Models (LLMs) is severely hampered by the extreme variance in context lengths. This data heterogeneity, amplified by conventional packing strategies and asymmetric forward-backward costs, leads to critical inefficiencies such as cascading workload imbalances and severe hardware underutilization. Existing solutions attempt to mitigate these challenges, but often at the expense of memory or communication efficiency. To address these challenges, we introduce SlimPack, a framework that fundamentally rethinks data packing and scheduling by decomposing samples into fine-grained slices. This slice-level decomposition immediately mitigates critical memory and communication bottlenecks by transforming large, volatile workloads into a stream of smaller, manageable units. This flexibility is then harnessed for our core innovation, Asymmetric Partitioning, which assembles balanced scheduling units uniquely optimized for the different demands of the forward and backward passes. Orchestrated by a two-phase solver and a high-fidelity simulator, SlimPack holistically resolves imbalances across all parallel dimensions. Extensive experiments demonstrate that SlimPack achieves up to a 2.8× training throughput improvement over baselines, breaking the conventional trade-off by delivering both superior balance and high resource efficiency.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 27 of its 41 references that are in the map. First posted on arXiv 30 Sep 2025.
Builds on
27 of 41 references.
References · 27
- 2017Attention Is All You Needcited by 60,361
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
- 2023GQA: Training Generalized Multi-Query Transformer Models from Multi-Head Checkpointscited by 1,346
- 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
- 2016Training Deep Nets with Sublinear Memory Costcited by 848
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 632
- 2023Ring Attention with Blockwise Transformers for Near-Infinite Contextcited by 457
- 2022Reducing Activation Recomputation in Large Transformer Modelscited by 302
- 2021ZeRO-Infinity: Breaking the GPU Memory Wall for Extreme Scale Deep Learningcited by 256
15 more
- 2020Memory-Efficient Pipeline-Parallel DNN Trainingcited by 134
- 2018PipeDream: Fast and Efficient Pipeline Parallel DNN Trainingcited by 117
- 2020DAPPLE: A Pipelined Data Parallel Approach for Training Large Modelscited by 101
- 2021Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Trainingcited by 96
- 2019Checkmate: Breaking the Memory Wall with Optimal Tensor Rematerializationcited by 91
- 2021Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelinescited by 68
- 2023Striped Attention: Faster Ring Attention for Causal Transformerscited by 54
- 2022Alpa: Automating Inter- and Intra-Operator Parallelism for Distributed Deep Learningcited by 20
- 2024FlexSP: Accelerating Large Language Model Training via Flexible Sequence Parallelismcited by 19
- 2025WLB-LLM: Workload-Balanced 4D Parallelism for Large Language Model Trainingcited by 17
- 2023DynaPipe: Optimizing Multi-task Training through Dynamic Pipelinescited by 12
- 2024Packing Analysis: Packing Is More Appropriate for Large Models or Datasets in Supervised Fine-tuningcited by 11
- 2025ByteScale: Efficient Scaling of LLM Training with a 2048K Context Length on More Than 12,000 GPUscited by 10
- 2025SlimPipe: Memory-Thrifty and Efficient Pipeline Parallelism for Long-Context LLM Trainingcited by 6
- 2024Re-evaluating the Memory-balanced Pipeline Parallelism: BPipecited by 3
Built on by
Not cited yet.