Softmax is $1/2$-Lipschitz: A tight bound across all $\ell_p$ norms
Abstract
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 with respect to the norm. In this work, we prove that the softmax function is contractive with the Lipschitz constant , uniformly across all norms with . We also show that the local Lipschitz constant of softmax attains for and , and for , the constant remains strictly below and the supremum 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 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.
Cite
@article{arxiv.2510.23012,
title = {Softmax is $1/2$-Lipschitz: A tight bound across all $\ell_p$ norms},
author = {Pravin Nair},
journal= {arXiv preprint arXiv:2510.23012},
year = {2025}
}
Comments
Under review