A Unified Analysis of Importance-Corrected Momentum Actor-Critic over a Markovian Trajectory
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 consecutive transitions, the ICM actors incur a frame-dependent gradient bias of order . 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 updates is bounded by for normalized steps and by 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 associated with an 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
| Actor | Class | Constants | |
|---|---|---|---|
| Muon | in ( 13 ) | normalized in | , , |
| NSGDM | normalized in | , , | |
| SGDM | unnormalized | , , | |
| AdaNorm | in ( 14 ) | unnormalized | , , |
| Actor | Rate | Optimization | Noise ( ) | Importance correction ( ) | Second order | Error floor c |
|---|---|---|---|---|---|---|
| Muon a | ||||||
| NSGDM a | ||||||
| SGDM b | – | |||||
| AdaNorm b | – |
Reinforcement Learning
Reinforcement learning studies how agents learn sequential decisions from rewards and interaction, balancing exploration, performance, safety, and adaptation.
See all 1,739 papers