Jargonautica
How the Machinery Works

Gradient Checkpointing

Also called: Activation Checkpointing

Throwing away intermediate results during the forward pass and recomputing them later when they are needed, in order to fit a larger model into memory. You pay perhaps a third more compute time to save a great deal of memory, which is usually the binding constraint.

Technical view

Also called activation checkpointing. Stores activations only at selected layer boundaries and recomputes the rest during the backward pass, reducing activation memory from linear in depth to roughly the square root of it with optimal placement. Combines with sharding and offloading as one of the standard levers for fitting large models onto available accelerators, and it is orthogonal to model parallelism.


Part of AI Terminology Index, section: How the Machinery Works.

How it connects

Gradient Checkpointing links to 4 other terms. The faint lines are links between those terms. See it in the full graph.

ComputeComputeBackpropagationBackpropagationGradient AccumulationGradient Accumulati…Sharding and ZeROSharding and ZeROGradient Checkpointing