Search arXiv⌕ Search

arXiv · 2611.01658

Long-Time Behavior of Zero-Noise Mean-Field Adam

Abstract

Adaptive methods such as Adam are the most commonly used methods for training neural networks, but their long-time behavior is not as well understood as that of non-adaptive methods. This is particularly true in the mean-field limit, where the dynamics of infinitely many neurons are described as a distribution over parameters which satisfies a self-consistency condition. In this paper, we study a deterministic continuous-time model of Adam in the mean-field setting. Under a condition relating the averaging rates of the momentum and second-moment variables, the size of the gradients, and a regularization parameter, we prove that the distribution of parameters converges to the set of equilibria. This condition extends the range known for finite-dimensional models, and we show that it is sharp. Under an additional Lojasiewicz-type inequality for distributions, we prove that individual parameter trajectories converge and obtain quantitative rates for this convergence. We verify the qualitative convergence assumptions for a regularized two-layer neural network. We also show that adding noise to the momentum prevents the system from approaching these equilibria and rules out many free-energy type Lyapunov functionals. Finally, numerical experiments illustrate the energy estimates, give evidence of persistent oscillations outside the sufficient parameter range, and compare the observed decay with the predicted rates.

Explore related subjects

Keep this discovery

Explore connections, maps & timelines

BibTeXRIS

Ethan Huffman, Asaf Cohen. 2026-11-01. Long-Time Behavior of Zero-Noise Mean-Field Adam. https://arxiv.org/abs/2611.01658

Cite the original work for its findings. Save a collection to share your selection of sources.

KEEP EXPLORING

Related papers

Common Noise by Random Measures: Constructing Mean-Field Equilibria for Competitive Investment and Hedging

We construct Nash equilibria in mean-field portfolio games of optimal investment and hedging under relative performance concerns with exponential (CARA) utility preferences. Common noise dynamics are modeled by integer-valued random measures, for instance Poisson random measures, in addition to Brownian motions. Agents differ in individual risk aversions, competition weights, and initial capital endowments, while their contingent claim liabilities depend on both common and idiosyncratic risk factors. Mean-field equilibria are characterized by solutions to McKean-Vlasov backward stochastic differential equations with jumps, for which we prove existence and uniqueness of solutions, without assuming mean-field interaction to be small. Moreover, we show how the equilibrium can be constructed from the optimal strategy of a single-agent optimization problem (without mean-field interaction) via an appropriate projection. Using successive changes of measure, our derivation provides a decomposition of the equilibrium strategy into three components with clear interpretations. Finally, we show how a limiting mean-field game of quadratic (instead of utility-based) hedging with relative performance concerns arises for vanishing risk aversion.

math.OC↗

Riemannian EXTRA: Communication-efficient decentralized optimization over compact submanifolds with data heterogeneity

We consider decentralized optimization over a compact Riemannian submanifold in a network of $n$ nodes, where each node $i$ owns a smooth nonconvex local objective $f_i$ defined by its private data, and the goal is to minimize the global objective $F(x)=\frac{1}{n}\sum_{i=1}^n f_i(x)$ using only local communications. A naive Riemannian extension of EXTRA leaves the auxiliary normal-space error uncontrolled and does not inherit the Euclidean convergence guarantees. We develop projection-based Riemannian EXTRA (REXTRA) and a retraction-based counterpart that incorporate a normal correction into both the communicated variable and the auxiliary update, while using only one local-variable-sized message per optimization iteration. By jointly controlling consensus, tangent, and normal errors, our analysis interprets the update as a network-based Riemannian inexact-gradient step. For projection-based REXTRA, a sufficiently small constant stepsize and a local initialization yield vanishing consensus, stationary accumulation points, and an $\mathcal O(1/k)$ best-iterate stationarity--consensus bound. For the retraction-based method, we prove that its difference from the projection update is uniformly second order and establish the same convergence properties and rate under a local initialization. Numerical experiments on representative manifold-valued learning tasks demonstrate the communication efficiency of the REXTRA updates.

math.OC↗

Quantification of Tail-trimming Effects for Interval CVaR Based Nonlinear Regression

The tail trimming strategy is widely adopted for robustness against outlier contamination in statistical learning and operations research. However, it generally lacks the thorough understanding that to what extent the tail-trimming technique is effective for nonlinear regression. In this paper, we address the robust nonlinear regression based on the so-called interval conditional value-at-risk (In-CVaR), which is introduced to guard against outliers by trimming extreme losses. We first analyze the trimming effect of the In-CVaR based estimator under distributional perturbation. Our first finding is that the In-CVaR based estimator is qualitatively robust under distributional perturbation in terms of the Prokhorov metric if and only if the largest portion of losses is trimmed. Furthermore, under sample contamination, we derive a probably approximately correct (PAC) bound for the In-CVaR based estimator, showing that the preliminary covariate filtering that is customary in the literature is not needed for In-CVaR based regression. The nonconvexity of the In-CVaR objective, together with unboundedness of covariates, responses and parameter space create the technical challenges for our developed theory, and thus necessitate the new treatment of the qualitative robustness and PAC analysis. Overall, this study rigorously analyzes robustness properties of In-CVaR based nonlinear regression models, which illustrates to what extent the right tail-trimming of the loss by In-CVaR risk measure is essential and effective for robust nonlinear regression. These theoretical findings are further supported by numerical experiments.

math.OC↗