cs.LGOct 24, 2026

A Unified Analysis of Importance-Corrected Momentum Actor-Critic over a Markovian Trajectory

Authors: Yanjie Dong, Zhou Su, Wei Wang, Gang Wang

Organizations: School of Cyber-Science and Engineering, Xi’an Jiaotong University · School of Information and Communications Engineering, Xi’an Jiaotong University · State Key Laboratory of Autonomous Intelligent Unmanned Systems, Beijing Institute of Technology

Abstract

We present a generic actor-critic method that combines a ridge-regularized pathwise least-squares temporal-difference (LSTD) critic with importance-corrected momentum (ICM) actors. The critic solves a ridge-regularized system per frame by Newton-Schulz inversion and needs no stepsize, no knowledge of the mixing time, and no rank condition. Averaging stochastic policy gradients over frames of TT consecutive transitions, the ICM actors incur a frame-dependent gradient bias of order 1T\frac{1}{T}. To our knowledge, we establish the first finite-time bounds for ICM actor-critic methods with normalized and unnormalized steps along a single Markovian trajectory. Up to a stationary-to-discounted distribution mismatch, the averaged gradient norm over KK updates is bounded by O(K−1/3+1T){\cal O}(K^{-1/3}+\frac{1}{T}) for normalized steps and by O(K−1/4+1T){\cal O}(K^{-1/4}+\frac{1}{T}) for unnormalized steps. We specialize these bounds to Muon, normalized stochastic gradient descent (SGD) with momentum, SGD with momentum, and AdaNorm, and compare them in the nuclear norm. With a dimension-free stepsize, Muon confines the factor min⁡{m,n}1/2\min\{m,n\}^{1/2} associated with an m×nm\times n actor parameter matrix to the noise and correction terms. The comparison also characterizes when orthogonalization guarantees a larger noiseless ascent than normalization: exactly when the gradient's nuclear rank exceeds the spectral-to-Frobenius curvature ratio.

Figures & tables

Explore similar work

CardsList
  1. Refined Analysis of Entropy-Regularized Actor-Critic

    May 23, 2026Safwan Labbi, Paul Mangold, Daniil Tiapkin +1Reinforcement LearningMaximum Entropy RL

  2. A Concentration Bound for Two-Timescale Actor-Critic Algorithm

    Sep 24, 2026Prashansa Panda, Shalabh BhatnagarAverage-Reward RLTwo-Timescale Stochastic Approximation

  3. Unregularized Convergence of Single-Loop, Entropy-Regularized Natural Actor-Critic

    Aug 20, 2026Zhiqiang TanPolicy OptimizationActor-Critic Methods