Choco-SGD and Choco-Gossip Decentralized Optimization with Compressed Communication
Algorithms AIMech:ChocoSgdAndChocoGossipDecentralizedCompressedOptimization DRAFT
Aliases: Choco-SGD, Choco-Gossip, Compressed Decentralized Gossip and Stochastic Gradient Descent
A family of decentralized gossip and stochastic optimization algorithms (Choco-Gossip and Choco-SGD) over a connected network with symmetric doubly stochastic mixing matrix W = [w_{ij}] and spectral gap delta > 0, in which each node i maintains local parameters x_i^{(t)} and public replica states x_hat_j^{(t)} for its neighbors j in N_i, transmits only omega-contractive compressed state differences q_i^{(t)} = Q(x_i^{(t)} - x_hat_i^{(t)}) to update x_hat_i^{(t+1)} = x_hat_i^{(t)} + q_i^{(t)}, and applies consensus mixing x_i^{(t+1)} = x_i^{(t)} + gamma * sum_{j=1}^n w_{ij} (x_hat_j^{(t+1)} - x_hat_i^{(t+1)}) (preceded in Choco-SGD by a local stochastic gradient step x_i^{(t+1/2)} = x_i^{(t)} - eta_t * g_i(x_i^{(t)})), preserving the exact global average (1/n) sum_{i=1}^n x_i^{(t)} under arbitrary biased or randomized contractive compression and achieving linear consensus convergence O((1 - delta^2 * omega / 82)^t) and optimal O(sigma_bar^2 / (mu * n * T)) decentralized stochastic optimization speedup.
Identity Notes & Concept Scope
Distinct from centralized/parameter-server error-compensated compression (AIMech:SparsifiedSgdWithErrorMemoryForKContractiveGradientCompression and AIMech:Ef21MarkovCompressorErrorFeedbackForHeterogeneousNonconvexOptimization, which assume a central coordinator or all-to-all aggregation rather than decentralized graph gossip with spectral gap delta), and from uncompressed periodic local SGD (AIMech:LocalSgdPeriodicAveragingParallelStochasticGradientDescent). No pinned AIO class covers decentralized compressed gossip or Choco-SGD.
Claims & Grounded Evidence
4 Claims
Mechanism
#r382-exact-average-preservation-and-linear-gossip-convergence
Because W is symmetric doubly stochastic (sum_i w_{ij} = 1 = sum_j w_{ij}), the gossip update x_i^{(t+1)} = x_i^{(t)} + gamma * sum_j w_{ij} (x_hat_j^{(t+1)} - x_hat_i^{(t+1)}) preserves the exact global mean x_bar^{(t+1)} = x_bar^{(t)} at every step regardless of compressor bias (Remark 14), and Theorem 2 proves that for gamma = delta * omega / (16 * delta + delta^2 + 4 * beta^2 + 2 * delta * beta^2 - 8 * delta * omega) the joint Lyapunov function e_t = E_Q sum_{i=1}^n (||x_i^{(t)} - x_bar||^2 + ||x_i^{(t)} - x_hat_i^{(t+1)}||^2) contracts geometrically as e_t <= (1 - delta^2 * omega / 82) * e_{t-1}.
Mechanism
#r382-decentralized-linear-speedup-rate
For L-smooth mu-strongly convex objectives with local variance sigma_i^2 and second-moment bound G^2, Theorem 4 proves that Choco-SGD (Algorithm 2) with stepsize eta_t = 4 / (mu * (a + t)) and weights w_t = (a + t)^2 achieves E[f(x_avg^{(T)}) - f*] = O(sigma_bar^2 / (mu * n * T)) + O(kappa * G^2 / (mu * omega * delta^2 * T^2)) + O(G^2 / (mu * omega^3 * delta^6 * T^3)), where the leading O(1/(n*T)) term matches centralized uncompressed mini-batch SGD independent of graph spectral gap delta and compression ratio omega.
Limitation
#r382-neighbor-replica-memory-and-spectral-gap-limitation
Each node i must maintain replica vectors x_hat_j^{(t)} in R^d for itself and its graph neighbors j in N_i (requiring O(deg(i) * d) memory unless nodes share a common random seed for sparsification), and higher-order transient terms scale inversely with delta^2 * omega.
Application
#r382-decentralized-ring-and-torus-optimization-application
Applied to decentralized optimization and consensus averaging over ring and 2D torus network topologies with 1% Top-k sparsification and 1-bit/low-bit quantization.
⚠️ Curated Limitations & Negative Results
- Requires each node i to store state replicas x_hat_j for its graph neighbors unless an unweighted shared-seed random sparsifier is used.
- Requires a connected network topology with a symmetric doubly stochastic mixing matrix W (delta > 0) and bounded second moments E ||g_i(x)||^2 <= G^2.
Artificial Intelligence Ontology (AIO) Grounding
Mapped AIO Classes
| AIO Target Class | Mapping Relation | Mapping Rationale |
|---|---|---|
AIO:FederatedLearning
Federated Learning
|
related_match | Choco-SGD performs decentralized peer-to-peer collaborative training across networked workers without a central server. |
AIO:MarkovChain
Markov Chain
|
related_match | Choco-Gossip mixes compressed node states over a doubly stochastic Markov transition matrix W with spectral gap delta > 0. |
Evidence Sources & Primary Literature
3 Sources| Source ID | Publication / Document | Type / Access | Reference & Checksum |
|---|---|---|---|
r382-primary-pdf
|
Anastasia Koloskova, Sebastian U. Stich, Martin Jaggi
|
primary paper
full_text
|
ID: https://proceedings.mlr.press/v97/koloskova19a.html
sha256:38a4b7820c0c…
|
r382-arxiv-abs
|
Anastasia Koloskova, Sebastian U. Stich, Martin Jaggi
|
primary paper
abstract_only
|
ID: https://arxiv.org/abs/1902.00340
sha256:2dcc881d5ef7…
|
r382-metadata-cr
|
Anastasia Koloskova, Sebastian U. Stich, Martin Jaggi
|
primary paper
abstract_only
|
ID: https://proceedings.mlr.press/v97/koloskova19a.html
sha256:ab928e62ff57…
|
Ontology & Grounding Topology
Open in interactive network graph →Curation Provenance
Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T07:45:00Z
Updated: 2026-10-19T07:45:00Z
Source File:
data/curated/ChocoSgdAndChocoGossipDecentralizedCompressedOptimization.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