An Exponential Averaging Process with Strong Convergence Properties
Frederik Köhne, Anton Schiela
TL;DR
This work addresses the limitation of exponential moving averages (EMA) in providing strong stochastic convergence when observations are taken along trajectories of random dynamical systems. It introduces p-EMA, a time-dependent smoothing with weights decaying as $\gamma_n=1-1/(n+1)^p$ for $p\in(\tfrac{1}{2},1]$, and develops an averaging-scheme framework to establish almost-sure convergence under summable decay of correlations. The paper proves convergence guarantees for p-EMA, connects these results to SGD dynamics via invariant measures, and shows how p-EMA can improve adaptive step-size estimators in stochastic optimization, supported by numerical studies. The findings offer a principled smoothing approach with strong theoretical guarantees that enhances online estimation and adaptive algorithms in stochastic, potentially non-stationary, environments.
Abstract
Averaging, or smoothing, is a fundamental approach to obtain stable, de-noised estimates from noisy observations. In certain scenarios, observations made along trajectories of random dynamical systems are of particular interest. One popular smoothing technique for such a scenario is exponential moving averaging (EMA), which assigns observations a weight that decreases exponentially in their age, thus giving younger observations a larger weight. However, EMA fails to enjoy strong stochastic convergence properties, which stems from the fact that the weight assigned to the youngest observation is constant over time, preventing the noise in the averaged quantity from decreasing to zero. In this work, we consider an adaptation to EMA, which we call $p$-EMA, where the weights assigned to the last observations decrease to zero at a subharmonic rate. We provide stochastic convergence guarantees for this kind of averaging under mild assumptions on the autocorrelations of the underlying random dynamical system. We further discuss the implications of our results for a recently introduced adaptive step size control for Stochastic Gradient Descent (SGD), which uses $p$-EMA for averaging noisy observations.
