Implicitly Normalized Online PCA: A Regularized Algorithm with Exact High-Dimensional Dynamics
Not on the map. Why?
Abstract (arXiv)
Many online learning algorithms, including classical online PCA methods, enforce explicit normalization steps that discard the evolving norm of the parameter vector. We show that this norm can in fact encode meaningful information about the underlying statistical structure of the problem, and that exploiting this information leads to improved learning behavior. Motivated by this principle, we introduce Implicitly Normalized Online PCA (INO-PCA), an online PCA algorithm that removes the unit-norm constraint and instead allows the parameter norm to evolve dynamically through a simple regularized update. We prove that in the high-dimensional limit the joint empirical distribution of the estimate and the true component converges to a deterministic measure-valued process governed by a nonlinear PDE. This analysis reveals that the parameter norm obeys a closed-form ODE coupled with the cosine similarity, forming an internal state variable that regulates learning rate, stability, and sensitivity to signal-to-noise ratio (SNR). The resulting dynamics uncover a three-way relationship between the norm, SNR, and optimal step size, and expose a sharp phase transition in steady-state performance. Both theoretically and experimentally, we show that INO-PCA consistently outperforms Oja's algorithm and adapts rapidly in non-stationary environments. Overall, our results demonstrate that relaxing norm constraints can be a principled and effective way to encode and exploit problem-relevant information in online learning algorithms.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 19 of its 33 references that are in the map. First posted on arXiv 1 Dec 2025.
Builds on
19 of 33 references.
References · 19
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 822
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 142
- 2019An Exponential Learning Rate Schedule for Deep Learningcited by 121
- 2020Online stochastic gradient descent on non-convex losses from high-dimensional inferencecited by 119
- 2020Effects of Parameter Norm Growth During Transformer Training: Inductive Bias from Gradient Descentcited by 40
- 2015The Fast Convergence of Incremental PCAcited by 15
- 2023Latent State Models of Training Dynamicscited by 15
- 2025Grokking Beyond the Euclidean Norm of Model Parameterscited by 14
- 2018A Solvable High-Dimensional Model of GANcited by 13
- 2015Online Principal Component Analysis in High Dimension: Which Algorithm to Choose?cited by 12
- 2017Scaling Limit: Exact and Tractable Analysis of Online Learning Algorithms with Applications to Regularized Regression and PCAcited by 11
- 2019AdaOja: Adaptive Learning Rates for Streaming PCAcited by 10
7 more
- 2023Streaming PCA for Markovian Datacited by 9
- 2024Oja's Algorithm for Streaming Sparse PCAcited by 9
- 2019Robust Streaming PCAcited by 6
- 2017The Scaling Limit of High-Dimensional Online Independent Component Analysiscited by 5
- 2023Fair Streaming Principal Component Analysis: Statistical and Algorithmic Viewpointcited by 5
- 2024Spectral Phase Transition and Optimal PCA in Block-Structured Spiked modelscited by 4
- 2024Exploring the Precise Dynamics of Single-Layer GAN Models: Leveraging Multi-Feature Discriminators for High-Dimensional Subspace Learningcited by 2
Built on by
Not cited yet.