SearcharxivSearch

arXiv subjects

Weiming Ou

Publications and source records attributed to Weiming Ou.

2 recordsLinked to original sources

Heavy-Tailed First-Order Optimization for Polyak-\L{}ojasiewicz Condition: High-Dimensional Minimax Bounds, High-Probability Guarantee, and Fixed-Dimensional Improvements

We study smooth Polyak--\L{}ojasiewicz (PL) optimization with conditionally unbiased stochastic gradients satisfying \[ \mathbb E\!\left[ \|G_t-\nabla f(x_t)\|^\alpha \mid\mathcal F_{t-1} \right]\le \sigma^\alpha, \qquad 1<\alpha\le2. \] When the dimension may depend on the oracle budget, we prove the noise-adaptive lower bound \[ T_\epsilon = \Omega_\alpha\!\left[ \kappa\log\frac{\Delta_0}{\epsilon} + \kappa \left( \frac{\sigma^2}{\mu\epsilon} \right)^{\frac{\alpha}{2(\alpha-1)}} \right], \] which recovers the noiseless PL lower bound when $\sigma=0$. Under the appropriate mirror-PL condition, we give a centered-clipped mirror-descent method attaining the matching high-probability upper bound up to logarithmic factors, without bounded-domain, bounded-gradient, or sub-Gaussian assumptions. We further characterize the stochastic complexity in prescribed fixed dimensions. For $d=1,2,3$, the optimal stochastic term is \[ \widetilde\Theta_\alpha\!\left[ \left( \frac{\sigma^2}{\mu\epsilon} \right)^{\frac{\alpha}{2(\alpha-1)}} \right]. \] For every fixed $d>3$, the same characterization holds whenever \[ \frac{\alpha}{\alpha-1}\ge d-1. \] In the complementary regime, we provide an upper bound with an additional surface-entropy factor and explicitly identify the remaining gap.

math.OC

Understanding Dynamics of Adam in Zero-Sum Games: An ODE Approach

The remarkable success of the Adam in training neural networks has naturally led to the widespread use of its descent-ascent counterpart, Adam-DA, for solving zero-sum games. Despite its popularity in practice, a rigorous theoretical understanding of Adam-DA still lags behind. In this paper, we derive ordinary differential equations (ODEs) that serve as continuous-time limits of the Adam-DA. These ODEs closely approximate the discrete-time dynamics of Adam-DA, providing a tractable analytical framework for understanding its behavior in zero-sum games. Using this ODE approach, we investigate two fundamental aspects of Adam-DA: local convergence and implicit gradient regularization. Our analysis reveals that the roles of the first- and second-order momentum parameters in zero-sum games are exactly the opposite of their well-documented effects in minimization problems. We validate these predictions through GAN experiments across multiple architectures and datasets, demonstrating the practical implications of this reversed momentum effect.

cs.LG