AIMech AI

TernGrad (Stochastic Ternary Gradient Quantization for Distributed Deep Learning)

Algorithms AIMech:TernGradTernaryGradientsForDistributedDeepLearning DRAFT
Aliases: TernGrad, Stochastic Ternary Gradient Quantization, Ternary Gradients for Distributed Deep Learning
An unbiased stochastic ternary gradient quantization scheme for distributed synchronous SGD (Wen et al. 2017, Section 3, Equations 4-7) that compresses each worker's gradient vector g_t into ternary levels {-1, 0, +1} scaled by s_t = max_k |g_{tk}| = ||g_t||_infty via g_tilde_t = s_t * sign(g_t) o b_t where each coordinate Bernoulli mask b_{tk} in {0, 1} is sampled independently with probability P(b_{tk} = 1 | g_t) = |g_{tk}| / s_t, combined with layer-wise scalars and coordinate clipping |g_{tk}| <= c * sigma_t to bound the maximum magnitude s_t.
Identity Notes & Concept Scope Distinct from deterministic sign-based compression SignSGD (AIMech:SignSgdAndMajorityVoteCompressedNonconvexOptimization, Bernstein et al. 2018), which maps coordinates deterministically to {-1, +1} without zero level or unbiased probabilistic rounding, and from multi-level L2-norm quantized SGD QSGD (AIMech:QsgdQuantizedStochasticGradientDescentWithEliasCoding, Alistarh et al. 2017), which normalizes by the Euclidean norm ||g_t||_2 rather than the infinity norm s_t = ||g_t||_infty with layer-wise sigma-clipping.

Claims & Grounded Evidence

4 Claims
Mechanism #r417-unbiased-ternary-rounding-and-infinity-l1-variance-identity
Conditioned on g_t, TernGrad satisfies E[g_tilde_{tk} | g_t] = s_t sign(g_{tk}) (|g_{tk}| / s_t) = g_{tk} and coordinate second moment E[g_tilde_{tk}^2 | g_t] = s_t |g_{tk}| = ||g_t||_infty |g_{tk}|, yielding total quantization variance E[||g_tilde_t - g_t||_2^2 | g_t] = ||g_t||_infty ||g_t||_1 - ||g_t||_2^2 and almost-sure convergence under light-tailed bounded-variance assumptions (Section 4, Theorem 1).
Mechanism #r417-layer-wise-quantization-and-sigma-gradient-clipping-variance-reduction
Replacing a single model-wide scalar s_t = ||g_t||_infty with per-layer scalars s_t^{(l)} = ||g_t^{(l)}||_infty (Section 3.2) and clipping coordinate outliers to [-c sigma_t, +c sigma_t] with c = 2.5 before quantization (Section 3.3, Eq. 7) shrinks the maximum scalar s_t and reduces the variance inflation factor kappa = max(|g_t|) / |g_t| across heterogeneous network layers.
Limitation #r417-parameter-localization-for-unshared-random-seeds-and-clipping-bias-limitation
In parameter-server topologies where parameters are evenly sharded across workers, sharing a random seed for b_t allows floating-point scalars s_t to be omitted during worker-to-server push only if each parameter tensor is localized to a single server shard (Section 3.1); additionally, pre-quantization gradient clipping introduces a small truncation bias when |g_{tk}| > c sigma_t.
Application #r417-alexnet-googlenet-and-vgg-imagenet-distributed-training-application
Applied to distributed synchronous SGD training of LeNet, CifarNet, AlexNet, GoogLeNet, and VGG-A on MNIST, CIFAR-10, and ImageNet, achieving 3.04x communication speedup without accuracy loss.

⚠️ Curated Limitations & Negative Results

Artificial Intelligence Ontology (AIO) Grounding

Mapped AIO Classes
AIO Target Class Mapping Relation Mapping Rationale
AIO:FederatedLearning
Federated Learning
related_match TernGrad is a ternary gradient quantization method for distributed data-parallel and federated optimization.

Evidence Sources & Primary Literature

3 Sources
Source ID Publication / Document Type / Access Reference & Checksum
r417-primary-pdf
Wei Wen, Cong Xu, Feng Yan, Chunpeng Wu, Yandan Wang, Yiran Chen, Hai Li
primary paper
full_text
ID: https://papers.nips.cc/paper/2017/hash/2bf283c05b601f21364d052ca0ec798d-Abstract.html
sha256:0c06647103d5…
r417-arxiv-abs
Wei Wen, Cong Xu, Feng Yan, Chunpeng Wu, Yandan Wang, Yiran Chen, Hai Li
primary paper
abstract_only
ID: https://arxiv.org/abs/1705.07878
sha256:977c2ea5ce24…
r417-metadata-cr
Wei Wen, Cong Xu, Feng Yan, Chunpeng Wu, Yandan Wang, Yiran Chen, Hai Li
primary paper
abstract_only
ID: https://papers.nips.cc/paper/2017/hash/2bf283c05b601f21364d052ca0ec798d-Abstract.html
sha256:2cd4f4b4695b…

Ontology & Grounding Topology

Open in interactive network graph →
CONCEPT KIND Algorithms AIMech:TernGradTernaryGradientsForDistributedDeepLearning TernGrad (Stochastic... 4 claims · 3 sources AIO GROUNDING AIO:FederatedLearning (1)

Curation Provenance

Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T08:20:00Z
Updated: 2026-10-19T08:20:00Z
Source File: data/curated/TernGradTernaryGradientsForDistributedDeepLearning.yaml · Schema: curated_concept.yaml

Explore across X-Mech

Related computational & biological Mechs: Embedding Map Network Graph ProteinTraitsMech (ESM-2 pLM) DUFMech (Domain pLM) TraitMech CultureMech All 14 X-Mech Repositories