English

Logit Dynamics in Softmax Policy Gradient Methods

Machine Learning 2025-06-17 v1 Artificial Intelligence Machine Learning

Abstract

We analyzes the logit dynamics of softmax policy gradient methods. We derive the exact formula for the L2 norm of the logit update vector: Δz212Pc+C(P) \|\Delta \mathbf{z}\|_2 \propto \sqrt{1-2P_c + C(P)} This equation demonstrates that update magnitudes are determined by the chosen action's probability (PcP_c) and the policy's collision probability (C(P)C(P)), a measure of concentration inversely related to entropy. Our analysis reveals an inherent self-regulation mechanism where learning vigor is automatically modulated by policy confidence, providing a foundational insight into the stability and convergence of these methods.

Cite

@article{arxiv.2506.12912,
  title  = {Logit Dynamics in Softmax Policy Gradient Methods},
  author = {Yingru Li},
  journal= {arXiv preprint arXiv:2506.12912},
  year   = {2025}
}

Comments

7 pages

R2 v1 2026-07-01T03:18:35.108Z