AIMech AI

Local SGD Periodic Model Averaging Parallel Stochastic Gradient Descent

Algorithms AIMech:LocalSgdPeriodicAveragingParallelStochasticGradientDescent DRAFT
Aliases: Local SGD, Parallel Restarted SGD, Periodic Model Averaging SGD, K-Step Averaging Stochastic Gradient Descent
A communication-efficient parallel stochastic optimization algorithm (Local SGD) in which K workers independently advance local parameter vectors via x_{t+1}^k = x_t^k - eta_t * grad f_{i_t^k}(x_t^k) and synchronize by global averaging x_{t+1}^k = (1/K) sum_{j=1}^K (x_t^j - eta_t * grad f_{i_t^j}(x_t^j)) only at communication steps t + 1 in I_T with maximum synchronization gap gap(I_T) <= H, bounding the mean squared worker drift from the virtual average x_bar_t = (1/K) sum_{k=1}^K x_t^k by 4 * eta_t^2 * G^2 * H^2 and achieving the optimal O(sigma^2 / (mu * K * T)) linear K-worker speedup with up to H = O(sqrt(T / K)) fewer communication rounds.
Identity Notes & Concept Scope Distinct from control-variate client-drift correction in heterogeneous federated learning (AIMech:ScaffoldStochasticControlledAveragingForFederatedLearning), from proximal heterogeneous federated optimization with variable inexact local solvers (AIMech:FedProxProximalHeterogeneousFederatedOptimization), and from compressed every-step SGD with error memory (AIMech:SparsifiedSgdWithErrorMemoryForKContractiveGradientCompression). No pinned AIO class covers Local SGD periodic averaging or its virtual-average drift analysis.

Claims & Grounded Evidence

4 Claims
Mechanism #r383-virtual-average-variance-reduction-and-drift-bound
For the virtual average sequence x_bar_t = (1/K) sum_{k=1}^K x_t^k, Lemma 3.2 establishes that the variance of the averaged stochastic gradient is reduced by a factor of K (E ||g_t - g_bar_t||^2 <= sigma^2 / K), and Lemma 3.3 proves that for stepsizes satisfying eta_t <= 2 * eta_{t+H} and synchronization gap at most H, the expected squared worker deviation from x_bar_t is uniformly bounded by (1/K) sum_{k=1}^K E ||x_bar_t - x_t^k||^2 <= 4 * eta_t^2 * G^2 * H^2.
Mechanism #r383-linear-parallel-speedup-and-communication-reduction
Under L-smoothness and mu-strong convexity with stepsizes eta_t = 4 / (mu * (a + t)), shift a > max(16 * kappa, H), and quadratic iterate weights w_t = (a + t)^2, Theorem 2.2 proves E[f(x_hat_T)] - f(x*) <= O(sigma^2 / (mu * K * T)) + O(kappa * H^2 * G^2 / (mu * T^2)) + O(G^2 * (kappa + H)^3 / (mu * T^3)), achieving linear K-fold parallel speedup whenever H = O(sqrt(T / (K * b))).
Limitation #r383-bounded-second-moment-and-heterogeneous-drift-limitation
The drift bound in Lemma 3.3 assumes a uniform second-moment bound E ||grad f_i(x)||^2 <= G^2 and shared objective f across workers; when workers optimize heterogeneous local objectives f_k != f with large H, local iterates drift toward distinct local optima unless corrected.
Application #r383-communication-reduced-parallel-training-application
Applied to parallel mini-batch and asynchronous stochastic optimization of regularized logistic regression and convex empirical risk minimization with periodic worker averaging.

⚠️ 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 Local SGD provides the foundational periodic local-update and model-averaging mechanism underlying Federated Averaging (FedAvg) and parallel data-center training.
AIO:LogisticRegression
Logistic Regression
related_match Local SGD is benchmarked on parallel regularized logistic regression objectives.

Evidence Sources & Primary Literature

3 Sources
Source ID Publication / Document Type / Access Reference & Checksum
r383-primary-pdf
Sebastian U. Stich
primary paper
full_text
ID: https://openreview.net/forum?id=SJSuGs0KoFX
sha256:91dab280c55e…
r383-arxiv-abs
Sebastian U. Stich
primary paper
abstract_only
ID: https://arxiv.org/abs/1805.09767
sha256:0a9300c4815d…
r383-metadata-cr
Sebastian U. Stich
primary paper
abstract_only
ID: https://openreview.net/forum?id=SJSuGs0KoFX
sha256:e064dd4c0653…

Ontology & Grounding Topology

Open in interactive network graph →
CONCEPT KIND Algorithms AIMech:LocalSgdPeriodicAveragingParallelStochasticGradientDescent Local SGD Periodic... 4 claims · 3 sources AIO GROUNDING AIO:FederatedLearning (2)

Curation Provenance

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