When Learning Rates Go Wrong: Early Structural Signals in PPO Actor-Critic
Authors: Alberto Fernández-Hernández, Cristian Pérez-Corral, Jose I. Mestre, Manuel F. Dolz, Jose Duato, Enrique S. Quintana-Ortí
Organizations: Universitat Polit`ecnica de Val`encia, Valencia, Spain · Universitat Jaume I, Castell´o de la Plana, Spain · Openchip & Software Technologies, Barcelona, Spain
Deep Reinforcement Learning systems are highly sensitive to the learning rate (LR), and selecting stable and performant training runs often requires extensive hyperparameter search. In Proximal Policy Optimization (PPO) actor--critic methods, small LR values lead to slow convergence, whereas large LR values may induce instability or collapse. We analyse this phenomenon from the behavior of the hidden neurons in the network using the Overfitting-Underfitting Indicator (OUI), a metric that quantifies the balance of binary activation patterns over a fixed probe batch. We introduce an efficient batch-based formulation of OUI and derive a theoretical connection between LR and activation sign changes, clarifying how a correct evolution of the neuron's inner structure depends on the step size. Empirically, across three discrete-control environments and multiple seeds, we show that OUI measured at only 10% of training already discriminates between LR regimes. We observe a consistent asymmetry: critic networks achieving highest return operate in an intermediate OUI band (avoiding saturation), whereas actor networks achieving highest return exhibit comparatively high OUI values. We then compare OUI-based screening rules against early return, clip-based, divergence-based, and flip-based criteria under matched recall over successful runs. In this setting, OUI provides the strongest early screening signal: OUI alone achieves the best precision at broader recall, while combining early return with OUI yields the highest precision in best-performing screening regimes, enabling aggressive pruning of unpromising runs without requiring full training.
Figures & tables
Fig. 1: Return (top row), actor OUI ( OUIa , middle row), and critic OUI ( OUIc , bottom row) as a function of LR across CartPole, LunarLander, and MiniGrid. Metrics are measured at 10% of training. Points denote median across seeds; shaded regions indicate IQR.
Recall bin
Ret.
OUI
KL
Clip
Div.
Flip
Ret.+OUI
Ret.+KL
Ret.+Clip
Ret.+Div.
Ret.+Flip
(0.00,0.05]
—
—
0.01 0.03 153
0.03 0.05 116
—
0.09 0.04 34
0.40 0.05 10
0.36 0.05 11
0.40 0.05 10
0.40 0.05 10
0.40 0.05 10
(0.05,0.10]
—
0.58 0.09 12
0.04 0.07 173
—
0.62 0.10 13
0.42 0.10 19
0.80 0.10 10
0.47 0.10 17
0.47 0.10 17
0.47 0.10 17
0.46 0.06 11
(0.10,0.15]
0.42 0.14 26
0.64 0.11 14
0.50 0.11 18
0.55 0.14 20
0.56 0.11 16
0.18 0.15 68
0.82 0.11 11
0.53 0.12 19
0.50 0.14 22
0.53 0.12 19
0.39 0.12 26
(0.15,0.20]
—
0.47 0.19 32
0.40 0.20 40
0.42 0.20 38
0.42 0.20 38
0.45 0.16 29
0.59 0.16 22
0.50 0.20 32
0.50 0.19 30
0.50 0.20 32
—
(0.20,0.25]
0.41 0.22 44
0.47 0.22 38
—
—
0.40 0.25 50
0.30 0.22 60
0.42 0.22 43
0.47 0.25 43
0.46 0.22 39
0.47 0.25 43
—
(0.25,0.30]
0.38 0.28 60
0.50 0.28 46
0.38 0.28 60
0.39 0.28 59
0.40 0.28 58
0.27 0.26 78
0.38 0.27 58
0.46 0.26 46
0.45 0.26 47
0.45 0.26 47
—
TABLE I: Recall-matched comparison of early screening rules. Success is defined as belonging to the top 20% of final return within each environment. Each cell reports precision, recall, and nQ , stacked vertically. Only entries with nQ≥10 are shown. Best precision in each recall bin is highlighted in bold.
Reinforcement learning studies how agents learn sequential decisions from rewards and interaction, balancing exploration, performance, safety, and adaptation.
Activation functions are what make deep networks expressive: without them, the model collapses to a linear map. Yet we still evaluate training mostly from the outside, through loss, accuracy, return, or final calibration, while the internal structural evolution of the network remains largely unobserved. In this paper, we argue that the Overfitting--Underfitting Indicator (OUI) should be understood as a first practical observable of that internal structure. Across our recent results, OUI consistently appears as an early, label-free, activation-based signal that reveals whether a network is entering a poor or promising training regime before convergence. In supervised learning, it anticipates weight decay regimes; in reinforcement learning, it discriminates learning-rate regimes early in PPO actor--critic; and in online control, it can drive layer-wise weight decay adaptation. Read together with recent evidence that activation patterns tend to stabilize earlier than parameters, these results suggest a broader research direction: an activation-centric theory of training dynamics. OUI is becoming an empirical foothold toward this theory.
Alberto Fernández-Hernández, Jose I. Mestre, Cristian Pérez-Corral +3
Universitat Politècnica de València, Valencia, Spain · Universitat Jaume I, Castelló de la Plana, Spain · Openchip & Software Technologies S.L., Spain
Despite its widespread use, Proximal Policy Optimization with clipping (PPO-Clip) remains difficult to tune, and the interactions among critic learning, clipping, and rollout reuse remain incompletely understood. We develop a non-asymptotic analysis of PPO-Clip as a closed-loop actor--critic system. It captures actor--critic coupling, nonsmooth probability-ratio clipping, finite-batch reuse, and predictable early stopping under explicit coverage and critic regularity assumptions, using raw GAE and Monte Carlo critic targets. Our synchronous and asynchronous guarantees jointly characterize policy stationarity and the tracking accuracy of the learned critic, with explicit dependence on algorithmic parameters. A sufficient coupling condition gives optimization, critic tracking, clipping, and finite-batch errors a common amplification bound. The asynchronous result also requires a delay-dependent critic stepsize restriction; violating these conditions does not establish divergence. For finite layered MDPs with tabular critics, a uniform bound on the actual clipped-gradient class replaces complete-trajectory counting. A verified growing-horizon family has polynomial sample complexity, and a two-time-scale schedule gives O(T−2/5) stationarity and critic-tracking bounds with explicit fresh-rollout accounting. These results together advance our understanding about PPO and provide theoretical guidance in tuning.
Junwei Su, Mengfan Liu, Yanyong Zhang +1
School of Artificial Intelligence and Data Science, University of Science and Technology of China · Department of Computer Science, University of Hong Kong
An agent's performance stagnating at a suboptimal level is a common problem in deep on-policy RL. Focusing on PPO, we show that plateaus in certain regimes arise not because of known exploration, capacity, or optimisation challenges, but because sample-based estimates of the loss eventually become poor proxies for the true objective over the course of training. Looking deeper, PPO alternates between sampling rollouts from several parallel environments online using the current policy (which we call the "outer loop") and performing repeated minibatch SGD steps against this offline dataset (the "inner loop"). In our work, we abstract away the inner loop, and conceptually model the outer loop as standard stochastic optimisation. The step size is then controlled by the regularisation strength towards the previous policy and the gradient noise by the number of samples collected between policy update steps. This framing predicts that, much like in SGD, if the outer step size is too large relative to the noise, updates become uninformative and lead to the policy thrashing around a local optimum instead of converging. Recasting PPO in this light makes it clear that there are two ways to address this particular type of learning stagnation: either reduce the step size or increase the number of samples collected between updates. We validate the predictions of our model and conclude that increasing the number of parallel environments is a simple way to avoid these plateaus by simultaneously altering both these factors. Applying our analysis and scaling PPO to more than 1M parallel environments enables monotonic performance improvement up to one trillion transitions and leads to vastly superior performance compared to prior baselines in a complex open-ended domain.