FastPersist: Making Per-Iteration Deep Learning Checkpointing Nearly Free
Guanhua Wang, Olatunji Ruwase, Bing Xie, Yuxiong He (Microsoft DeepSpeed) · arXiv, June 2024
arXiv:2406.13768
Interactive page
References
- FastPersist: Accelerating Model Checkpointing in Deep Learning — Wang, Ruwase, Xie, He, 2024 (arXiv:2406.13768)
- CheckFreq: Frequent, Fine-Grained DNN Checkpointing — Mohan, Phanishayee, Chidambaram, 2021
- Gemini: Fast Failure Recovery in Distributed Training with In-Memory Checkpoints — Wang et al., 2023
- Check-N-Run — Eisenman et al., 2022
- ByteCheckpoint — Wan et al., 2024
- ZeRO-Infinity — Rajbhandari, Ruwase, Rasley, Smith, He, 2021
- Understanding Data Storage and Ingestion for Large-Scale Deep Recommendation Model Training — Zhao et al., 2022
- What Modern NVMe Storage Can Do, And How To Exploit It — Haas, Leis, 2023
- PyTorch Distributed Checkpoint (DCP) and async_save — PyTorch team, 2023–2024
- OPT-175B logbook and Llama 3 herd paper (failure analyses) — Zhang et al.; Llama Team, 2022; 2024
- Young/Daly optimal checkpoint interval; multi-level checkpointing (SCR, FTI) — Young; Daly; Moody et al.; Bautista-Gomez et al.
- GPUDirect Storage / cuFile and SPIN — NVIDIA; Bergman et al.
- ReCycle / Oobleck / Bamboo — resilient training without checkpoint restart, 2023–2024