Check-N-Run: a Checkpointing System for Training Deep Learning Recommendation Models
Talkat 19th USENIX Symposium on Networked Systems Design and Implementation (NSDI '22) · 2022-04-04..04-06·
summary
System implements a decoupled, atomic checkpoint pipeline that copies GPU state to CPU and uploads with dedicated CPU workers, targets terabyte-scale recommendation models whose embeddings are >99% of size, and combines differential checkpointing (one‑shot, consecutive, intermittent) with row‑wise quantization (gradient‑tuned ranges). Evaluation on Facebook workloads shows write bandwidth drops 6–17× and storage 2.5–8×; 3‑bit quantization tolerates up to ~3 resumes under a 0.01 accuracy loss threshold before falling back to 8‑bit.
AI-written summary · about
chapters (15)
- 0:00Intro
- 0:30Recommendation Models are important . Use cases include
- 1:00Recommendation Model Architecture
- 1:48High Performance Training at Meta
- 2:26The Criticality of Checkpointing • Failure recovery ensure progress
- 3:41Checkpoint Challenges
- 4:52Check-n-Run
- 6:20Checkpointing Workflow
- 7:18Reducing WB with Differential Checkpointing
- 8:09Approaches for Differential Checkpointing • One-Shot Differential Checkpoint . Consecutive Incremental Checkpoint - Intermittent Differential Checkpoint
- 10:00Checkpoint Quantization Compress checkpoint without degrading training accuracy
- 11:31Comparing Quantization Strategies . Uniform quantization . Non-uniform quantization using kmeans • Adaptive uniform quantization
- 12:47Quantization Bit-width Selection
- 14:35Overall Reduction
- 15:06Summary
related talks
15mUniversal Checkpointing: A Flexible and Efficient Distributed Checkpointing...
17mCheckmate: Zero Performance Overhead Model Checkpointing via Network Gradient Replication
13mByteCheckpoint: A Unified Checkpointing System for Large Foundation Model Development
12mCheckFreq: Frequent, Fine-Grained DNN Checkpointing
15mDistributed Checkpoint
15mAdaCheck: An Adaptive Checkpointing System for Efficient LLM Training with Redundancy...
32mEfficient Transparent Checkpointing of AI/ML Workloads in Kub...
12mQuickUpdate: a Real-Time Personalization System for Large-Scale Recommendation Models
