AIMech AI

ProxSkip and Scaffnew Probabilistic Proximal Skipping for Accelerated Local Gradient Descent

Algorithms AIMech:ProxSkipAndScaffnewProbabilisticProximalSkippingForAcceleratedLocalGradientDescent DRAFT
Aliases: ProxSkip, Scaffnew, SProxSkip, Probabilistic Proximal Skipping
A randomized composite first-order optimization algorithm (ProxSkip) and its federated consensus instantiation (Scaffnew) for minimizing f(x) + psi(x) where f is L-smooth and mu-strongly convex and psi is a proper closed convex function with an expensive proximity operator prox_{(gamma/p) psi}. At each step t, the algorithm forms a shifted gradient iterate x_hat_{t+1} = x_t - gamma * (nabla f(x_t) - h_t), draws theta_t ~ Bernoulli(p), evaluates x_{t+1} = prox_{(gamma/p) psi}(x_hat_{t+1} - (gamma/p) h_t) with probability p (and skips the prox via x_{t+1} = x_hat_{t+1} with probability 1 - p), and updates the control variate h_{t+1} = h_t + (p / gamma) * (x_{t+1} - x_hat_{t+1}) so h_t -> nabla f(x_*). Setting gamma = 1/L and p = 1/sqrt(kappa) (kappa = L/mu) achieves O(kappa log(1/eps)) gradient steps and O(sqrt(kappa) log(1/eps)) expected prox/communication evaluations without any data-similarity assumption.
Identity Notes & Concept Scope Distinct from SCAFFOLD (AIMech:ScaffoldStochasticControlledAveragingForFederatedLearning, which uses deterministic K-step local schedules and requires O(kappa log(1/eps)) communication rounds under arbitrary heterogeneity) and from FedProx (AIMech:FedProxProximalFederatedOptimization, which adds a static proximal penalty (mu/2)||w - w^t||^2 without control variates or O(sqrt(kappa)) communication acceleration). Scaffnew is the exact consensus-splitting specialization of ProxSkip where psi is the indicator of x_1 = ... = x_n and prox_{(gamma/p) psi} reduces to exact model averaging across clients.

Claims & Grounded Evidence

4 Claims
Mechanism #r392-firm-nonexpansiveness-and-lyapunov-contraction
Using firm nonexpansiveness of P(x) = prox_{(gamma/p) psi}(x) and Q(x) = x - P(x) (Lemma 3.3: ||P(x) - P(y)||^2 + ||Q(x) - Q(y)||^2 <= ||x - y||^2), Lemmas 3.4-3.5 and Theorem 3.6 prove that the Lyapunov potential Psi_t = ||x_t - x_*||^2 + (gamma^2 / p^2)||h_t - h_*||^2 with h_* = nabla f(x_*) satisfies E[Psi_{t+1}] <= (1 - zeta) Psi_t for any 0 < gamma <= 1/L and 0 < p <= 1, where zeta = min{gamma * mu, p^2}.
Mechanism #r392-scaffnew-accelerated-communication-complexity
When applied to the consensus reformulation min_{x_1,...,x_n} (1/n) sum_{i=1}^n f_i(x_i) + psi(x_1,...,x_n) where psi is the indicator of x_1 = ... = x_n, ProxSkip yields Scaffnew (Algorithm 2): with gamma = 1/L and communication probability p = 1/sqrt(kappa), Corollary 4.2 proves O(kappa log(1/eps)) local gradient steps and O(sqrt(kappa) log(1/eps)) expected communication rounds under arbitrary client heterogeneity, and Theorem 5.7 achieves O(sqrt(kappa / delta) log(1/eps)) decentralized mixing steps over a graph with spectral gap delta.
Limitation #r392-strong-convexity-and-stochastic-neighborhood-limitation
The O(sqrt(kappa) log(1/eps)) communication acceleration requires mu-strong convexity of f (mu > 0) and full client participation during the Bernoulli(p) synchronization rounds; under stochastic gradients with expected smoothness constants (A, B, C), SProxSkip (Theorem 5.5) converges to an O((gamma / mu) C) variance neighborhood unless step size gamma is reduced.
Application #r392-federated-and-decentralized-regularized-logistic-regression
Applied to federated and decentralized L2-regularized logistic regression on LIBSVM datasets (w8a, mushrooms) and synthetic heterogeneous quadratic problems, empirically confirming the 1/sqrt(kappa) optimal communication probability scaling.

⚠️ 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 Scaffnew instantiates ProxSkip on the consensus constraint of federated learning to achieve accelerated O(sqrt(kappa) log(1/eps)) communication rounds via local gradient steps.

Evidence Sources & Primary Literature

3 Sources
Source ID Publication / Document Type / Access Reference & Checksum
r392-primary-pdf
Konstantin Mishchenko, Grigory Malinovsky, Sebastian Stich, Peter Richtarik
primary paper
full_text
ID: https://doi.org/10.48550/arXiv.2202.09357
sha256:5076874962f8…
r392-arxiv-abs
Konstantin Mishchenko, Grigory Malinovsky, Sebastian Stich, Peter Richtarik
primary paper
abstract_only
ID: https://arxiv.org/abs/2202.09357
sha256:c50e2c7e49cd…
r392-metadata-cr
Konstantin Mishchenko, Grigory Malinovsky, Sebastian Stich, Peter Richtarik
primary paper
abstract_only
ID: https://doi.org/10.48550/arXiv.2202.09357
sha256:2128baf69899…

Ontology & Grounding Topology

Open in interactive network graph →
CONCEPT KIND Algorithms AIMech:ProxSkipAndScaffnewProbabilisticProximalSkippingForAcceleratedLocalGradientDescent ProxSkip and Scaffnew... 4 claims · 3 sources AIO GROUNDING AIO:FederatedLearning (1)

Curation Provenance

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