arXiv Science⌕ Search
Home/Archive/arXiv

arXiv · 2610.26555

Beyond Finite-Sum Optimization: A Primal-Dual Method for Decentralized Finite-Max Optimization

Zhouxing Luo·Lei Wang·Xin Liu
View original source ↗Read source document ↗Export metadata ↓

Abstract

Many decentralized systems are judged by their worst-performing agents rather than their average performance. We study finite-max optimization over a connected undirected network, where each agent knows only its own smooth, possibly nonconvex objective. Quadratic regularization yields a smooth finite-sum representation with a scalar dual variable and locally computable clipped weights. We develop an alternating primal-dual gradient tracking (APDGT) method, using neighbor communication without any global or centralized projections. Under smoothness and level-boundedness assumptions, a localized Lyapunov analysis with vanishing regularization establishes bounded primal iterates and Clarke stationarity of every accumulation point of their network average. The averaged sequence contains an $\varepsilon$-stationary point, measured by near-active gradients, within $O (\varepsilon^{-3})$ iterations. Synthetic tests and decentralized logistic regression on Fashion-MNIST illustrate convergence behavior and potential improvements in worst-agent performance.

Explore related subjects

math.OC

Keep this discovery

Explore connections, maps & timelines

BibTeXRIS

Zhouxing Luo, Lei Wang, Xin Liu. 2026-10-19. Beyond Finite-Sum Optimization: A Primal-Dual Method for Decentralized Finite-Max Optimization. https://arxiv.org/abs/2610.26555

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

KEEP EXPLORING

Related papers

arXiv2026

The Principle of Optimality in Dynamic Programming: A Pedagogical Note

The principle of optimality is a fundamental aspect of dynamic programming, which states that the optimal solution to a dynamic optimization problem can be found by combining the optimal solutions to its sub-problems. While this principle is generally applicable, it is often only taught for problems with finite or countable state spaces in order to sidestep measure-theoretic complexities. Therefore, it cannot be applied to classic models such as inventory management and dynamic pricing models that have continuous state spaces, and students may not be aware of the possible challenges involved in studying dynamic programming models with general state spaces. To address this, we provide conditions and a self-contained simple proof that establish when the principle of optimality for discounted dynamic programming is valid. These conditions shed light on the difficulties that may arise in the general state space case. We provide examples from the literature that include the relatively involved case of universally measurable dynamic programming and the simple case of finite dynamic programming where our main result can be applied to show that the principle of optimality holds.

math.OC↗
arXiv2026

A Geometric Framework for Stochastic Iterations

This paper concerns models and convergence principles for dealing with stochasticity in a wide range of algorithms arising in nonlinear analysis and optimization in Hilbert spaces. It proposes a flexible geometric framework within which existing solution methods can be recast and improved, and new ones can be designed. Almost sure weak, strong, and linear convergence results are established in particular for stochastic fixed point iterations, the stochastic gradient descent method, and stochastic extrapolated parallel algorithms for feasibility problems. In these areas, the proposed algorithms exceed the features and convergence guarantees of the state of the art. Numerical applications to signal and image recovery are provided.

math.OC↗
arXiv2026

Attention's forward pass and Frank-Wolfe

We study the hardmax limit of self-attention dynamics for token embeddings obtained in the zero-temperature ($β\to+\infty$) regime, and relate it to the finite-$β$ setting. In this limit, the update rule can be viewed as a Frank-Wolfe step for a quadratic objective over the convex hull of the current token embeddings. When the key-query matrix is negative semidefinite, the method linearly contracts all tokens to a single cluster at the origin. When it is positive semidefinite, extending the hardmax rule to the entire convex hull induces a Voronoi diagram: vertices are stationary, interior points remain in their initial cells, and each token moves along a straight line toward its cell's vertex, yielding (super-)exponential convergence. As a byproduct, we also establish well-posedness of the associated ODE limit in this regime. Returning to the finite-$β$ regime, we model self-attention dynamics as a Markov chain and prove dynamic metastability: with high probability, interior tokens reach near-vertex configurations in a constant number of steps and remain within a small neighborhood for times that grow exponentially in the inverse temperature $β$, before ultimately collapsing to the origin. Thus, the hardmax dynamics accurately approximate the finite-$β$ process over exponentially long time horizons.

math.OC↗