← Search

Alexey Naumov

17 accepted papers

2026

Gaussian Approximation for Two-Timescale Linear Stochastic Approximation

AAAI 2026technical

In this paper, we establish non-asymptotic bounds for accuracy of normal approximation for linear two-timescale stochastic approximation (TTSA) algorithms driven by martingale difference or Markov noise. Focusing on both the last iterate and Polyak–Ruppert averaging regimes, we derive bounds for nor

Cited by 0SourcePDFScholar
2026

High-Order Error Bounds for Markovian LSA with Richardson–Romberg Extrapolation

AAAI 2026technical

In this paper, we study the bias and high-order error bounds of the Linear Stochastic Approximation (LSA) algorithm with Polyak-Ruppert (PR) averaging under Markovian noise. We focus on the version of the algorithm with constant step size and propose a novel decomposition of the bias via a lineariza

Cited by 0SourcePDFScholar
2026

Tight Bounds for Schrodinger Potential Estimation in Unpaired Data Translation

ICLR 2026poster

Modern methods of generative modelling and unpaired data translation based on Schrodinger bridges and stochastic optimal control theory aim to transform an initial density to a target one in an optimal way. In the present paper, we assume that we only have access to i.i.d. samples from initial and f…

Cited by 0SourceScholar
2025

Nonasymptotic Analysis of Stochastic Gradient Descent with the Richardson–Romberg Extrapolation

ICLR 2025poster

We address the problem of solving strongly convex and smooth minimization problems using stochastic gradient descent (SGD) algorithm with a constant step size. Previous works suggested to combine the Polyak-Ruppert averaging procedure with the Richardson-Romberg extrapolation to reduce the asymptot…

Cited by 4SourcePDFScholar
2025

Statistical inference for Linear Stochastic Approximation with Markovian Noise

NeurIPS 2025poster

In this paper we derive non-asymptotic Berry–Esseen bounds for Polyak–Ruppert averaged iterates of the Linear Stochastic Approximation (LSA) algorithm driven by the Markovian noise. Our analysis yields $O(n^{-1/4})$ convergence rates to the Gaussian limit in the Kolmogorov distance. We further estab…

Cited by 0SourceScholar
2024

Demonstration-Regularized RL

ICLR 2024poster

Incorporating expert demonstrations has empirically helped to improve the sample efficiency of reinforcement learning (RL). This paper quantifies theoretically to what extent this extra information reduces RL's sample complexity. In particular, we study the demonstration-regularized reinforcement le…

Cited by 0SourcePDFScholar
2024

Gaussian Approximation and Multiplier Bootstrap for Polyak-Ruppert Averaged Linear Stochastic Approximation with Applications to TD Learning

NeurIPS 2024poster

In this paper, we obtain the Berry–Esseen bound for multivariate normal approximation for the Polyak-Ruppert averaged iterates of the linear stochastic approximation (LSA) algorithm with decreasing step size. Moreover, we prove the non-asymptotic validity of the confidence intervals for parameter es…

Cited by 4SourcePDFScholar
2024

Generative Flow Networks as Entropy-Regularized RL

AISTATS 2024poster

The recently proposed generative flow networks (GFlowNets) are a method of training a policy to sample compositional discrete objects with probabilities proportional to a given reward via a sequence of actions. GFlowNets exploit the sequential nature of the problem, drawing parallels with reinforcem…

2024

Group and Shuffle: Efficient Structured Orthogonal Parametrization

NeurIPS 2024poster

The increasing size of neural networks has led to a growing demand for methods of efficient finetuning. Recently, an orthogonal finetuning paradigm was introduced that uses orthogonal matrices for adapting the weights of a pretrained model. In this paper, we introduce a new class of structured matri…

Cited by 10SourcePDFScholar
2024

SCAFFLSA: Taming Heterogeneity in Federated Linear Stochastic Approximation and TD Learning

NeurIPS 2024poster

In this paper, we analyze the sample and communication complexity of the federated linear stochastic approximation (FedLSA) algorithm. We explicitly quantify the effects of local training with agent heterogeneity. We show that the communication complexity of FedLSA scales polynomially with the inver…

Cited by 9SourcePDFScholar
2023

Fast Rates for Maximum Entropy Exploration

ICML 2023poster

We address the challenge of exploration in reinforcement learning (RL) when the agent operates in an unknown environment with sparse or no rewards. In this work, we study the maximum entropy exploration problem of two different types. The first type is visitation entropy maximization previously cons…

2023

First Order Methods with Markovian Noise: from Acceleration to Variational Inequalities

NeurIPS 2023poster

This paper delves into stochastic optimization problems that involve Markovian noise. We present a unified approach for the theoretical analysis of first-order gradient methods for stochastic optimization and variational inequalities. Our approach covers scenarios for both non-convex and strongly co…

Cited by 22SourcePDFScholar
2023

Model-free Posterior Sampling via Learning Rate Randomization

NeurIPS 2023poster

In this paper, we introduce Randomized Q-learning (RandQL), a novel randomized model-free algorithm for regret minimization in episodic Markov Decision Processes (MDPs). To the best of our knowledge, RandQL is the first tractable model-free posterior sampling-based algorithm. We analyze the performa…

Cited by 3SourcePDFScholar
2022

From Dirichlet to Rubin: Optimistic Exploration in RL without Bonuses

ICML 2022oral

We propose the Bayes-UCBVI algorithm for reinforcement learning in tabular, stage-dependent, episodic Markov decision process: a natural extension of the Bayes-UCB algorithm by Kaufmann et al. 2012 for multi-armed bandits. Our method uses the quantile of a Q-value function posterior as upper confide…

Cited by 24SourcePDFScholar
2022

Local-Global MCMC kernels: the best of both worlds

NeurIPS 2022accept

Recent works leveraging learning to enhance sampling have shown promising results, in particular by designing effective non-local moves and global proposals. However, learning accuracy is inevitably limited in regions where little data is available such as in the tails of distributions as well as in…

2022

Optimistic Posterior Sampling for Reinforcement Learning with Few Samples and Tight Guarantees

NeurIPS 2022accept

We consider reinforcement learning in an environment modeled by an episodic, tabular, step-dependent Markov decision process of horizon $H$ with $S$ states, and $A$ actions. The performance of an agent is measured by the regret after interacting with the environment for $T$ episodes. We propose an…

2021

Tight High Probability Bounds for Linear Stochastic Approximation with Fixed Stepsize

NeurIPS 2021poster

This paper provides a non-asymptotic analysis of linear stochastic approximation (LSA) algorithms with fixed stepsize. This family of methods arises in many machine learning tasks and is used to obtain approximate solutions of a linear system $\bar{A}\theta = \bar{b}$ for which $\bar{A}$ and $\bar{b…

Cited by 30SourcePDFScholar