mlsanityData to 7 Oct 2026

When Learning Rates Go Wrong: Early Structural Signals in PPO Actor-Critic

arXiv 2603.09950 ↗First posted 10 Mar 2026cs.LG cs.AIcited by 2

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Builds on 6 of its 11 references that are in the map. First posted on arXiv 10 Mar 2026.

Builds on

6 of 11 references.

References · 6

  1. 2017Proximal Policy Optimization Algorithmscited by 14,441
  2. 2019Similarity of Neural Network Representations Revisitedcited by 1,695
  3. 2016OpenAI Gymcited by 1,684
  4. 2017Deep Reinforcement Learning that Matterscited by 727
  5. 2018BabyAI: A Platform to Study the Sample Efficiency of Grounded Language Learningcited by 238
  6. 2025OUI Need to Talk About Weight Decay: A New Perspective on Overfitting Detectioncited by 7

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.