Polyak-Ruppert Averaged Q-learning is a technique that averages Q-iterates to mitigate transient bias and stabilize learning.
It leverages full and tail averaging schemes to derive asymptotic linearity, Gaussian approximations, and optimal finite-sample error bounds.
The approach extends to synchronous, asynchronous, and function-approximation settings, enabling robust online inference and improved convergence guarantees.
Searching arXiv for the cited papers to ground the article with fresh references.
Polyak–Ruppert averaged Q-learning is the practice of replacing the last Q-learning iterate by an average of iterates, most commonly the Cesàro average
QˉT=T1∑t=1TQt,
or, in some regimes, a tail average over the final segment of the trajectory. In the stochastic-approximation view of Q-learning, averaging suppresses transient bias and exposes a more regular fluctuation structure, which has led to central limit theorems, functional central limit theorems, non-asymptotic concentration bounds, and parameter-free optimal-rate results across synchronous, asynchronous, discounted, average-reward, and function-approximation settings (Li et al., 2021, Khodadadian et al., 27 May 2025, Naskar et al., 8 Aug 2025, Liu, 23 Sep 2025, Rubtsov et al., 8 Apr 2026, Rubtsov et al., 17 May 2026).
1. Basic formulation and averaging schemes
In discounted tabular Q-learning, the asynchronous update studied in several recent analyses updates only the visited coordinate: Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),
with all other entries unchanged (Rubtsov et al., 8 Apr 2026). A closely related formulation writes
which is the asynchronous tabular recursion used in a high-probability concentration analysis (Khodadadian et al., 27 May 2025). In synchronous tabular settings with a generative model, all state-action coordinates are updated simultaneously through an empirical Bellman operator Tt, and the averaged estimator remains
The most common averaging rule is full-trajectory Polyak–Ruppert averaging. In that setting, inference and concentration are formulated in terms of QˉT−Q∗, or equivalently in terms of the normalized partial sum
Recent work shows that the averaging window is itself a design choice rather than a fixed convention. For power-law decay to zero step sizes ηt,n=η(1−t/n)ν, including the linear-decay-to-zero case ν=1, the usual full Polyak–Ruppert average is described as not appropriate, because early iterates behave too much like constant-step-size iterates and may not be centered well enough around Q⋆. In that regime, the relevant estimator is a tail average over the last Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),0 iterates (Bonnerjee et al., 5 Apr 2026).
A related variant, sample-averaged Q-learning, combines iterate averaging with within-update batch averaging. Its update uses a batch-averaged empirical Bellman operator, while inference is still based on the time-average
This places Polyak–Ruppert averaging inside a broader family of averaged Q-learning estimators (Panda et al., 2024).
2. Asymptotic statistical theory in synchronous tabular settings
A detailed asymptotic theory was established for synchronous, tabular, discounted Q-learning under a generative model (Li et al., 2021). The central object is the standardized partial-sum process
Under bounded fourth moments, a Lipschitz or gap condition controlling the greedy-policy switch, and slowly decaying step sizes such as Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),3 with Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),4, this process converges weakly to a rescaled Brownian motion: Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),5
At Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),6, this yields the ordinary central limit theorem
and the estimator attains the semiparametric efficiency bound among regular asymptotically linear estimators in that model (Li et al., 2021).
A practical consequence is fully online inference through a self-normalized statistic constructed from the full partial-sum path. The limiting distribution is pivotal and does not require explicit estimation of the asymptotic covariance, which differentiates the method from batch-mean or covariance-plug-in procedures (Li et al., 2021).
The finite-sample side of the same theory gives a non-asymptotic Qk+1(s,a)=Qk(s,a)+αk1{(s,a)=(Sk,Ak)}(R(Sk,Ak)+γa′maxQk(Sk′,a′)−Qk(s,a)),1 analysis for Qk+1(s,a)=Qk(s,a)+αk1{(s,a)=(Sk,Ak)}(R(Sk,Ak)+γa′maxQk(Sk′,a′)−Qk(s,a)),2. For polynomial step sizes, the dominant term matches the instance-dependent lower bound in the tabular setting, and in the worst case the leading term yields the optimal minimax sample complexityQk+1(s,a)=Qk(s,a)+αk1{(s,a)=(Sk,Ak)}(R(Sk,Ak)+γa′maxQk(Sk′,a′)−Qk(s,a)),3 for Qk+1(s,a)=Qk(s,a)+αk1{(s,a)=(Sk,Ak)}(R(Sk,Ak)+γa′maxQk(Sk′,a′)−Qk(s,a)),4-accuracy in Qk+1(s,a)=Qk(s,a)+αk1{(s,a)=(Sk,Ak)}(R(Sk,Ak)+γa′maxQk(Sk′,a′)−Qk(s,a)),5 (Li et al., 2021). Entropy-regularized Q-learning admits analogous asymptotic and finite-sample results without the hard-max Lipschitz assumption, because the soft Bellman operator is smooth and has a unique fixed point (Li et al., 2021).
3. High-probability concentration and the stochastic-approximation viewpoint
A recent general theorem treats Polyak–Ruppert averaging as a modular stochastic-approximation construction rather than a Q-learning-specific device (Khodadadian et al., 27 May 2025). The base recursion is
The theorem assumes a norm Qk+1(s,a)=Qk(s,a)+αk1{(s,a)=(Sk,Ak)}(R(Sk,Ak)+γa′maxQk(Sk′,a′)−Qk(s,a)),8 whose square is Qk+1(s,a)=Qk(s,a)+αk1{(s,a)=(Sk,Ak)}(R(Sk,Ak)+γa′maxQk(Sk′,a′)−Qk(s,a)),9-smooth, a fixed point Tt0 of the mean operator Tt1, invertibility of Tt2, local pseudo-smoothness of Tt3, and sub-Gaussian noise at the fixed point and in the Jacobian (Khodadadian et al., 27 May 2025).
The theorem is deliberately modular. If, for every Tt4, the non-averaged iterates satisfy
Tt5
with probability at least Tt6, then the averaged iterate obeys
Tt7
with probability at least Tt8. The term Tt9 is the main averaging term and contains the QˉT=T1∑t=1TQt0 rate with leading QˉT=T1∑t=1TQt1 concentration, whereas QˉT=T1∑t=1TQt2 is a transient term associated with the part of the trajectory preceding entry into the local pseudo-smoothness region. If the operator is globally smooth, or if QˉT=T1∑t=1TQt3, then QˉT=T1∑t=1TQt4 disappears entirely. The leading behavior is summarized as
Specialized to asynchronous tabular Q-learning with i.i.d. samples, the same framework uses the average
QˉT=T1∑t=1TQt6
The sampling distribution is induced by a behavior policy QˉT=T1∑t=1TQt7, and the exploration parameter is
QˉT=T1∑t=1TQt8
A key structural condition is greedy uniqueness:
for every QˉT=T1∑t=1TQt9, the greedy action QˉT−Q∗0 is unique. This yields the local pseudo-smoothness required by the general theorem. With QˉT−Q∗1, QˉT−Q∗2, the analysis uses QˉT−Q∗3 and QˉT−Q∗4, and the mean operator satisfies an explicit contraction bound whose factor is
QˉT−Q∗5
provided
QˉT−Q∗6
With step size QˉT−Q∗7 and QˉT−Q∗8, the averaged Q-learning iterate satisfies
That result is positioned as the first high-probability bound on the full error distribution for averaged Q-learning, with a step size that does not depend on T1∑t=1T(Qt−Q∗).1 (Khodadadian et al., 27 May 2025). A plausible implication is that Polyak–Ruppert averaging can be analyzed in Q-learning without choosing a confidence-specific schedule, provided a sufficiently sharp raw-iterate bound is available.
The broader stochastic-approximation background for such results includes linear analyses in which averaged iterates admit exact covariance formulas in the CLT and directional non-asymptotic concentration inequalities whose leading term matches that covariance up to universal constants (Mou et al., 2020).
4. Asynchronous updates, Markovian sampling, and Gaussian approximation
The asynchronous case is fundamentally different from the synchronous generative-model setting because the data are Markovian and only one coordinate is updated at each iteration. One line of work formulates the trajectory T1∑t=1T(Qt−Q∗).2 as an irreducible and aperiodic finite-state Markov chain with stationary distribution T1∑t=1T(Qt−Q∗).3, defines the state-action visitation matrix
T1∑t=1T(Qt−Q∗).4
and measures exploration through
T1∑t=1T(Qt−Q∗).5
Under the local quadratic regularity condition
T1∑t=1T(Qt−Q∗).6
polynomial step sizes T1∑t=1T(Qt−Q∗).7 with T1∑t=1T(Qt−Q∗).8, and the matrix
T1∑t=1T(Qt−Q∗).9
the normalized averaged error converges to a Gaussian law with covariance ηt,n=η(1−t/n)ν0, and the partial-sum process converges weakly in ηt,n=η(1−t/n)ν1 to a scaled Brownian motion (Liu, 23 Sep 2025). The same work gives a non-asymptotic central limit theorem in 1-Wasserstein distance; its simplified corollary with ηt,n=η(1−t/n)ν2 has rate
ηt,n=η(1−t/n)ν3
up to the manuscript’s formatting imperfections (Liu, 23 Sep 2025).
A more refined Gaussian approximation result assumes that the chain of transition triples ηt,n=η(1−t/n)ν4 is uniformly geometrically ergodic and studies the usual Polyak–Ruppert average
ηt,n=η(1−t/n)ν5
With polynomial stepsize ηt,n=η(1−t/n)ν6, ηt,n=η(1−t/n)ν7, the finite-sample distribution of ηt,n=η(1−t/n)ν8 is approximated by a Gaussian law over hyper-rectangles, with the best leading rate obtained at ηt,n=η(1−t/n)ν9: ν=10
up to logarithmic factors (Rubtsov et al., 8 Apr 2026). The limiting covariance is
ν=11
and the proof relies on a Poisson-equation decomposition of the Markovian noise into a martingale-difference component plus smaller remainders (Rubtsov et al., 8 Apr 2026).
These Markovian analyses correct a common oversimplification. Averaged Q-learning is not restricted to i.i.d. or synchronous sampling. The available theory covers non-IID trajectories, exploration-dependent covariances, and process-level weak convergence, although at the cost of stronger mixing and local regularity assumptions (Liu, 23 Sep 2025, Rubtsov et al., 8 Apr 2026).
5. Step-size design, tail averaging, and inferential regimes
The choice of learning-rate schedule determines not only optimization dynamics but also the correct averaging window. For the power-law decay-to-zero family
ν=12
with linear decay to zero as the special case ν=13, the iterates behave like a constant-step-size method for ν=14 with ν=15, because ν=16 there, and only stabilize near the end where ν=17 (Bonnerjee et al., 5 Apr 2026). In this triangular-array setting, the correct Polyak–Ruppert estimator is not the full average over ν=18, but a tail average over the last ν=19 iterates (Bonnerjee et al., 5 Apr 2026).
The non-asymptotic theory for that regime shows two phases. The first is fast initial forgetting, with Q⋆0-type decay. The second is a terminal statistical error of order
Q⋆1
which is analogous in spirit to the Q⋆2 error of polynomially decaying schedules (Bonnerjee et al., 5 Apr 2026). Under Assumptions 1 and 2, compactness of Q⋆3, and
Q⋆4
the tail average satisfies the central limit theorem
Q⋆5
for any Q⋆6 and Q⋆7 (Bonnerjee et al., 5 Apr 2026). The covariance Q⋆8 is stated to be positive definite or semidefinite and independent of Q⋆9, but an exact closed-form expression is described as highly intractable (Bonnerjee et al., 5 Apr 2026).
That same analysis establishes a strong invariance principle for the tail partial sums and derives a time-uniform Gaussian approximation. Because direct estimation of Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),00 is difficult, the paper argues that the strong approximation hints at an easily implementable Gaussian bootstrap procedure by running multiple independent chains of the approximating Gaussian recursion in parallel (Bonnerjee et al., 5 Apr 2026).
The main interpretive distinction is therefore not merely between averaged and non-averaged Q-learning, but between full averaging and tail averaging under different step-size geometries. A plausible implication is that “Polyak–Ruppert averaged Q-learning” is not a single estimator class with a universal windowing rule; the averaging operator must be matched to the learning-rate schedule.
6. Extensions beyond standard discounted tabular Q-learning
Recent work extends Polyak–Ruppert averaging to settings where the Bellman operator is a nonlinear semi-norm contraction rather than a standard norm contraction. In that framework, the update has mean map Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),01 satisfying
and discounted Q-learning in Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),04 (Naskar et al., 8 Aug 2025). The difficulty is that the span semi-norm is not monotone, which had obstructed earlier Polyak–Ruppert analyses. The key technical device is to couple the semi-norm contraction with a suitably induced monotone norm and to recast the averaged error as a linear recursion plus a nonlinear perturbation (Naskar et al., 8 Aug 2025). With the universal step size
the averaged iterate achieves parameter-free optimal rates: Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),06
with Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),07 independent agents (Naskar et al., 8 Aug 2025). The stated framework covers synchronous and asynchronous updates, single-agent and distributed settings, and both simulator-generated and Markovian data (Naskar et al., 8 Aug 2025).
Entropy-regularized asynchronous Q-learning with linear function approximation furnishes another extension. There, the parameter update is
with polynomially decaying step sizes Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),09, Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),10, and Polyak–Ruppert average
Under uniform geometric ergodicity, bounded features, and regularity of the projected soft Bellman equation, the normalized averaged error Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),12 admits a Gaussian approximation in convex distance. Optimizing over Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),13 gives Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),14 and rate
and entropy regularization smooths the Bellman operator and guarantees uniqueness of the soft optimal policy, avoiding degeneracy that can obstruct central limit theorems in ordinary Q-learning (Rubtsov et al., 17 May 2026).
Sample-averaged Q-learning provides a different extension in which averaging occurs both within updates and across time. With batch size Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),17, step size Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),18, Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),19, Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),20, and iterate average Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),21, the asymptotic normalization becomes
so the batch schedule directly changes the rate. The resulting functional central limit theorem supports random-scaling confidence intervals that avoid direct estimation of the covariance matrix Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),23 (Panda et al., 2024).
Setting
Averaging object
Representative result
Synchronous discounted tabular
Full average Qt+1(st,at)=Qt(st,at)+αt(rt+γa∈AmaxQt(st+1,a)−Qt(st,at)),24
Across these variants, the common role of Polyak–Ruppert averaging is stable: it converts a recursively updated Q-estimator into an object with sharper asymptotic variance, tractable Gaussian limits, or improved finite-sample concentration. The main points of divergence are equally stable: the admissible step-size schedules, the required local smoothness or uniqueness assumptions, the sampling model, and whether the correct averaging operator is the full Cesàro average or a tail window.
“Emergent Mind helps me see which AI papers have caught fire online.”
Philip
Creator, AI Explained on YouTube
Sign up for free to explore the frontiers of research
Discover trending papers, chat with arXiv, and track the latest research shaping the future of science and technology.Discover trending papers, chat with arXiv, and more.