atlas

Research map / Optimization / Optimizers

Accelerating Distributed Stochastic Optimization via Self-Repellent Random Walks

ICLR 2024 · Optimization › Optimizers · approach: Stochastic gradient MCMC · 1 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Replaces the i.i.d. or Markovian token in distributed stochastic approximation with a self-repellent random walk (SRRW) and proves a.s. convergence and CLTs showing the optimization error's asymptotic variance decreases at rate O(1/alpha^2) in the slow-timescale case.

Problem

Distributed stochastic optimization with token algorithms driven by Markov chains: standard random walks sample gradients suboptimally, and finite-time bounds don't capture the effect of the walk on asymptotic efficiency.

Approach

Drive the token with a generalized non-linear self-repellent random walk (SRRW) built on a time-reversible base chain, analyze the coupled SA-SRRW system via a CLT (including two-timescale controlled Markov noise), and show that the asymptotic covariance of optimization error shrinks as alpha grows, faster than the sampling covariance.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use SRRW with increasing α and βₙ = o(γₙ) (case i), instead of the base Markov chainSA-SRRW · Distributed stochastic optimizationasymptotic covariance of optimization error: O(1/α²)Helped
Use case i instead of case iii (γₙ = o(βₙ))SA-SRRW · Distributed stochastic optimizationasymptotic covariance of optimization errorHelped
Use SRRW with γₙ = o(βₙ) (case iii), instead of the base Markov chainSA-SRRW · Distributed stochastic optimizationasymptotic covariance of optimization errorNo clear effect
Use SRRW with α ≥ 5 instead of i.i.d. samplingSGD-SRRW and SHB-SRRW · wikiVote graph, ijcnn1MSEHelped
Use case i instead of cases ii and iii at identical αSGD-SRRW and SHB-SRRW · wikiVote graph, ijcnn1MSEHelped
Increase α in case iSGD-SRRW · Dolphins graph, ijcnn1, n = 10^7MSE: O(1/α²)Helped
Increase α in the SRRW sampling processSRRW weighted empirical measure · Target-distribution samplingasymptotic sampling covariance: O(1/α)Helped

Surprising findings

Limitations

Experimental setup

Datasets: ijcnn1 (LIBSVM) wikiVote graph Dolphins graph

Benchmarks: ijcnn1 binary classification with L2 regularization

Metrics: mean squared error of SA iterates

Baselines: i.i.d. sampling SGD Metropolis-Hastings random walk (MHRW) driven SGD SHB with MHRW

Builds on: Self-Repellent Random Walk Metropolis-Hastings random walk stochastic heavy ball

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy