paper-with-me

홈 › Papers

Softmax is $1/2$-Lipschitz: A tight bound across all $\ell_p$ norms

2025-10-27 · Pravin Nair arxiv

The softmax function is a basic operator in machine learning and optimization, used in classification, attention mechanisms, reinforcement learning, game theory, and problems involving log-sum-exp terms. Existing robustness guarantees of learning models and convergence analysis of optimization algorithms typically consider the softmax operator to have a Lipschitz constant of $1$ with respect to the $\ell_2$ norm. In this work, we prove that the softmax function is contractive with the Lipschitz constant $1/2$, uniformly across all $\ell_p$ norms with $p \ge 1$. We also show that the local Lipschitz constant of softmax attains $1/2$ for $p = 1$ and $p = \infty$, and for $p \in (1,\infty)$, the constant remains strictly below $1/2$ and the supremum $1/2$ is achieved only in the limit. To our knowledge, this is the first comprehensive norm-uniform analysis of softmax Lipschitz continuity. We demonstrate how the sharper constant directly improves a range of existing theoretical results on robustness and convergence. We further validate the sharpness of the $1/2$ Lipschitz constant of the softmax operator through empirical studies on attention-based architectures (ViT, GPT-2, Qwen3-8B) and on stochastic policies in reinforcement learning.

📄 PDF Abstract BibTeX arXiv:2510.23012

Code (0)

등록된 구현이 없습니다.

Tasks

Reinforcement Learning

Similar Papers 제목 키워드 기반

Data-dependent Sample Complexity of Deep Neural Networks via Lipschitz Augmentation

2019-05-09 · NeurIPS 2019 12 · Colin Wei, Tengyu Ma

Existing Rademacher complexity bounds for neural networks rely only on norm control of the weight matrices and depend exponentially on depth via a product of the matrix norms. Lower bounds show that this exponential depe…

Analytical bounds on the local Lipschitz constants of affine-ReLU functions

2020-08-14 · Trevor Avant, Kristi A. Morgansen

In this paper, we determine analytical bounds on the local Lipschitz constants of of affine functions composed with rectified linear units (ReLUs). Affine-ReLU functions represent a widely used layer in deep neural netwo…

On Tighter Generalization Bounds for Deep Neural Networks: CNNs, ResNets, and Beyond

2019-05-01 · ICLR 2019 5 · Xingguo Li, Junwei Lu, Zhaoran Wang, Jarvis Haupt 외

We propose a generalization error bound for a general family of deep neural networks based on the depth and width of the networks, as well as the spectral norm of weight matrices. Through introducing a novel characteriza…

Generalization Bounds

On bounds for norms of reparameterized ReLU artificial neural network parameters: sums of fractional powers of the Lipschitz norm control the network parameter vector

2022-06-27 · Arnulf Jentzen, Timo Kröger

It is an elementary fact in the scientific literature that the Lipschitz norm of the realization function of a feedforward fully-connected rectified linear unit (ReLU) artificial neural network (ANN) can, up to a multipl…

Pay Attention to Attention Distribution: A New Local Lipschitz Bound for Transformers

2025-07-10 · Nikolay Yudin, Alexander Gaponov, Sergei Kudriashov, Maxim Rakhuba arxiv

We present a novel local Lipschitz bound for self-attention blocks of transformers. This bound is based on a refined closed-form expression for the spectral norm of the softmax function. The resulting bound is not only m…