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
- Accelerated O(sqrt(kappa) log(1/eps)) prox/communication complexity requires mu-strong convexity (mu > 0).
- Each client i must maintain a local d-dimensional control variate h_{i,t} across communication rounds.
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 →Curation Provenance
Curator: Antigravity
Method: ai_assisted
Model: Gemini
Created: 2026-10-19T07:55:00Z
Updated: 2026-10-19T07:55:00Z
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