AIMech AI

QSGD (Quantized Stochastic Gradient Descent) with Elias Coding

Algorithms AIMech:QsgdQuantizedStochasticGradientDescentWithEliasCoding DRAFT
Aliases: QSGD, Quantized SGD, Quantized Stochastic Gradient Descent, QSVRG (Quantized SVRG)
A communication-efficient family of unbiased stochastic gradient quantization and lossless prefix-coding schemes (Alistarh, Grubic, Li, Tomioka, & Vojnovic 2017) for distributed data-parallel optimization on R^n that, given a number of quantization levels s >= 1 and a vector v != 0, maps each coordinate v_i to Q_s(v)_i = ||v||_2 * sgn(v_i) * xi_i(v, s) where xi_i(v, s) is independently rounded to (l_i + 1)/s with probability p_i = s * |v_i| / ||v||_2 - l_i and to l_i/s with probability 1 - p_i (for l_i = floor(s * |v_i| / ||v||_2)), and encodes the resulting sparse integer tuple (||v||_2, signs, non-zero run-length gaps, and quantization levels k_i = s * xi_i) via Elias recursive gamma/omega integer coding, guaranteeing unbiasedness E[Q_s(v)] = v, variance E[||Q_s(v) - v||_2^2] <= min(n / s^2, n^{1/2} / s) * ||v||_2^2, and expected communication <= (3/2 + 1/2 * log_2(2(s^2 + n)/s)) * s(s + n^{1/2}) + F bits per step (<= 2.818 * n + 32 bits when s = n^{1/2} with only 2x variance blowup).
Identity Notes & Concept Scope Distinct from biased 1-bit sign compression signSGD (Bernstein et al. 2018, AIMech:SignSgdAndMajorityVoteCompressedNonconvexOptimization), which discards the l_2 norm ||v||_2 and uses deterministic sign(v_i) without unbiased stochastic rounding or Elias run-length coding; from 1-bit SGD with error feedback (Seide et al. 2014; Karimireddy et al. 2019), which maintains a local residual memory e_t; and from post-training neural weight quantization (such as GPTQ or AWQ), which quantizes trained model weights for inference rather than stochastically quantizing and Elias-coding gradients during distributed SGD/SVRG training.

Claims & Grounded Evidence

3 Claims
Mechanism #r378-claim-unbiased-stochastic-quantization-variance-and-sparsity-bounds
For any v in R^n and quantization level count s >= 1, the stochastic rounding operator Q_s(v) satisfies E[Q_s(v)] = v (unbiasedness), E[||Q_s(v) - v||_2^2] = (||v||_2^2 / s^2) * sum_{i=1}^n p_i * (1 - p_i) <= min(n / s^2, n^{1/2} / s) * ||v||_2^2 (since sum_i p_i(1 - p_i) <= min(n, sum_i p_i) and sum_i p_i <= s * ||v||_1 / ||v||_2 <= s * n^{1/2}), and expected support size (induced sparsity) E[||Q_s(v)||_0] <= s^2 + s * n^{1/2} (Lemma 3.1, Lemma A.5).
Mechanism #r378-claim-elias-coding-communication-complexity-tradeoff
By encoding the index gaps between non-zero coordinates of Q_s(v) and their non-zero integer quantization levels k_i in {1, ..., s} with Elias recursive prefix codes (which use log_2(k) + O(log log k) bits for integer k >= 1 and satisfy Jensen's inequality on concave code-length upper bounds), QSGD transmits at most (3/2 + 1/2 * log_2(2(s^2 + n) / s)) * s * (s + n^{1/2}) + F bits in expectation (where F = 32 is float precision for ||v||_2); in particular, at s = n^{1/2}, QSGD increases gradient second-moment variance by at most 1 + min(n/s^2, n^{1/2}/s) = 2 while transmitting <= 2.818 * n + 32 bits per step instead of 32*n bits (~11.3x compression) (Theorem 3.4, Corollary 3.3, Theorem A.6).
Application #r378-claim-qsvrg-nonconvex-and-multi-gpu-bucketed-training
Extending Q_{n^{1/2}} to variance-reduced finite-sum optimization in Quantized SVRG (QSVRG, Theorem B.2 / G.2) achieves linear convergence E[f(y^{(P+1)}) - f(x^*)] <= (1/2)^P * (f(x_0) - f(x^*)) using <= P * (F + 2.8*n) * (T + 1) bits per processor, matching optimal communication lower bounds up to constants; in smooth non-convex and asynchronous optimization (Theorems C.1 and D.1) and bucketed d-block quantization (Section 3.3, variance min(d/s^2, d^{1/2}/s)) on 16-GPU ImageNet AlexNet, CIFAR-10 ResNet-110, and AN4 speech LSTM, 4-bit and 2-bit QSGD achieve 2.5x-3x end-to-end training speedups without accuracy loss.

⚠️ Curated Limitations & Negative Results

Artificial Intelligence Ontology (AIO) Grounding

Mapped AIO Classes
AIO Target Class Mapping Relation Mapping Rationale
AIO:LogisticRegression
Logistic Regression
related_match Provides communication-efficient quantized SGD and SVRG for convex, strongly convex, and non-convex empirical risk and neural network classification.

Evidence Sources & Primary Literature

3 Sources
Source ID Publication / Document Type / Access Reference & Checksum
r378-primary-pdf
Dan Alistarh, Demjan Grubic, Jerry Li, Ryota Tomioka, Milan Vojnovic
primary paper
full_text
ID: https://papers.nips.cc/paper/2017/hash/6c340f25839e6acdc73414517203f5f0-Abstract.html
sha256:3bd13b1eab43…
r378-arxiv-abs
Dan Alistarh, Demjan Grubic, Jerry Li, Ryota Tomioka, Milan Vojnovic
primary paper
abstract_only
ID: https://arxiv.org/abs/1610.02132
sha256:a259a0dd960b…
r378-metadata-cr
Dan Alistarh, Demjan Grubic, Jerry Li, Ryota Tomioka, Milan Vojnovic
primary paper
abstract_only
ID: https://papers.nips.cc/paper/2017/hash/6c340f25839e6acdc73414517203f5f0-Abstract.html
sha256:f2bfd70587b8…

Ontology & Grounding Topology

Open in interactive network graph →
CONCEPT KIND Algorithms AIMech:QsgdQuantizedStochasticGradientDescentWithEliasCoding QSGD (Quantized... 3 claims · 3 sources AIO GROUNDING AIO:LogisticRegression (1)

Curation Provenance

Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T07:41:00Z
Updated: 2026-10-19T07:41:00Z
Source File: data/curated/QsgdQuantizedStochasticGradientDescentWithEliasCoding.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