alphaFeedback

WorksComputer Science

On Lipschitz Explosion in Deep Neural Networks with Normalization: Consequences for Optimization and Adversarial Robustness

AI contributions
Not recorded for this paper
View details

Abstract

Can a network look stable at every layer and still be exponentially unstable as a whole? The Lipschitz constant of a neural network is a basic stability quantity appearing in optimization guarantees, generalization bounds, and robustness certificates. Deep networks naturally admit two such notions, input Lipschitzness, measuring sensitivity to data perturbations, and parameter Lipschitzness, measuring sensitivity to weight perturbations. We prove that, for deep networks with normalization layers, with batch normalization as the canonical example, both Lipschitz constants can grow exponentially with depth and hence with parameter dimension, even under strong per-layer norm control such as ∥Wℓ∥2≤1\|W_\ell\|_2\le 1, and already for linear activations. Thus, a mechanism widely used to stabilize training can create severe worst-case instabilities that are invisible locally, i.e., from layerwise norm bounds alone. Parameter-Lipschitz explosion turns Lipschitz-dependent nonsmooth optimization guarantees into exponential-in-dimension bounds for deep normalized networks, while input-Lipschitz explosion makes worst-case Lipschitz-based generalization and robustness certificates vacuous. The latter also yields a concrete rank-separation mechanism for adversarial vulnerability. The theory predicts that perturbations introducing new singular directions should be amplified much more strongly than perturbations at the same noise level that remain within the input's original singular subspace. Experiments on MNIST, Fashion-MNIST, and CIFAR-10 support this prediction, showing that rank-creating perturbations cause substantially sharper drops in accuracy, confidence, and margins than same-subspace perturbations.

Introduction

Increasing the depth of neural networks is a key driver of modern performance, boosting image classification [1] and text generation in large language models [2]. Yet increased depth also makes optimization more challenging, motivating architectural stabilizers such as residual connections [1] and normalization layers [3], which are now standard components of modern architectures. In this work, we take a closer look at batch normalization in particular, a seemingly routine component of modern deep neural networks, and study its unexpected implications for optimization, generalization, and adversarial robustness.

A recurring quantity linking depth to both optimization and stability is a model’s (local) Lipschitz constant. Crucially, deep networks come with two different worst-case Lipschitz notions, depending on what is perturbed. Writing FθF_\theta for the network map with parameters θ\theta (and using Frobenius / Euclidean norms for concreteness), we consider

Lip⁡X(Fθ):=sup⁡X≠Y∥Fθ(X)−Fθ(Y)∥F∥X−Y∥F,Lip⁡Θ(F;Xin):=sup⁡θ≠θ′∥Fθ(Xin)−Fθ′(Xin)∥F∥θ−θ′∥2.\operatorname{Lip}_{X}(F_{\theta}) := \sup_{X\ne Y} \frac{\lVert F_{\theta}(X)-F_{\theta}(Y)\rVert_{F}}{\lVert X-Y\rVert_{F}}, \qquad\operatorname{Lip}_{\Theta}(F;X_{\mathrm{in}}) := \sup_{\theta\ne\theta'} \frac{\lVert F_{\theta}(X_{\mathrm{in}})-F_{\theta'}(X_{\mathrm{in}})\rVert_{F}}{\lVert\theta-\theta'\rVert_{2}}.

The parameter Lipschitz constant Lip⁡Θ\operatorname{Lip}_{\Theta} governs how sharply the network output (and hence the induced training objective) can change as the weights move, and it therefore enters worst-case iteration-complexity guarantees in nonsmooth optimization. The input Lipschitz constant Lip⁡X\operatorname{Lip}_{X} governs worst-case sensitivity to data perturbations, and it appears (often via upper bounds) in several generalization and robustness analyses based on margins and stability, including spectrally-normalized bounds and related Lipschitz controls [4, 5]. We use these doubly Lipschitz notions as a lens to study unexpected implications of normalization layers in optimization, generalization, and adversarial robustness.

Main result: both Lipschitz constants can be exponentially large. We show that for deep networks with normalization layers (with batch normalization as the canonical example), both Lip⁡X(Fθ)\operatorname{Lip}_{X}(F_{\theta}) and Lip⁡Θ(F;Xin)\operatorname{Lip}_{\Theta}(F; X_{\mathrm{in}}) can grow exponentially with depth mm, and consequently exponentially with the ambient parameter dimension dd. This occurs even under strong per-layer norm control (e.g., ∥Wℓ∥2≤1\lVert W_{\ell}\rVert_{2} \le1, and in the input-Lipschitz construction even WℓW_{\ell} orthogonal), and already for linear activations. Put differently, we find that while normalization stabilizes many practical training pipelines [3, 6, 7, 8, 9, 10], it causes worst-case doubly Lipschitz explosion as depth increases.

Implications of parameter-Lipschitz explosion for nonsmooth optimization

For smooth objectives, first-order methods admit dimension-independent rates for reaching approximate stationarity [11]. Neural-network training objectives, however, are typically nonsmooth (e.g., due to ReLU activations, pooling layers or normalizations), so stationarity is captured by generalized-gradient notions. A classical tractable notion is Goldstein stationarity [12], and recent algorithms can find approximate Goldstein stationary points with oracle complexity polynomial in the Lipschitz constant LL of the objective [13, 14, 15]. In our setting, LL is not an external constant; instead, it is induced by the network and our lower bound shows it can be exp⁡(Ω(d))\exp(\Omega(d)) for deep normalized architectures. Consequently, Lipschitz-based “dimension-free” guarantees of the form poly⁡(L,1/ε,1/δ)\operatorname{poly}(L,1/\varepsilon,1/\delta) can unfortunately translate into exponential dependence on parameter dd for this model class. Moreover, lower bounds in nonsmooth optimization show that some polynomial dependence on LL is unavoidable in the worst case [16], underscoring that one cannot expect a uniform theory for Goldstein stationarity that is both fully general and only polylogarithmic in LL. Motivated by the optimization barrier created by exploding Lipschitz constants, we consider a relaxed directional notion (probabilistic Goldstein stationarity) and give a basic algorithm that finds such points with oracle complexity that does not depend polynomially on LL. This begets a new quest to find more refined solution concepts (with probabilistic Goldstein stationarity as a basic starting point) that are theoretically tractable in poly⁡(d,1/ε)\operatorname{poly}(d,1/\varepsilon) time and hence in poly⁡(d,1/ε,log⁡L)\operatorname{poly}(d,1/\varepsilon,\log L) time.

Implications of input-Lipschitz explosion for generalization and adversarial robustness

The same phenomenon impacts learning-theoretic uses of Lipschitz control. Since Lip⁡X(Fθ)\operatorname{Lip}_{X}(F_{\theta}) (or upper bounds on it) enters several margin-based generalization bounds and Lipschitz-based stability arguments [4, 5, 17], an exponential lower bound indicates that these worst-case guarantees can deteriorate rapidly with depth, even when each layer is individually norm-controlled. Therefore, we conclude that the Lipschitzness assumption may not be realistic for studying generalization in deep neural networks with normalization layers, since this assumption becomes vacuous exponentially fast with depth.

The input Lipschitzness Lip⁡X(Fθ)\operatorname{Lip}_{X}(F_{\theta}) lens on batch normalization also sheds new light on recent empirical evidence that batch normalization can have an outsized effect on adversarial robustness [18, 19, 20]. By showing theoretically that the Lipschitzness of deep neural networks grows exponentially fast with depth as a byproduct of batch normalization, we show that normalization layers significantly increase the sensitivity of neural-network outputs to noise in their inputs. Therefore, we can partially explain this empirical observation.

At a high level, our hardness construction for exponential growth of Lip⁡X(Fθ)\operatorname{Lip}_{X}(F_{\theta}) with depth exploits a rank-separation effect. The witness pair consists of a low-rank matrix and a nearby matrix that is full-rank but nearly rank-deficient. Although these two inputs are close in norm, normalization can amplify their separation across layers, leading to an exponential growth in the input Lipschitz constant. This mechanism suggests a corresponding empirical prediction for adversarial vulnerability under batch normalization: If a clean image, or a patch-based representation of it, has approximate low-rank structure (as suggested by classical low-rank patch models for natural images and by low-dimensional structure observed in datasets such as MNIST, Fashion-MNIST, CIFAR-10, and ImageNet [21, 22, 23, 24, 25]) then adding small i.i.d. noise can generically make the representation full-rank while keeping it close to the low-rank set. Our theory predicts that batch-normalized networks amplify this small perturbation through depth, resulting in a large change in the output. We experimentally validate this prediction (see Figure 1).

Original image with pixelwise Gaussian noise and subspace Gaussian noise, showing differing neural-network predictions

Figure 1. We study the effect of different noise structures on the robustness of a multilayer neural network with normalization layers. Above, we add standard i.i.d. Gaussian noise; below, we instead apply noise restricted to the subspace of the original image. In the noisy images, red indicates positive values and blue indicates negative values. As depicted, even though the structured noise in the same subspace as the original image changes the image much more meaningfully (here, from a human perspective, making the 5 look more like an 8 as also observed in the noise pattern) the model still correctly classifies the image with high confidence. In contrast, under pixel-wise noise, despite the change being barely noticeable from a human perspective, the model is completely fooled and misclassifies the image with relatively high confidence. Our theoretical analysis predicts this phenomenon: the original image is low-rank, while i.i.d. Gaussian pixel noise makes the noisy image full-rank but nearly rank-deficient. Due to this relationship between the original image and the noisy one, normalization layers can amplify their separation across layers, causing neural networks’ vulnerability to noise. This separation and vulnerability do not arise when Gaussian noise is added within the same subspace as the original image, even at a much higher magnitude.

Our main contributions are as follows:

  • We distinguish two natural Lipschitz notions for deep networks: parameter Lipschitzness and input Lipschitzness. Through this lens, we reveal unexpected implications of normalization layers for optimization, generalization, and adversarial robustness.

  • For deep networks with normalization layers, we show that both input and parameter Lipschitz constants can grow exponentially with depth (and hence with parameter dimension) even under strong per-layer norm constraints and even with linear activations.

  • We clarify how parameter-Lipschitz explosion propagates to worst-case nonsmooth optimization. Existing guarantees for Goldstein stationarity that are polynomial in the objective Lipschitz constant LL become exponential in the parameter dimension for deep normalized networks. This motivates a new search for tractable stationarity notions with complexity polynomial in dimension and accuracy, but only polylogarithmic in LL. We initiate this search by introducing probabilistic Goldstein stationarity, a basic directional relaxation of Goldstein stationarity derived from its dual formulation. We give a simple function-value algorithm that finds such points with oracle complexity that avoids polynomial dependence on LL.

  • We connect input-Lipschitz explosion to generalization and adversarial robustness. In particular, our results show that worst-case Lipschitz-based generalization, margin, and stability arguments

can become vacuous exponentially fast with depth in normalized networks, even when each layer is norm-controlled.

  • More broadly, our construction identifies a simple mechanism for adversarial sensitivity in batch-normalized networks. Normalization can amplify small input differences when two nearby inputs have different rank structure. Consequently, when the data are approximately low rank, perturbations that introduce new singular directions can be magnified across layers, while perturbations that remain within the original singular subspace may be much less disruptive. Our experiments support this prediction across MNIST, Fashion-MNIST, and CIFAR-10. Perturbations that change the input’s singular structure lead to substantially sharper drops in accuracy, confidence, and classification margins than perturbations at the same noise level that stay within the input’s original singular subspace. This suggests a geometry-dependent failure mode of normalized networks and helps explain why batch normalization can have an outsized effect on adversarial robustness.

Exponential Explosion of Two Lipschitz Constants with Dimension

In this section, we study the Lipschitz constant of deep neural networks. In particular, we show that the presence of normalization layers, with a focus on batch normalization, causes the Lipschitz constant of the network to grow exponentially with depth, and consequently with the dimension of the parameter space dd. We observe this explosion in two notions of Lipschitz continuity: (i) the Lipschitz constant of the deep neural network with fixed weights, viewed as an input–output map with respect to the inputs, and (ii) the Lipschitz constant of the deep neural network with fixed input data, viewed as a function on the parameter space with respect to the weights.

Preliminaries and Notation

Notation. For a matrix XX, ∥X∥F\lVert X\rVert_F is the Frobenius norm and ∥X∥2\lVert X\rVert_2 the operator norm. For a square matrix AA, λmin⁡(A)\lambda_{\min}(A) denotes its smallest eigenvalue. We write Xi:X_{i:} for the ii-th row of XX and X:jX_{:j} for the jj-th column of XX. We denote the set {1,2,…,m}\{1,2,\ldots,m\} by [m][m].

Batch normalization (Column-wise normalization on Rn×k\mathbb{R}^{n\times k}). Let X∈Rn×kX\in\mathbb{R}^{n\times k}. Define the diagonal matrix of column-squared norms

D(X):=diag⁡(XTX)∈Rk×k,D(X)jj=∥X:j∥22.D(X) := \operatorname{diag}(X^{\mathsf{T}}X)\in\mathbb{R}^{k\times k}, \qquad D(X)_{jj}=\lVert X_{:j}\rVert_2^2.

When every column of XX is nonzero (equivalently D(X)≻0D(X)\succ0), the column-wise batch normalization map is BN⁡(X):=XD(X)−1/2\operatorname{BN}(X):=XD(X)^{-1/2}. Equivalently, (BN⁡(X))ij=Xij/∥X:j∥2(\operatorname{BN}(X))_{ij}=X_{ij}/\lVert X_{:j}\rVert_2. We emphasize that the batch normalization operator defined above omits the mean subtraction step for convenience, in line with prior formulations and theoretical studies [9, 26, 27]. Our results extend to mean-subtracted batch normalization by an isometric embedding into the mean-zero subspace; Appendix § 8.8 gives the construction. If D(X)≻0D(X)\succ0, then every column of BN⁡(X)\operatorname{BN}(X) has unit norm, hence

∥BN⁡(X)∥F2=tr⁡(BN⁡(X)TBN⁡(X))=k.(1)\lVert\operatorname{BN}(X)\rVert_F^2=\operatorname{tr}(\operatorname{BN}(X)^{\mathsf{T}}\operatorname{BN}(X))=k. \tag*{(1)}

Remark 2.1 (Regularization at zero columns). If one wants BN⁡\operatorname{BN} to be defined on all of Rn×k\mathbb{R}^{n\times k}, one may work with the standard ε\varepsilon-regularized variant

BN⁡ε(X):=X(D(X)+εIk)−1/2,ε>0,\operatorname{BN}_{\varepsilon}(X):=X\left(D(X)+\varepsilon I_k\right)^{-1/2}, \qquad\varepsilon>0,

and (when desired) take ε↓0\varepsilon\downarrow0 at the end. Our lower bound proof below is written for BN⁡(X)=XD(X)−1/2\operatorname{BN}(X)=XD(X)^{-1/2} on the open set {D(X)≻0}\{D(X)\succ0\}, and remains valid for BN⁡ε\operatorname{BN}_{\varepsilon} with the same qualitative conclusion.

A deep network with normalization layers and linear activation. Fix an integer n≥1n\ge1, i.e., the batch size. Let k0,k1,…,kmk_0,k_1,\ldots,k_m be positive integers (layer widths / feature dimensions). For each layer ℓ∈{1,…,m}\ell\in\{1,\ldots,m\} let Wℓ∈Rkℓ−1×kℓW_\ell\in\mathbb{R}^{k_{\ell-1}\times k_\ell} be a weight matrix. Given an input matrix X(0)∈Rn×k0X^{(0)}\in\mathbb{R}^{n\times k_0}, define the forward iterates

X(ℓ):=fWℓ(X(ℓ−1)):=BN⁡(X(ℓ−1)Wℓ)∈Rn×kℓ,ℓ=1,…,m,(2)X^{(\ell)}:=f_{W_\ell}(X^{(\ell-1)})\mathrel{\mathop:}= \operatorname{BN}(X^{(\ell-1)}W_\ell)\in\mathbb{R}^{n\times k_\ell}, \qquad\ell=1,\ldots,m, \tag*{(2)}

whenever the normalization is well-defined. The depth-mm network map is

Fm:=fWm∘⋯∘fW1:Rn×k0→Rn×km.F_m:=f_{W_m}\circ\cdots\circ f_{W_1}:\mathbb{R}^{n\times k_0}\to\mathbb{R}^{n\times k_m}.

Why linear activations? We focus on linear activations because they provide a conservative setting for a lower bound. By removing nonlinearities, we rule out amplification effects that could come from the activation function itself, so any growth we prove must come from the interaction between the weights and normalization. This choice is also consistent with the mean-field analysis of batch-normalized networks in Yang et al. [ [28], where linear activations exhibit the slowest gradient-norm growth among the activation functions considered, even though the growth can still be exponential in depth. Thus, showing exponential Lipschitz growth already in the linear case demonstrates that the phenomenon is not caused by nonlinear activations. We expect the same mechanism to persist for a broader class of standard nonlinearities, we show this result for networks with ReLU activations in Appendix § 8.9 and leave the rest to future work.

Input and parameter Lipschitzness. We use two Lipschitz notions, depending on whether the perturbation is applied to the input or to the weights. First, fix a sequence of weights (W1,…,Wm)(W_1,\ldots,W_m) and an input domain X\mathcal{X} on which every normalization step is defined. View the network as a map from inputs to outputs. Its input Lipschitz constant is

Lip⁡X(Fm):=sup⁡X,Y∈XX≠Y∥Fm(X)−Fm(Y)∥F∥X−Y∥F.\operatorname{Lip}_{\mathcal{X}}(F_m) := \sup_{\substack{X,Y\in\mathcal{X}\\X\ne Y}} \frac{\lVert F_m(X)-F_m(Y)\rVert_F}{\lVert X-Y\rVert_F}.

Second, fix an input batch Xin∈Rn×k0X_{\mathrm{in}}\in\mathbb{R}^{n\times k_0} and view the network output as a function of the parameters. For a weight tuple θ=(W1,…,Wm)∈Θ⊆∏ℓ∈[m]Rkℓ−1×kℓ\theta=(W_1,\ldots,W_m)\in\Theta\subseteq\prod_{\ell\in[m]}\mathbb{R}^{k_{\ell-1}\times k_\ell}, we use the Euclidean norm obtained by vectorizing all weights, ∥θ∥2:=(∑ℓ=1m∥Wℓ∥F2)1/2\lVert\theta\rVert_2:=\left(\sum_{\ell=1}^{m}\lVert W_\ell\rVert_F^2\right)^{1/2}. The corresponding parameter-to-output map and its Lipschitz constant are

Fm(θ):=Fmθ(Xin),Lip⁡Θ(Fm):=sup⁡θ,θ′∈Θθ≠θ′∥Fm(θ)−Fm(θ′)∥F∥θ−θ′∥2.F_m(\theta):=F_m^\theta(X_{\mathrm{in}}),\qquad\operatorname{Lip}_{\Theta}(F_m):=\sup_{\substack{\theta,\theta'\in\Theta\\\theta\ne\theta'}}\frac{\lVert F_m(\theta)-F_m(\theta')\rVert_F}{\lVert\theta-\theta'\rVert_2}.

The first quantity measures sensitivity to input perturbations, while the second measures sensitivity to changes in the weights.

Remark 2.2. Normalization layer is invariant under positive rescaling since BN⁡(αX)=BN⁡(X)\operatorname{BN}(\alpha X)=\operatorname{BN}(X) for every α>0\alpha>0. Our exponential lower bounds are established on compact admissible domains with scales bounded away from zero uniformly in depth. Specifically, the input domains satisfy 1≤∥X∥F≤k1\le\lVert X\rVert_F\le\sqrt{k}, while every weight matrix in the parameter witnesses has operator norm one.

Explosion of the Input Lipschitz Constant Lip⁡X(Fm)\operatorname{Lip}_{\mathcal{X}}(F_m)

We first show that the input Lipschitz constant Lip⁡X(Fm)\operatorname{Lip}_{\mathcal{X}}(F_m) can grow exponentially with depth. This growth is not an artifact of a large or expansive linear map. In fact, it already occurs in the following highly controlled setting with orthogonal weights:

n=k0=k1=⋯=km=k≥2,X(0)∈Rk×k,Wℓ∈O(k).(3)n=k_0=k_1=\cdots=k_m=k\ge2,\qquad X^{(0)}\in\mathbb{R}^{k\times k},\qquad W_\ell\in\mathbb{O}(k). \tag*{(3)}

Here each weight matrix is orthogonal Wℓ∈O(k)W_\ell\in\mathbb{O}(k), so ∥Wℓ∥2=1\lVert W_\ell\rVert_2=1 and therefore ∥XWℓ∥F≤∥X∥F\lVert XW_\ell\rVert_F\le\lVert X\rVert_F. Thus, the linear part of every layer is non-expansive.

The setting is deliberately restrictive. The linear maps do not increase Frobenius norm, and by (1), every post-normalization activation has fixed norm, ∥X(ℓ)∥F=k\lVert X^{(\ell)}\rVert_F=\sqrt{k}. In particular, the activations remain in a bounded region throughout the network. Nevertheless, we show below that the input–output map can still have an input Lipschitz constant that grows exponentially with the depth mm. Moreover, this growth is witnessed by uncountably many input pairs.

Theorem 2.3 (Exponential lower bound on Lip⁡X(Fm)\operatorname{Lip}_{\mathcal{X}}(F_m)). Fix k≥2k\ge2 and constant η∈(0,1)\eta\in(0,1). There exist constants Ck,η>0C_{k,\eta}>0 and γk,η>0\gamma_{k,\eta}>0 such that, for every depth m∈Nm\in\mathbb{N}, one can choose orthogonal weights W1,…,Wm∈O(k)W_1,\ldots,W_m\in\mathbb{O}(k) with ∥Wℓ∥2=1\lVert W_\ell\rVert_2=1 for all ℓ∈[m]\ell\in[m], for which the depth-mm map FmF_m defined in (2)–(3) satisfies Lip⁡X(Fm)≥Ck,ηeγk,ηm\operatorname{Lip}_{\mathcal{X}}(F_m)\ge C_{k,\eta}e^{\gamma_{k,\eta}m}, where constants Ck,ηC_{k,\eta} and γk,η\gamma_{k,\eta} depend on k,ηk,\eta. Moreover, for this same fixed choice of weights, the lower bound is witnessed by an uncountable family of input pairs.

Our proof construction uses orthogonal weights, so the linear maps themselves do not create any expansion. The instability comes from the interaction between normalization and near rank-deficiency.

We choose two inputs, X0X_0 and Y0Y_0, that are exponentially close, with Y0Y_0 rank-deficient and X0X_0 only slightly perturbed away from being rank-deficient. After mm carefully chosen normalization steps, the trajectory starting from X0X_0 becomes uniformly well-conditioned, while the trajectory starting from Y0Y_0 remains rank-deficient. Thus the two outputs are separated by at least λmin⁡(XmTXm)\sqrt{\lambda_{\min}(X_m^{\mathsf{T}}X_m)} which is relatively high in a well-conditioned matrix, even though the original perturbation was exponentially small. This creates the exponential separation-to-perturbation ratio. We give the full proof of this result in Appendix § 8.6, and provide a proof sketch in Appendix § 8.§ 1.

The next corollary translates the depth-dependent lower bound into a lower bound in terms of the number of parameters. Thus, even when the parameter count is the measure of complexity, the input Lipschitz constant can still grow exponentially.

Corollary 2.4. Fix η∈(0,1)\eta\in(0,1) and set k=2k=2. There exist constants Cη>0C_\eta>0 and Γη>0\Gamma_\eta>0 such that, for every integer d≥4d \ge4, there is a depth-mm network with at most dd scalar weight parameters whose input–output map satisfies Lip⁡X(Fm)≥CηeΓηd\operatorname{Lip}_{X}(F_m) \ge C_\eta e^{\Gamma_\eta d}.

Remark 2.5. Although the constants in our general bounds in Theorem 2.3 depend on kk, the exponential lower bounds hold uniformly over all widths k≥2k \ge2. Indeed, the width-two construction embeds into any larger width via block-diagonal matrices, preserving the same exponential rate and constants independently of kk; see Appendix § 8.10.

Explosion of the Parameter Lipschitz Constant Lip⁡Θ(Fm)\operatorname{Lip}_{\Theta}(F_m)

In Section 2.2, we showed that the input–output map can have an input Lipschitz constant Lip⁡X(Fm)\operatorname{Lip}_{X}(F_m) that grows exponentially with depth. We now show that the same phenomenon occurs when the input is fixed and the network is viewed as a function of its parameters. In other words, normalization can also lead to exponential growth of the parameter Lipschitz constant Lip⁡Θ(Fm)\operatorname{Lip}_{\Theta}(F_m).

Theorem 2.6 (Exponential lower bound on Lip⁡Θ(Fm)\operatorname{Lip}_{\Theta}(F_m)). Fix k≥2k \ge2 and constant η∈(0,1)\eta\in(0,1). There exist constants C~k,η>0\widetilde{C}_{k,\eta}>0 and γ~k,η>0\widetilde{\gamma}_{k,\eta}>0 depending on kk and η\eta such that, for every depth m≥2m \ge2, there is a depth-mm network of the form (2), with ∥Wℓ∥2≤1\lVert W_\ell\rVert_2 \le1 for all ℓ∈[m]\ell\in[m], for which the fixed-input parameter-to-output map Fm(θ):=Fmθ(Xin)F_m(\theta):=F_m^\theta(X_{\mathrm{in}}) with Xin=IkX_{\mathrm{in}}=I_k satisfies Lip⁡Θ(Fm)≥C~k,ηeγ~k,ηm\operatorname{Lip}_{\Theta}(F_m) \ge\widetilde{C}_{k,\eta}e^{\widetilde{\gamma}_{k,\eta}m}.

The proof is a direct reduction from the input-Lipschitz construction in Theorem 2.3. We fix the input to be Xin=IkX_{\mathrm{in}}=I_k and encode the two input witnesses from Theorem 2.3 into the first-layer weights. The two parameter choices differ only in this first layer, while all subsequent weights are kept the same as in the input construction. By the scale-invariance of batch normalization, the resulting forward pass reproduces the same two trajectories after the first layer. Thus the parameter perturbation is as small as the original input perturbation, while the output separation remains bounded below, giving the same exponential separation ratio. Full details are provided in Appendix § 8.7. As in Corollary 2.4, this depth-dependent lower bound can also be translated into an exponential lower bound in the parameter dimension dd.

Corollary 2.7. Fix any η∈(0,1)\eta\in(0,1) and set k=2k=2. There exist constants C~η>0\widetilde{C}_\eta>0 and Γ~η>0\widetilde{\Gamma}_\eta>0 such that, for every integer d≥8d \ge8, one can construct a depth-mm network with at most dd scalar weight parameters for which Lip⁡Θ(Fm)≥C~ηeΓ~ηd\operatorname{Lip}_{\Theta}(F_m) \ge\widetilde{C}_\eta e^{\widetilde{\Gamma}_\eta d}.

From the optimization viewpoint, the relevant quantity is the Lipschitz constant of the training objective that is fed by the network. To make this connection explicit, fix a realizable target parameter θ⋆∈Θ\theta^\star\in\Theta and define

f(θ):=∥Fm(θ)−Fm(θ⋆)∥F,Lip⁡(f):=sup⁡θ,θ′∈Θθ≠θ′∣f(θ′)−f(θ)∣∥θ′−θ∥2,(4)f(\theta):=\lVert F_m(\theta)-F_m(\theta^\star)\rVert_F,\qquad\operatorname{Lip}(f):=\sup_{\substack{\theta,\theta'\in\Theta\\ \theta\ne\theta'}}\frac{\lvert f(\theta')-f(\theta)\rvert}{\lVert\theta'-\theta\rVert_2}, \tag*{(4)}

Here Fm(θ⋆)F_m(\theta^\star) plays the role of the observed label generated by the ground-truth network. The objective function f(θ)f(\theta) is equivalent to the mean-squared loss evaluated on the outputs of the neural network.

The next corollary shows that the exponential growth of Lip⁡Θ(Fm)\operatorname{Lip}_{\Theta}(F_m) transfers to the Lipschitz constant of this induced objective.

Corollary 2.8. Let FmF_m be the parameter-to-output map constructed in Theorem 2.6. There exists a realizable target parameter θ⋆∈Θ\theta^\star\in\Theta such that the objective ff defined in (4) satisfies

Lip⁡(f)≥C~ηeΓ~ηd,\operatorname{Lip}(f) \ge\widetilde{C}_\eta e^{\widetilde{\Gamma}_\eta d},

up to the constants in Corollary 2.7.

Thus, the same exponential sensitivity appears in the induced optimization objective and directly affects convergence guarantees that scale with its Lipschitz constant.

Experiments for Lipschitz Explosion

We now empirically validate the exponential input-Lipschitz growth predicted by our theory. The results are shown in Figure 2 on a logarithmic scale. We focus on the setting of Corollary 2.4 with k=2k = 2. Since the proof of Theorem 2.3 establishes existence through a nonconstructive probabilistic argument over the weights {Wℓ}ℓ=1m\{W_{\ell}\}_{\ell=1}^{m}, we use a constructive heuristic to select the weights in our experiments. Specifically, at each layer ℓ∈[m]\ell\in[m], we choose WℓW_{\ell} greedily so as to maximize the isotropy functional from Definition B.1. In the two-dimensional setting, this amounts to a deterministic grid search over rotation angles. At each layer, we scan a grid of candidate rotations and select the one that maximizes the isotropy of the next activation.

Figure 2. Empirical validation of Lipschitz explosion. The constructed witness pair (X0,Y0)(X_0,Y_0) and its random rotations (QX0,QY0)(QX_0,QY_0) grow exponentially with depth, while the median and 95th percentile over generic random local pairs remain nearly flat.

Figure 2a compares the empirical Lipschitz ratio obtained from the constructed witness pair (X0,Y0)(X_0,Y_0) with the predicted exponential growth. The same plot also shows witnesses obtained by applying random rotations QQ to the input witness pair, yielding pairs of the form (QX0,QY0)(QX_0,QY_0). As predicted by the orthogonal equivariance of the construction, these rotated witnesses exhibit the same exponential growth behavior. Figure 2b compares the constructed witness with generic local input pairs. The random local pairs are generated from i.i.d. standard Gaussian perturbations, and we report both the median and the 95th percentile of their empirical Lipschitz ratios. Unlike the constructed witness, these generic local pairs do not exhibit exponential growth with depth. These experiments support two complementary conclusions. First, the exponential growth predicted by our theory is clearly observable for the constructed witnesses. Second, this behavior is not typical for arbitrary local perturbations. Thus, the witness construction captures a genuine worst-case instability of normalized networks rather than a numerical artifact.

Implications for Optimization Theory in Deep Neural Networks

We now turn to the optimization consequences of the lower bound L:=Lip⁡(f)≥exp⁡(Ω(d))L := \operatorname{Lip}(f) \ge\exp(\Omega(d)) from § 2.3. Dimension-free convergence rates are often viewed as especially desirable in deep learning, since modern networks can have a very large number of parameters. Our results show that, for normalized networks, this interpretation can be misleading. A guarantee may be formally independent of the parameter dimension dd, but if it depends polynomially on the objective Lipschitz constant LL, then Corollary 2.8 implies that the same guarantee can still scale exponentially with dd. In this sense, Lipschitz-dependent “dimension-free” rates need not be truly dimension-free for deep normalized architectures.

This issue is particularly relevant in nonsmooth optimization. Neural-network training objectives are nonconvex, and they are often nonsmooth as well. Even if the outer loss is smooth, standard architectural components such as ReLU [29] and leaky-ReLU activations [30], max-pooling [31], absolute-value regularizers [32], and piecewise-linear combinatorial layers [33, 34] can introduce nonsmoothness into the objective. At the same time, these objectives are typically locally Lipschitz and differentiable almost everywhere, so generalized-gradient notions of stationarity provide the natural language for worst-case analysis. The relevant background on nonsmooth stationarity is recalled in Appendix § 9.§ 1; here, we focus on how the Lipschitz explosion above changes the interpretation of existing convergence guarantees.

Complexity of Finding Goldstein Stationary Points

Existing algorithms for finding (δ,ϵ)(\delta,\epsilon)-approximate Goldstein stationary points have oracle complexity that depends polynomially on the objective Lipschitz constant LL [13, 14, 35, 15, 36]. Such a dependence is acceptable when LL is viewed as a fixed property of the optimization problem. In deep normalized networks, however, LL is not an external constant; it is determined by the architecture and can grow with depth. Our results in § 2 show that LL can be as large as exp⁡(Ω(d))\exp(\Omega(d)). Consequently, a guarantee of the form O(poly⁡(L,1/ϵ,1/δ))O(\operatorname{poly}(L,1/\epsilon,1/\delta)) can translate into exponential dependence on the parameter dimension dd, even when the stated rate has no explicit dependence on dd.

This raises a natural question: can one design algorithms for Goldstein stationarity whose dependence on LL is only logarithmic, for example with oracle complexity polynomial in dd, 1/ϵ1/\epsilon, 1/δ1/\delta, and log⁡L\log L?

We show that, in full generality, the answer is negative. One cannot hope for a randomized first-order method whose oracle complexity is polynomial in dd, 1/ϵ1/\epsilon, 1/δ1/\delta, and only logarithmic in LL, uniformly over all Lipschitz objectives. The obstruction is not specific to neural networks or to nonconvexity: it already appears for convex Lipschitz functions. We rely directly on Kornowski and Shamir [ [16], which gives a lower bound on the oracle complexity of finding (δ,ϵ)(\delta,\epsilon)-Goldstein stationary points of 1-Lipschitz functions.

Corollary 3.1 (Informal). In the worst case, polynomial dependence on the Lipschitz constant LL is unavoidable for finding (δ,ϵ)(\delta,\epsilon)-Goldstein stationary points. In particular, no randomized first-order algorithm can have oracle complexity only poly⁡(d,1/ϵ,1/δ,log⁡L)\operatorname{poly}(d,1/\epsilon,1/\delta,\log L) uniformly over all Lipschitz objectives.

Toward Tractable Solution Concepts

These observations highlight a mismatch between existing nonsmooth optimization guarantees and the regimes that arise in deep normalized networks. As we observed, current guarantees for Goldstein stationarity do not yield truly polynomial-time bounds once the architecture-induced Lipschitz constant LL is taken into account. This suggests a natural quest: to identify weaker, but still meaningful, stationarity notions whose complexity is polynomial in dd, 1/ϵ1/\epsilon, and 1/δ1/\delta, and polynomial only in log⁡L\log L. As a first step in this direction, we introduce probabilistic Goldstein stationarity, a simple directional relaxation of Goldstein stationarity. We use it as a basic baseline rather than a final answer, and give a simple function-value algorithm that finds such points with number of queries independent of LL.

Goldstein stationarity is a worst-case directional condition. To see this, fix δ>0\delta>0 and let ∂δf(x)\partial_\delta f(x) denote the Goldstein subdifferential. For a direction v∈Sd−1v\in\mathbb{S}^{d-1}, consider the support function of the set of Goldstein subdifferentials mδ(x;v):=min⁡g∈∂δf(x)⟨g,v⟩m_\delta(x;v):=\min_{g\in\partial_\delta f(x)}\langle g,v\rangle. By Lemma C.3, Goldstein stationarity can be written in the equivalent form dist⁡(0,∂δf(x))≤ϵ⟺max⁡v∈Sd−1−mδ(x;v)≤ϵ\operatorname{dist}(0,\partial_\delta f(x))\leq\epsilon\Longleftrightarrow\max_{v\in\mathbb{S}^{d-1}}-m_\delta(x;v)\leq\epsilon. Thus, a (δ,ϵ)(\delta,\epsilon)-Goldstein stationary point is one for which no direction has a large negative margin. This dual viewpoint goes back to the work of Goldstein [ [12]. We now relax this worst-case requirement by asking only that large-margin directions occupy a small fraction of the sphere.

Definition 3.2 (Probabilistic Goldstein stationarity (PGS)). Fix δ>0\delta>0, a margin threshold τ>0\tau>0, and a probability level ρ∈(0,1)\rho\in(0,1). Let V∼Unif⁡(Sd−1)V\sim\operatorname{Unif}(\mathbb{S}^{d-1}). We say that xx is (δ,τ,ρ)(\delta,\tau,\rho)-PGS if Pr⁡[mδ(x;V)≥−τ]≥ρ\Pr[m_\delta(x;V)\geq-\tau]\geq\rho. We write xx is (δ,ρ)(\delta,\rho)-PGS as shorthand for (δ,τ,ρ)=(δ,ϵ,ϵ)(\delta,\tau,\rho)=(\delta,\epsilon,\epsilon).

We view probabilistic Goldstein stationarity (PGS) as a first step toward understanding which stationarity notions are tractable for deep normalized networks, rather than as a definitive solution concept. Additionally, we give a simple algorithm for finding PGS points that uses only function evaluations and prove its convergence guarantee.

Theorem 3.3. There exists a randomized function-value algorithm that finds a (δ,τ,ρ)(\delta,\tau,\rho)-PGS point with probability at least 1−γ1-\gamma using O(Δτδρlog⁡Δγτδ)O\left(\frac{\Delta}{\tau\delta\rho}\log\frac{\Delta\gamma}{\tau\delta}\right) queries, with no polynomial dependence on LL.

Implications for Generalization and Robustness in Deep Neural Networks

The implications of Lipschitz explosion results for generalization are relatively direct. As discussed in § 1, blindly assuming Lipschitzness LL for deep normalized neural networks can make the resulting generalization bounds vacuous with depth (and number of parameters). We now turn to the unexpected implications of our results for the robustness of neural networks.

We next connect Lipschitz explosion to decision flips, showing that it can directly undermine the robustness of deep normalized networks to input noise. The input-Lipschitz lower bound in Theorem 2.3 shows that an exponentially small input perturbation can create a large separation in the network output. One might suspect that this only makes global Lipschitz certificates overly pessimistic, without leading to an actual change in the predicted label. The next two informal consequences show that this is not the case. The same construction can be converted into an explicit binary robustness failure. We defer the formal statements and proofs to Appendix § 10. We first show that a constant-margin decision flip can occur within an exponentially small adversarial radius. This provides a theoretical mechanism that helps explain why batch normalization can have an outsized effect on adversarial robustness [18, 19, 20].

Theorem 4.1 (Informal). For every depth mm, consider the same batch-normalized network as in Theorem 2.3 and append a linear readout of Frobenius norm one. The resulting binary classifier has a clean input with score margin Ωk,η(1)\Omega_{k,\eta}(1), but the clean input can be misclassified after a perturbation of Frobenius norm at most e−Ωk,η(m)e^{-\Omega_{k,\eta}(m)}.

The readout is the hyperplane placed halfway between the two output features generated by the witness pair in the proof of Theorem 2.3. Thus, the result does not rely on a separate instability mechanism. It shows that the Lipschitz explosion itself can produce a genuine small-radius adversarial example once the learned representation is passed through a linear classifier. We now explain the mechanism behind the separation observed in Figure 1. The key point is that the adversarial perturbation is not only small in norm, but also rank-creating. This allows normalization to amplify the perturbation across layers, whereas perturbations that remain within the same low-rank branch do not experience the same amplification. This perspective is also consistent with the empirical geometry of standard image datasets, since MNIST, Fashion-MNIST, and CIFAR-10 images exhibit strong approximate low-rank structure in Figure 3 and Figure 5.

Figure 3. Spectral structure of CIFAR-10 luminance and RGB channel matrices. Left: normalized Gram eigenvalues λj(X⊤X)/tr⁡(X⊤X)\lambda_j(X^\top X)/\operatorname{tr}(X^\top X). Right: cumulative spectral energy, with horizontal lines at 90% and 95%. Most of the energy lies in the first few of the 32 singular components.

Figure 5. Spectral evidence for approximate low-rank structure in MNIST and Fashion-MNIST. For each dataset, the left panel shows the median normalized Gram spectrum λj(X⊤X)/tr⁡(X⊤X)\lambda_j(X^\top X)/\operatorname{tr}(X^\top X) with interquartile range, the middle panel shows cumulative spectral energy with the 90% and 95% thresholds, and the right panel reports the distributions of effective rank and the rank needed to capture 95% of the energy. In both datasets, most of the spectral energy is concentrated in a few components, indicating that typical images lie close to a low-rank set.

Proposition 4.2 (Informal). In the same construction, the clean input is rank-deficient, while the adversarially perturbed input is full-rank and only e−Ωk,η(m)e^{-\Omega_{k,\eta}(m)} away. Moreover, the full-rank adversarial output is a constant distance away from the entire rank-deficient output branch of the

network. Thus perturbations that leave the low-rank branch and perturbations that stay within it have fundamentally different behavior under normalization.

We empirically validate the prediction of Proposition 4.2 in both realistic and controlled settings. In Figure 4, we use a ResNet56 model pretrained on CIFAR-10, while in Figure 8 we use toy CNNs with batch normalization trained on MNIST and Fashion-MNIST. In all experiments, we vary the noise variance and compare perturbations with matched pre-clipping energy but different geometric structure. We consider four perturbation families: i.i.d. pixel-wise noise, i.i.d. column-wise noise, i.i.d. row-wise noise, and singular-value noise, which perturbs only the singular values and therefore keeps the image in the same singular subspace.

Figure 4. Effect of noise structure on a pretrained BatchNorm ResNet56 on CIFAR-10. Same-subspace perturbations are substantially less harmful than pixel-, row-, and column-wise noise.

Figure 8. Matched-energy perturbation robustness across MNIST, Fashion-MNIST, and CIFAR-10 BatchNorm models. Each row shows performance as the noise variance σ2\sigma^2 increases under i.i.d. pixel noise, subspace-preserving perturbations, column offsets, and row offsets. Subspace-preserving perturbations typically degrade accuracy, confidence, and margin more slowly and induce fewer prediction changes, while perturbations that move away from the image subspace cause earlier instability.

The results are consistent with the mechanism suggested by Proposition 4.2. Perturbations that introduce new directions or alter the rank structure of the input lead to a much stronger degradation in accuracy, confidence, and margin. In contrast, same-subspace perturbations are substantially less effective, even at comparable noise energy. This supports the view that the vulnerability induced by normalization is not determined only by perturbation size, but also by whether the perturbation moves the input away from its underlying low-rank structure. We defer the detailed experimental setup and additional results to Appendix § 11.

Conclusion

We showed that normalization layers can induce exponential growth in both input and parameter Lipschitz constants, even under strong per-layer norm control and already with linear activations. This Lipschitz explosion changes the interpretation of several standard guarantees for deep normalized networks, making Lipschitz-based optimization, generalization, and robustness analyses potentially vacuous in worst-case regimes. Our results also identify a rank-separation mechanism behind adversarial sensitivity and show that this mechanism is reflected empirically. We hope these findings motivate new stability notions and analysis tools that better capture the behavior of modern normalized architectures.

Acknowledgement

The authors thank Suvrit Sra for introducing them to nonsmooth optimization and its literature.

A.S. and P.J. were partially supported by the ONR grant N00014-24-1-2470. Part of this research was performed while A.S. was visiting the Institute for Mathematical and Statistical Innovation (IMSI), which is supported by the National Science Foundation (Grant No. DMS-2425650).

Related Works

Deep learning theory. Our study is motivated by the fact that depth is both powerful and difficult to manage. Deeper networks have driven many empirical gains, but they also tend to make optimization more fragile. Many standard tools in modern architectures were introduced to make very deep models easier to train, including normalization layers [3, 7], residual skip connections [1], and modified activation functions [37]. In practice, these tools often stabilize training and make it possible to scale networks to much greater depths. We ask whether this practical stabilization can come with unanticipated costs for training, robustness, and statistical behavior. As a first step toward this broader question, in this work, we study batch normalization.

The Lipschitz constant of a neural network appears in several learning-theoretic analyses through Lipschitz- and margin-sensitive complexity measures. For example, Bartlett et al. [ [4] prove margin-based generalization bounds whose complexity term depends on a spectral complexity of the network, given by the product of the layer spectral norms together with an additional correction factor. Similarly, Bubeck and Sellke [ [5] study robustness through the Lipschitz constant, showing tradeoffs between interpolation, model size, and smoothness. These results make Lipschitz control a natural quantity to study when reasoning about generalization and robustness. Our contribution is complementary. We show that, in deep networks with normalization layers, the relevant input Lipschitz constant can itself grow exponentially with depth, even under strong per-layer norm control. Thus, Lipschitz-dependent generalization and robustness guarantees can become vacuous for deep normalized networks in the worst case.

Our analysis is closely related to mean-field studies of batch normalization. Yang et al. [ [28] show that, in the infinite-width limit, gradient norms in batch-normalized networks can grow exponentially with depth. More recently, Meterez et al. [ [27] showed that this explosion can be avoided for linear activations under suitable full-rank conditions on the input batch. Our results show that this conditioning requirement is essential. Even with linear activations and norm-controlled weights, inputs that are close to rank deficiency can lead to exponential growth of the worst-case input Lipschitz constant, and hence to large worst-case sensitivity. The mechanism is tied to the isotropizing effect of normalization layers. Batch normalization tends to push representations toward a more isotropic geometry [9], but near rank-deficiency this same effect can amplify small perturbations into large output separations.

Smooth optimization. Dimension-free convergence guarantees have a long history in smooth optimization. For smooth nonconvex objectives, gradient descent can find an approximate first-order stationary point at a rate that does not depend on the ambient dimension [11, 38]. A large body of work has tried to go beyond first-order stationarity and reach second-order stationary points, where the gradient is small and there are no directions of significant negative curvature [39, 40, 41]. Under smoothness assumptions, such guarantees are possible in polynomial time. However, dimension-free second-order guarantees cannot hold in full generality. Simchowitz et al. [ [42] show that some dependence on the dimension is unavoidable for general smooth functions. Motivated by this limitation, Daneshmand et al. [ [43] identify noise conditions under which stochastic gradient methods can still achieve dimension-free convergence to second-order stationary points, using structure that is natural in deep learning losses. These results rely on smoothness of the objective, an assumption that is often violated in modern neural networks because of ReLU activations, and max-pooling operations. For this reason, we focus here on nonsmooth optimization, where the objective need not be differentiable everywhere.

Nonsmooth optimization. The study of stationarity for nonsmooth Lipschitz functions goes back to the work of Goldstein [ [12], which introduced what are now known as Goldstein stationary points. Unlike in the smooth setting, exact first-order stationarity is not a finite-time target in general nonsmooth nonconvex optimization. This has led recent work to focus on relaxed notions such as (δ,ϵ)(\delta,\epsilon)-Goldstein stationarity. In this direction, Zhang et al. [ [13] gave one of the first non-asymptotic algorithms for finding such points, with complexity scaling polynomially in the Lipschitz constant LL, the radius parameter 1/δ1/\delta, and the accuracy 1/ϵ1/\epsilon. More recently, Davis et al. [ [14] developed a gradient-sampling approach under a standard oracle model, improving the dependence on ϵ\epsilon and LL in certain regimes, at the cost of an additional dependence on the dimension. While these guarantees are polynomial for a fixed Lipschitz constant LL, this distinction is important in deep learning, where LL is not an external constant. It is induced by the network architecture and can grow with depth. Our results show that, for deep networks with normalization layers, this growth can be exponential. Consequently, nonsmooth optimization guarantees that are polynomial in LL can become exponential in depth, even when they are formally dimension-free.

Proofs and Details for Section 2

We begin with a proof sketch of Theorem 2.3, and then give the full proof of Theorems 2.3 and 2.6 together with the required machinery and lemmas.

Proof Sketch of Theorem 2.3

We specialize to the square orthogonal regime (3) and the recursion (2). Since Wℓ∈O(k)W_\ell\in\mathrm{O}(k), the linear maps are non-expansive, so any growth must come from the normalization. Our argument exploits a tension between (i) the ability of BN⁡(⋅)\operatorname{BN}(\cdot) to regularize anisotropy when combined with orthogonal mixing, and (ii) the instability of this regularization near rank-deficient inputs.

We follow Joudaki et al. [ [26] and track isotropy via the functional I(⋅)I(\cdot) in Definition B.1. For normalized matrices X=BN⁡(⋅)X = \operatorname{BN}(\cdot) we have tr⁡(XTX)=k\operatorname{tr}(X^{\mathsf{T}}X) = k by (1), hence I(X)=det⁡(XTX)1/k∈(0,1]I(X) = \det(X^{\mathsf{T}}X)^{1/k} \in(0, 1]. A key input is the one-step lift inequality of Meterez et al. [ [27]: Theorem B.3 shows that for X=BN⁡(⋅)X = \operatorname{BN}(\cdot) and Haar-uniform W∈O(k)W \in\mathrm{O}(k),

E[I(BN⁡(XW))∣X]≥11−S(X)I(X),\mathbb{E}[I(\operatorname{BN}(XW)) \mid X] \ge\frac{1}{1-S(X)} I(X),

where S(⋅)S(\cdot) is the spectral deviation from Definition B.2. Lemma B.4 provides a uniform gap S(X)≥sk,η>0S(X) \ge s_{k,\eta} > 0 whenever I(X)≤1−ηI(X) \le1-\eta. Combining these yields a deterministic multiplicative lift: Corollary B.5 guarantees that, as long as I(X)≤1−ηI(X) \le1-\eta, we can choose WW so that I(BN⁡(XW))≥rk,ηI(X)I(\operatorname{BN}(XW)) \ge r_{k,\eta}I(X) with rk,η=(1−sk,η)−1>1r_{k,\eta} = (1-s_{k,\eta})^{-1} > 1. Iterating this choice gives the greedy geometric increase Corollary B.6, so after mm layers we can enforce I(Xm)≥1−ηI(X_m) \ge1-\eta provided the initial isotropy is at least rk,η−m(1−η)r_{k,\eta}^{-m}(1-\eta).

We realize such an initialization while keeping it exponentially close to rank deficiency. As in the appendix proof of Theorem 2.3, we set Gε=(1−ε)11T+εIG_\varepsilon= (1-\varepsilon)\mathbf{1}\mathbf{1}^{\mathsf{T}} + \varepsilon I and choose X0X_0 with X0TX0=GεX_0^{\mathsf{T}}X_0 = G_\varepsilon. Then diag⁡(Gε)=I\operatorname{diag}(G_\varepsilon) = I, so X0=BN⁡(X0)X_0 = \operatorname{BN}(X_0), and I(X0)=det⁡(Gε)1/k≥ε(k−1)/kI(X_0) = \det(G_\varepsilon)^{1/k} \ge\varepsilon^{(k-1)/k}. Choosing

ε=(1−η)kk−1rk,η−mkk−1\varepsilon= (1-\eta)^{\frac{k}{k-1}} r_{k,\eta}^{-\frac{mk}{k-1}}

ensures rk,ηmI(X0)≥1−ηr_{k,\eta}^{m}I(X_0) \ge1-\eta, hence (by Corollary B.6) we obtain I(Xm)≥1−ηI(X_m) \ge1-\eta along the resulting trajectory. Near-isotropy then forces uniform conditioning: Lemma B.7 yields λmin⁡(XmTXm)≥ckI(Xm)k≥ck(1−η)k=:δk,η\lambda_{\min}(X_m^{\mathsf{T}}X_m) \ge c_k I(X_m)^k \ge c_k(1-\eta)^k =: \delta_{k,\eta}.

To turn this into a Lipschitz lower bound, we compare X0X_0 to a rank-(k−1)(k-1) neighbor. Let X0=UΣVTX_0 = U\Sigma V^{\mathsf{T}} be an SVD with singular values σ1≥⋯≥σk>0\sigma_1 \ge\cdots\ge\sigma_k > 0 and define the matrix Y0:=Udiag⁡(σ1,…,σk−1,0)VTY_0 := U\operatorname{diag}(\sigma_1,\ldots,\sigma_{k-1},0)V^{\mathsf{T}}. Then rank⁡(Y0)<k\operatorname{rank}(Y_0) < k and ∥X0−Y0∥F=σk=ε\lVert X_0-Y_0\rVert_F = \sigma_k = \sqrt{\varepsilon}. Let Yt:=Ft(Y0)Y_t := F_t(Y_0) under the same weights (one may, if desired, interpret BN via the standard ε\varepsilon-regularization from Remark 2.1 to avoid degenerate columns; this does not change the bound). Since each layer is an invertible right-multiplication followed by column-wise rescaling, rank cannot increase, so rank⁡(Yt)<k\operatorname{rank}(Y_t) < k for all tt. Applying Lemma B.8 at depth mm gives ∥Xm−Ym∥F≥λmin⁡(XmTXm)≥δk,η\lVert X_m-Y_m\rVert_F \ge\lambda_{\min}(X_m^{\mathsf{T}}X_m) \ge\delta_{k,\eta}, and therefore

Lip⁡λ(Fm)≥∥Fm(X0)−Fm(Y0)∥F∥X0−Y0∥F≥δk,ηε=Ck,ηrk,ηmk2(k−1)=Ck,ηeγk,ηm,\operatorname{Lip}_{\lambda}(F_m) \ge\frac{\lVert F_m(X_0)-F_m(Y_0)\rVert_F}{\lVert X_0-Y_0\rVert_F} \ge\sqrt{\frac{\delta_{k,\eta}}{\varepsilon}} = C_{k,\eta}r_{k,\eta}^{\frac{mk}{2(k-1)}} = C_{k,\eta}e^{\gamma_{k,\eta}m},

with γk,η=k2(k−1)log⁡rk,η\gamma_{k,\eta} = \frac{k}{2(k-1)}\log r_{k,\eta} and Ck,ηC_{k,\eta} as in Theorem 2.3. Finally, the orthogonal equivariance BN⁡((QX)W)=QBN⁡(XW)\operatorname{BN}((QX)W) = Q\operatorname{BN}(XW) for Q∈O(k)Q \in\mathrm{O}(k) implies Fm(QX)=QFm(X)F_m(QX) = QF_m(X), so the same lower bound is witnessed by the uncountable family of pairs (QX0,QY0)(QX_0,QY_0), as recorded at the end of the proof of Theorem 2.3.

Isotropy and a One-step Lift Inequality

We use the notion of isotropy from Joudaki et al. [ [26]1, together with the one-step lift developed in Meterez et al. [ [27], to study the dynamics of linear neural networks with normalization across layers. Let X∈Rk×kX \in\mathbb{R}^{k \times k} and assume XTX≻0X^{\mathsf{T}}X \succ0.

Definition B.1 (Isotropy functional). Define

I(X):=det⁡(XTX)1/ktr⁡(XTX)/k∈(0,1].I(X) := \frac{\det(X^{\mathsf{T}}X)^{1/k}}{\operatorname{tr}(X^{\mathsf{T}}X)/k} \in(0,1].

The quantity I(X)I(X) is scale-invariant and satisfies I(X)=1I(X) = 1 if and only if XTXX^{\mathsf{T}}X is a scalar multiple of the identity. In this case, we say that we have “perfect isotropy”. If X=BN⁡(⋅)X = \operatorname{BN}(\cdot) (so that tr⁡(XTX)=k\operatorname{tr}(X^{\mathsf{T}}X) = k), then I(X)=det⁡(XTX)1/kI(X) = \det(X^{\mathsf{T}}X)^{1/k}.

Definition B.2 (Spectral deviation). Suppose X=BN⁡(⋅)X = \operatorname{BN}(\cdot) and let λ1,…,λk\lambda_1,\ldots,\lambda_k be the eigenvalues of XTXX^{\mathsf{T}}X (so ∑i=1kλi=tr⁡(XTX)=k\sum_{i=1}^{k}\lambda_i = \operatorname{tr}(X^{\mathsf{T}}X) = k). Define

S(X):=12k2(k+2)∑i=1k(λi−1)2.S(X) := \frac{1}{2k^2(k+2)}\sum_{i=1}^{k}(\lambda_i-1)^2.

Theorem B.3 (One-step isotropy lift [27 Theorem A.4]). Let X=BN⁡(⋅)X = \operatorname{BN}(\cdot) and let WW be Haar-uniform on O(k)O(k). Then

EW[I(BN⁡(XW))∣X]≥11−S(X)I(X).\mathbb{E}_{W}[I(\operatorname{BN}(XW)) \mid X] \geq\frac{1}{1-S(X)}I(X).

A Uniform Multiplicative Lift Below an Isotropy Cap

Fix η∈(0,1)\eta\in(0,1) and define the compact set of eigenvalue vectors

Kη:={λ∈[0,∞)k: ∑i=1kλi=k, (∏i=1kλi)1/k≤1−η}.K_{\eta}:=\left\{\lambda\in[0,\infty)^k:\ \sum_{i=1}^{k}\lambda_i=k,\ \left(\prod_{i=1}^{k}\lambda_i\right)^{1/k}\leq1-\eta\right\}.

Consider S(λ):=12k2(k+2)∑i=1k(λi−1)2S(\lambda):=\frac{1}{2k^2(k+2)}\sum_{i=1}^{k}(\lambda_i-1)^2 from Definition B.2.

We now prove a uniform spectral gap away from I(X)=1I(X)=1.

Lemma B.4. Let η∈(0,1)\eta\in(0,1) and set sk,η:=min⁡λ∈KηS(λ)s_{k,\eta}:=\min_{\lambda\in K_{\eta}}S(\lambda). Then sk,η>0s_{k,\eta}>0. Moreover, if X=BN⁡(⋅)X=\operatorname{BN}(\cdot) and I(X)≤1−ηI(X)\leq1-\eta, then S(X)≥sk,ηS(X)\geq s_{k,\eta}.

Proof. KηK_{\eta} is nonempty, closed, and bounded, hence compact, and SS is continuous, so the minimum is attained. Also S(λ)=0S(\lambda)=0 iff λ1=⋯=λk=1\lambda_1=\cdots=\lambda_k=1, but that vector is not in KηK_{\eta} since its geometric mean is 1>1−η1>1-\eta. Thus sk,η>0s_{k,\eta}>0. The final statement follows because S(X)=S(λ(XTX))S(X)=S(\lambda(X^{\mathsf{T}}X)) and, for X=BN⁡(⋅)X=\operatorname{BN}(\cdot), I(X)=(∏i=1kλi)1/kI(X)=\left(\prod_{i=1}^{k}\lambda_i\right)^{1/k}.

In turn, we can convert this result into a deterministic multiplicative lift for the isotropy functional I(X)I(X), as long as it is below 1−η1-\eta.

Corollary B.5. Fix η∈(0,1)\eta\in(0,1) and define

rk,η:=11−sk,η>1.r_{k,\eta}:=\frac{1}{1-s_{k,\eta}}>1.

If X=BN⁡(⋅)X=\operatorname{BN}(\cdot) and I(X)≤1−ηI(X)\leq1-\eta, then there exists an orthogonal matrix W∈O(k)W\in O(k) such that

I(BN⁡(XW))≥rk,ηI(X).I(\operatorname{BN}(XW))\geq r_{k,\eta}I(X).

Proof. By Lemma B.4, S(X)≥sk,ηS(X)\geq s_{k,\eta}. Theorem B.3 then gives EW[I(BN⁡(XW))∣X]≥rk,ηI(X)\mathbb{E}_{W}[I(\operatorname{BN}(XW))\mid X]\geq r_{k,\eta}I(X). Since I(BN⁡(XW))≥0I(\operatorname{BN}(XW))\geq0, there must exist a realization of WW achieving at least the expectation.

Consequently, we can show that the isotropy functional I(X)I(X) increases geometrically until XX reaches near-isotropy. In the proof, we choose the weight matrices greedily according to Corollary B.5.

Corollary B.6. Fix η∈(0,1)\eta\in(0,1) and let rk,η>1r_{k,\eta}>1 be as above. Let X0=BN⁡(⋅)X_0=\operatorname{BN}(\cdot) and choose W1,…,Wm∈O⁡(k)W_1,\ldots,W_m\in\operatorname{O}(k) greedily as follows: for each t≥1t\ge1, if I(Xt−1)≤1−ηI(X_{t-1})\le1-\eta choose WtW_t satisfying Corollary B.5, and set Xt:=BN⁡(Xt−1Wt)X_t:=\operatorname{BN}(X_{t-1}W_t). Then for every t≥0t\ge0,

I(Xt)≥min⁡{rk,ηtI(X0),1−η}.I(X_t)\ge\min\left\{r_{k,\eta}^{t}I(X_0),1-\eta\right\}.

Proof. If I(Xt−1)≤1−ηI(X_{t-1})\le1-\eta, then by construction I(Xt)≥rk,ηI(Xt−1)I(X_t)\ge r_{k,\eta}I(X_{t-1}). Once I(Xt−1)>1−ηI(X_{t-1})>1-\eta we do not require any further increase. An induction yields the claimed bound.

Near-isotropy Forces a Floor on λmin⁡\lambda_{\min}

A useful basic result is the inequality that connects the minimum eigenvalue in the spectrum of X⊤XX^\top X to the isotropy functional I(X)I(X).

Lemma B.7. Let X=BN⁡(⋅)X=\operatorname{BN}(\cdot) and let λ1≥⋯≥λk\lambda_1\ge\cdots\ge\lambda_k be the eigenvalues of X⊤XX^\top X (so ∑i=1kλi=k\sum_{i=1}^{k}\lambda_i=k and ∏i=1kλi=I(X)k\prod_{i=1}^{k}\lambda_i=I(X)^k). Then

λmin⁡(X⊤X)=λk≥ckI(X)k,ck:=(k−1k)k−1.\lambda_{\min}(X^\top X)=\lambda_k\ge c_k I(X)^k,\qquad c_k:=\left(\frac{k-1}{k}\right)^{k-1}.

Proof. Set β:=λk\beta:=\lambda_k and note ∑i=1k−1λi=k−β\sum_{i=1}^{k-1}\lambda_i=k-\beta. By AM–GM, for fixed sum k−βk-\beta the product of the first k−1k-1 eigenvalues is maximized when they are equal, hence

∏i=1k−1λi≤(k−βk−1)k−1≤(kk−1)k−1.\prod_{i=1}^{k-1}\lambda_i\le\left(\frac{k-\beta}{k-1}\right)^{k-1}\le\left(\frac{k}{k-1}\right)^{k-1}.

Therefore

I(X)k=∏i=1kλi=β∏i=1k−1λi≤β(kk−1)k−1,I(X)^k=\prod_{i=1}^{k}\lambda_i=\beta\prod_{i=1}^{k-1}\lambda_i\le\beta\left(\frac{k}{k-1}\right)^{k-1},

which rearranges to β≥((k−1)/k)k−1I(X)k\beta\ge((k-1)/k)^{k-1}I(X)^k.

Distance to a Rank-deficient Trajectory

We next use a simple geometric fact that turns the rank mismatch between the two trajectories into a quantitative separation. In our construction, one trajectory becomes well-conditioned, while the other remains rank-deficient throughout the network. The following lemma shows that any rank-deficient matrix must stay at least a smallest-singular-value distance away from a well-conditioned full-rank matrix.

Lemma B.8. Let X,Y∈Rk×kX,Y\in\mathbb{R}^{k\times k} and suppose that rank⁡(Y)<k\operatorname{rank}(Y)<k. Then

∥X−Y∥F2≥λmin⁡(X⊤X).\lVert X-Y\rVert_F^2\ge\lambda_{\min}(X^\top X).

Proof. Since rank⁡(Y)<k\operatorname{rank}(Y)<k there exists a unit vector v∈ker⁡(Y)v\in\ker(Y). Then

∥X−Y∥F2=tr⁡((X−Y)⊤(X−Y))≥v⊤(X−Y)⊤(X−Y)v=∥(X−Y)v∥22=∥Xv∥22.\lVert X-Y\rVert_F^2=\operatorname{tr}\left((X-Y)^\top(X-Y)\right)\ge v^\top(X-Y)^\top(X-Y)v=\lVert(X-Y)v\rVert_2^2=\lVert Xv\rVert_2^2.

Finally, ∥Xv∥22=v⊤X⊤Xv≥λmin⁡(X⊤X)\lVert Xv\rVert_2^2=v^\top X^\top Xv\ge\lambda_{\min}(X^\top X).

Exponential growth of Lip⁡X(Fm)\operatorname{Lip}_X(F_m) with depth

Theorem 2.3 (Exponential lower bound on Lip⁡X(Fm)\operatorname{Lip}_X(F_m)). Fix k≥2k\ge2 and constant η∈(0,1)\eta\in(0,1). There exist constants Ck,η>0C_{k,\eta}>0 and γk,η>0\gamma_{k,\eta}>0 such that, for every depth m∈Nm\in\mathbb{N}, one can choose orthogonal weights W1,…,Wm∈O⁡(k)W_1,\ldots,W_m\in\operatorname{O}(k) with ∥Wℓ∥2=1\lVert W_\ell\rVert_2=1 for all ℓ∈[m]\ell\in[m], for which the depth-mm map FmF_m defined in (2)–(3) satisfies

Lip⁡X(Fm)≥Ck,ηeγk,ηm,\operatorname{Lip}_X(F_m)\ge C_{k,\eta}e^{\gamma_{k,\eta}m},

where constants Ck,ηC_{k,\eta} and γk,η\gamma_{k,\eta} depend on k,ηk,\eta. Moreover, for this same fixed choice of weights, the lower bound is witnessed by an uncountable family of input pairs.

Proof. Fix m∈Nm \in\mathbb{N} and let r:=rk,η>1r := r_{k,\eta} > 1 from Corollary B.5. Define

δk,η:=ck(1−η)k,γk,η:=k2(k−1)log⁡r,Ck,η:=δk,η(1−η)−k2(k−1).\delta_{k,\eta} := c_k(1-\eta)^k,\qquad\gamma_{k,\eta} := \frac{k}{2(k-1)}\log r,\qquad C_{k,\eta} := \sqrt{\delta_{k,\eta}}(1-\eta)^{-\frac{k}{2(k-1)}}.

Step 1: choose an initialization X0X_0 with controlled anisotropy. Let

Gε:=(1−ε)11T+εI∈Rk×k,ε∈(0,1].G_{\varepsilon} := (1-\varepsilon)\mathbf{1}\mathbf{1}^{\mathsf{T}}+\varepsilon I \in\mathbb{R}^{k\times k},\qquad\varepsilon\in(0,1].

Then diag⁡(Gε)=I\operatorname{diag}(G_{\varepsilon})=I and the eigenvalues of GεG_{\varepsilon} are λ1=k−(k−1)ε\lambda_1=k-(k-1)\varepsilon and λ2=⋯=λk=ε\lambda_2=\cdots=\lambda_k=\varepsilon. Choose any X0∈Rp×kX_0\in\mathbb{R}^{p\times k} such that X0TX0=GεX_0^{\mathsf{T}}X_0=G_{\varepsilon} (e.g. take X0=LTX_0=L^{\mathsf{T}} where Gε=LLTG_{\varepsilon}=LL^{\mathsf{T}} is a Cholesky factorization). Since diag⁡(Gε)=I\operatorname{diag}(G_{\varepsilon})=I, every column of X0X_0 has norm 11, hence X0=BN⁡(X0)X_0=\operatorname{BN}(X_0).

Moreover,

I(X0)=det⁡(Gε)1/k=(εk−1(k−(k−1)ε))1/k≥εk−1k,I(X_0)=\det(G_{\varepsilon})^{1/k}=\left(\varepsilon^{k-1}(k-(k-1)\varepsilon)\right)^{1/k}\geq\varepsilon^{\frac{k-1}{k}},

because k−(k−1)ε≥1k-(k-1)\varepsilon\geq1 for ε∈(0,1]\varepsilon\in(0,1].

Step 2: pick ε\varepsilon so that mm greedy steps reach near-isotropy. Choose

ε:=(1−η)kk−1r−mkk−1.\varepsilon:= (1-\eta)^{\frac{k}{k-1}}r^{-\frac{mk}{k-1}}.

Then rmI(X0)≥rmε(k−1)/k=1−ηr^m I(X_0)\geq r^m\varepsilon^{(k-1)/k}=1-\eta.

Now choose W1,…,WmW_1,\ldots,W_m greedily as in Corollary B.6 and set Xt:=BN⁡(Xt−1Wt)X_t:=\operatorname{BN}(X_{t-1}W_t). By Corollary B.6, I(Xm)≥1−ηI(X_m)\geq1-\eta.

Step 3: eigenvalue floor after reaching near-isotropy. Lemma B.7 yields

λmin⁡(XmTXm)≥ckI(Xm)k≥ck(1−η)k=δk,η.\lambda_{\min}(X_m^{\mathsf{T}}X_m)\geq c_k I(X_m)^k\geq c_k(1-\eta)^k=\delta_{k,\eta}.

Step 4: a rank-deficient companion Y0Y_0 and distance growth. Let X0=UΣVTX_0=U\Sigma V^{\mathsf{T}} be an SVD with singular values σ1≥⋯≥σk>0\sigma_1\geq\cdots\geq\sigma_k>0. Define the rank-(k−1)(k-1) matrix

Y0:=Udiag⁡(σ1,…,σk−1,0)VT.Y_0:=U\operatorname{diag}(\sigma_1,\ldots,\sigma_{k-1},0)V^{\mathsf{T}}.

Then rank⁡(Y0)<k\operatorname{rank}(Y_0)<k and

∥X0−Y0∥F=σk=λmin⁡(X0TX0)=ε.\lVert X_0-Y_0\rVert_F=\sigma_k=\sqrt{\lambda_{\min}(X_0^{\mathsf{T}}X_0)}=\sqrt{\varepsilon}.

Let Yt:=Ft(Y0)Y_t:=F_t(Y_0). Since each layer is multiplication by an invertible matrix (WtW_t) followed by multiplication by a diagonal matrix (coming from BN), the rank cannot increase, hence rank⁡(Yt)<k\operatorname{rank}(Y_t)<k for all tt.

Applying Lemma B.8 at depth mm gives

∥Xm−Ym∥F2≥λmin⁡(XmTXm)≥δk,η.\lVert X_m-Y_m\rVert_F^2\geq\lambda_{\min}(X_m^{\mathsf{T}}X_m)\geq\delta_{k,\eta}.

Therefore

Lip⁡X(Fm)≥∥Fm(X0)−Fm(Y0)∥F∥X0−Y0∥F≥δk,ηε=Ck,ηrmk2(k−1)=Ck,ηeγk,ηm.\operatorname{Lip}_{\mathcal{X}}(F_m)\geq\frac{\lVert F_m(X_0)-F_m(Y_0)\rVert_F}{\lVert X_0-Y_0\rVert_F}\geq\sqrt{\frac{\delta_{k,\eta}}{\varepsilon}}=C_{k,\eta}r^{\frac{mk}{2(k-1)}}=C_{k,\eta}e^{\gamma_{k,\eta}m}.

Uncountably many witness pairs for the same weights. For any orthogonal Q∈O(k)Q\in O(k), we have for all XX and all orthogonal WW,

BN⁡((QX)W)=QBN⁡(XW),\operatorname{BN}((QX)W)=Q\operatorname{BN}(XW),

because ((QX)W)T((QX)W)=WTXTQTQXW=WTXTXW((QX)W)^{\mathsf{T}}((QX)W)=W^{\mathsf{T}}X^{\mathsf{T}}Q^{\mathsf{T}}QXW=W^{\mathsf{T}}X^{\mathsf{T}}XW. Hence Fm(QX)=QFm(X)F_m(QX)=QF_m(X) by induction. Consequently, the ratio ∥Fm(QX0)−Fm(QY0)∥F/∥QX0−QY0∥F\lVert F_m(QX_0)-F_m(QY_0)\rVert_F/\lVert QX_0-QY_0\rVert_F equals the ratio for (X0,Y0)(X_0,Y_0) for every Q∈O(k)Q\in O(k). Since O(k)O(k) is uncountable, this yields uncountably many witness pairs.

Corollary 2.4. Fix η∈(0,1)\eta\in(0,1) and set k=2k=2. There exist constants Cη>0C_\eta>0 and Γη>0\Gamma_\eta>0 such that, for every integer d≥4d\ge4, there is a depth-mm network with at most dd scalar weight parameters whose input–output map satisfies

Lip⁡X(Fm)≥CηeΓηd.\operatorname{Lip}_{\mathcal{X}}(F_m) \ge C_\eta e^{\Gamma_\eta d}.

Proof. When k=2k=2, each layer contains 4 scalar parameters if the weight matrix is counted as an unconstrained 2×22 \times2 matrix. Thus, a depth-mm network has 4m4m scalar parameters. Given a budget d≥4d\ge4, choose m=⌊d/4⌋m=\lfloor d/4\rfloor, so that 4m≤d4m\le d. Applying Theorem 2.3 with k=2k=2 gives

Lip⁡X(Fm)≥C2,ηeγ2,η⌊d/4⌋≥C2,ηe−γ2,ηe(γ2,η/4)d.\operatorname{Lip}_{\mathcal{X}}(F_m) \ge C_{2,\eta}e^{\gamma_{2,\eta}\lfloor d/4\rfloor} \ge C_{2,\eta}e^{-\gamma_{2,\eta}}e^{(\gamma_{2,\eta}/4)d}.

The claim follows by setting

Cη:=C2,ηe−γ2,η,Γη:=γ2,η/4.C_\eta:= C_{2,\eta}e^{-\gamma_{2,\eta}}, \qquad\Gamma_\eta:= \gamma_{2,\eta}/4.

Exponential growth of Lip⁡Θ(Fm)\operatorname{Lip}_{\Theta}(F_m) with depth

Theorem 2.6 (Exponential lower bound on Lip⁡Θ(Fm)\operatorname{Lip}_{\Theta}(F_m)). Fix k≥2k\ge2 and constant η∈(0,1)\eta\in(0,1). There exist constants C~k,η>0\widetilde{C}_{k,\eta}>0 and γ~k,η>0\widetilde{\gamma}_{k,\eta}>0 depending on kk and η\eta such that, for every depth m≥2m\ge2, there is a depth-mm network of the form (2), with ∥Wℓ∥2≤1\lVert W_\ell\rVert_2\le1 for all ℓ∈[m]\ell\in[m], for which the fixed-input parameter-to-output map Fm(θ):=Fmθ(Xin)F_m(\theta):=F_m^\theta(X_{\mathrm{in}}) with Xin=IkX_{\mathrm{in}}=I_k satisfies

Lip⁡Θ(Fm)≥C~k,ηeγ~k,ηm.\operatorname{Lip}_{\Theta}(F_m) \ge\widetilde{C}_{k,\eta}e^{\widetilde{\gamma}_{k,\eta}m}.

Proof. Apply Theorem 2.3 with depth m−1m-1 to obtain orthogonal matrices W~2,…,W~m∈O(k)\widetilde{W}_2,\ldots,\widetilde{W}_m\in O(k) and matrices X0,Y0∈Rk×kX_0,Y_0\in\mathbb{R}^{k\times k} with rank⁡(Y0)<k\operatorname{rank}(Y_0)<k such that the corresponding depth-(m−1)(m-1) map

Gm−1(Z):=BN⁡(⋯BN⁡(BN⁡(ZW~2)W~3)⋯W~m)G_{m-1}(Z):=\operatorname{BN}(\cdots\operatorname{BN}(\operatorname{BN}(Z\widetilde{W}_2)\widetilde{W}_3)\cdots\widetilde{W}_m)

satisfies

∥Gm−1(X0)−Gm−1(Y0)∥F≥δk,η,∥X0−Y0∥F=ε,\lVert G_{m-1}(X_0)-G_{m-1}(Y_0)\rVert_F\ge\sqrt{\delta_{k,\eta}}, \qquad\lVert X_0-Y_0\rVert_F=\sqrt{\varepsilon},

where ε\varepsilon is exponentially small in mm as in the proof of Theorem 2.3.

Now consider a depth-mm network with input Xin=IkX_{\mathrm{in}}=I_k and weights

W1:=αX0,W1′:=αY0,Wℓ:=W~ℓ(ℓ=2,…,m),W_1:=\alpha X_0,\qquad W'_1:=\alpha Y_0,\qquad W_\ell:=\widetilde{W}_\ell\quad(\ell=2,\ldots,m),

where α>0\alpha>0 is chosen so that ∥W1∥2,∥W1′∥2≤1\lVert W_1\rVert_2,\lVert W'_1\rVert_2\le1. Since column-wise batch normalization is scale-invariant,

BN⁡(αZ)=αZ(D(αZ))−1/2=αZ(α2D(Z))−1/2=BN⁡(Z),\operatorname{BN}(\alpha Z)=\alpha Z\left(D(\alpha Z)\right)^{-1/2}=\alpha Z\left(\alpha^2D(Z)\right)^{-1/2}=\operatorname{BN}(Z),

we have BN⁡(XinW1)=BN⁡(X0)\operatorname{BN}(X_{\mathrm{in}}W_1)=\operatorname{BN}(X_0) and BN⁡(XinW1′)=BN⁡(Y0)\operatorname{BN}(X_{\mathrm{in}}W'_1)=\operatorname{BN}(Y_0). Moreover, rank⁡(BN⁡(Y0))<k\operatorname{rank}(\operatorname{BN}(Y_0))<k.

Let θ=(W1,…,Wm)\theta=(W_1,\ldots,W_m) and θ′=(W1′,W2,…,Wm)\theta'=(W'_1,W_2,\ldots,W_m). Then

Fm(θ)=Gm−1(BN⁡(X0))andFm(θ′)=Gm−1(BN⁡(Y0)).F_m(\theta)=G_{m-1}(\operatorname{BN}(X_0)) \qquad\text{and}\qquad F_m(\theta')=G_{m-1}(\operatorname{BN}(Y_0)).

The second trajectory remains rank-deficient at every depth, so Lemma B.8 (applied at depth m−1m-1) and the conditioning lower bound from the proof of Theorem 2.3 yield

∥Fm(θ)−Fm(θ′)∥F≥δk,η.\lVert F_m(\theta)-F_m(\theta')\rVert_F\ge\sqrt{\delta_{k,\eta}}.

On the other hand,

∥θ−θ′∥2=∥W1−W1′∥F=α∥X0−Y0∥F=αε.\lVert\theta-\theta'\rVert_2=\lVert W_1-W'_1\rVert_F=\alpha\lVert X_0-Y_0\rVert_F=\alpha\sqrt{\varepsilon}.

Therefore

Lip⁡Θ(Fm)≥∥Fm(θ)−Fm(θ′)∥F∥θ−θ′∥2≥δk,ηαε=C~k,ηeγ~k,ηm,\operatorname{Lip}_{\Theta}(F_m)\ge\frac{\lVert F_m(\theta)-F_m(\theta')\rVert_F}{\lVert\theta-\theta'\rVert_2}\ge\frac{\sqrt{\delta_{k,\eta}}}{\alpha\sqrt{\varepsilon}}=\widetilde{C}_{k,\eta}e^{\widetilde{\gamma}_{k,\eta}m},

since ε\varepsilon is exponentially small in mm (same as in Theorem 2.3).

Corollary 2.8. Let FmF_m be the parameter-to-output map constructed in Theorem 2.6. There exists a realizable target parameter θ⋆∈Θ\theta^\star\in\Theta such that the objective ff defined in (4) satisfies

Lip⁡(f)≥C~ηeΓ~ηd,\operatorname{Lip}(f) \ge\widetilde{C}_\eta e^{\widetilde{\Gamma}_\eta d},

up to the constants in Corollary 2.7.

Proof. In the construction of Theorem 2.6, let θ⋆\theta^\star be one of the two parameter choices witnessing the lower bound on Lip⁡Θ(Fm)\operatorname{Lip}_{\Theta}(F_m), and let θ\theta be the other one. Since the target is realizable, f(θ⋆)=0f(\theta^\star)=0. Therefore,

Lip⁡(f)≥∣f(θ)−f(θ⋆)∣∥θ−θ⋆∥2=∣f(θ)∣∥θ−θ⋆∥2=∥Fm(θ)−Fm(θ⋆)∥F∥θ−θ⋆∥2.\operatorname{Lip}(f) \ge\frac{\lvert f(\theta)-f(\theta^\star)\rvert}{\lVert\theta-\theta^\star\rVert_2} = \frac{\lvert f(\theta)\rvert}{\lVert\theta-\theta^\star\rVert_2} = \frac{\lVert F_m(\theta)-F_m(\theta^\star)\rVert_F}{\lVert\theta-\theta^\star\rVert_2}.

The right-hand side is exactly the same separation ratio used in the proof of Theorem 2.6. Applying Corollary 2.7 gives the claimed exponential lower bound. □\square

Extension to Normalization with Mean Subtraction

The results of Sections 2.2 and 2.3 extend to batch normalization with mean subtraction. Under our notation in § 2, the centered normalization map is defined, for X∈Rn×kX \in\mathbb{R}^{n \times k}, by

BN⁡ctr(X):=BN⁡((In−1n11T)X),\operatorname{BN}_{\mathrm{ctr}}(X) := \operatorname{BN}\left(\left(I_n-\frac{1}{n}\mathbf{1}\mathbf{1}^{\mathsf T}\right)X\right),

whenever every centered column is nonzero. This corresponds to the standard batch-variance normalization with a fixed affine scale of 1/n1/\sqrt{n}.

We now try to embed our hard instances into the centered subspace. Suppose n≥k+1n \ge k+1, and choose U∈Rn×kU \in\mathbb{R}^{n \times k} satisfying

UTU=Ik,1TU=0.U^{\mathsf T}U=I_k,\qquad\mathbf{1}^{\mathsf T}U=0.

Such a matrix exists since the orthogonal complement of 1\mathbf{1} in Rn\mathbb{R}^n has dimension n−1≥kn-1 \ge k. Left multiplication by UU is an isometry and maps every matrix into the subspace of column-centered matrices. In particular, for every Z∈Rk×kZ \in\mathbb{R}^{k \times k},

(UZ)T(UZ)=ZTZ,(UZ)^{\mathsf T}(UZ)=Z^{\mathsf T}Z,

so the embedding preserves rank, Gram eigenvalues, column norms, and Frobenius distances.

Moreover, since 1TU=0\mathbf{1}^{\mathsf T}U=0, mean subtraction leaves UZWUZW unchanged. Consequently, for every ZZ and WW for which normalization is defined,

BN⁡ctr(UZW)=UZWD(ZW)−1/2=UBN⁡(ZW).(5)\operatorname{BN}_{\mathrm{ctr}}(UZW)=UZW D(ZW)^{-1/2}=U\operatorname{BN}(ZW). \tag*{(5)}

Thus, centered normalization reproduces the original normalization dynamics exactly on the embedded subspace.

Let F‾m\overline{F}_m denote the depth-mm network obtained by replacing each normalization layer of FmF_m with BN⁡ctr\operatorname{BN}_{\mathrm{ctr}}, while keeping the same weight matrices. Applying (5) inductively yields

F‾m(UZ)=UFm(Z).\overline{F}_m(UZ)=UF_m(Z).

In particular, applying this identity to the witness pair (X0,Y0)(X_0,Y_0) from Theorem 2.3 gives

∥F‾m(UX0)−F‾m(UY0)∥F∥UX0−UY0∥F=∥Fm(X0)−Fm(Y0)∥F∥X0−Y0∥F.\frac{\lVert\overline{F}_m(UX_0)-\overline{F}_m(UY_0)\rVert_F}{\lVert UX_0-UY_0\rVert_F} = \frac{\lVert F_m(X_0)-F_m(Y_0)\rVert_F}{\lVert X_0-Y_0\rVert_F}.

Hence the exponential input Lipschitz lower bound carries over with exactly the same initial perturbation size and final output separation. The Gram-eigenvalue bounds underlying the construction are also preserved by the embedding.

The parameter Lipschitz lower bound follows from the same identity. Indeed, replacing the fixed input IkI_k in Theorem 2.6 with UU, and using the same two weight tuples, preserves both their parameter distance and the corresponding output separation. Since the weight matrices remain k×kk \times k, the parameter count is unchanged.

Finally, the robustness construction of Theorem 4.1 extends directly to the centered network. Using the readout UAmUA_m and the same offset bmb_m, we have ∥UAm∥F=1\lVert UA_m\rVert_F = 1 and

⟨UAm,F~m(UZ)⟩F+bm=⟨Am,Fm(Z)⟩F+bm.\langle UA_m, \widetilde{F}_m(UZ)\rangle_F + b_m = \langle A_m, F_m(Z)\rangle_F + b_m.

Thus, the embedded clean and adversarial inputs achieve exactly the same classification margins and perturbation distance as in Theorem 4.1. In particular, the exponential input and parameter sensitivity bounds, as well as the exponentially small adversarial radius, hold for mean-subtracted batch normalization.

Extension to ReLU Activations

Our construction in Appendix § 8.8 also extends to networks with centered batch normalization followed by ReLU activations. We use the standard affine shift of batch normalization, fixed to 2 in every channel, and define

f~W(H):=ReLU⁡(BN⁡ctr(HW)+211T),\widetilde{f}_W(H) := \operatorname{ReLU}\left(\operatorname{BN}_{\mathrm{ctr}}(HW) + 2\mathbf{1}\mathbf{1}^{\mathsf{T}}\right),

where ReLU acts entrywise and 11T∈Rn×k\mathbf{1}\mathbf{1}^{\mathsf{T}} \in\mathbb{R}^{n \times k} denotes the all-ones matrix. Due to normalization layers, each column of BN⁡ctr(HW)\operatorname{BN}_{\mathrm{ctr}}(HW) has unit Euclidean norm, so every entry lies in [−1,1][-1,1]. Consequently, the affine shift makes every entry strictly positive, and ReLU acts as the identity.

The key observation is that mean subtraction eliminates any matrix with identical rows, even after multiplication by a weight matrix. Combining this property with (5), we obtain

f~W(UZ)=UfW(Z)+211T,\widetilde{f}_W(UZ) = U f_W(Z) + 2\mathbf{1}\mathbf{1}^{\mathsf{T}},
f~W(UZ+211T)=UfW(Z)+211T.\widetilde{f}_W(UZ + 2\mathbf{1}\mathbf{1}^{\mathsf{T}}) = U f_W(Z) + 2\mathbf{1}\mathbf{1}^{\mathsf{T}}.

Thus, the affine shift introduced by one layer is removed by the centering operation in the next layer, while the centered component follows exactly the dynamics of the original network.

Writing F~m=f~Wm∘⋯∘f~W1\widetilde{F}_m = \widetilde{f}_{W_m} \circ\cdots\circ\widetilde{f}_{W_1}, induction yields

F~m(UZ)=UFm(Z)+211T.\widetilde{F}_m(UZ) = UF_m(Z) + 2\mathbf{1}\mathbf{1}^{\mathsf{T}}.

In particular, for the witness pair (X0,Y0)(X_0,Y_0),

F~m(UX0)−F~m(UY0)=U(Fm(X0)−Fm(Y0)).\widetilde{F}_m(UX_0) - \widetilde{F}_m(UY_0) = U\left(F_m(X_0) - F_m(Y_0)\right).

Since UTU=IkU^{\mathsf{T}}U = I_k, the input and output distances are preserved exactly. The exponential input Lipschitz lower bound therefore holds with the same witness ratio. Likewise, using the fixed input UU and the weight tuples from Theorem 2.6 preserves their parameter distance and output separation, yielding the same parameter Lipschitz lower bound. All affine shifts are fixed, so the number of free weight parameters is unchanged.

Finally, the robustness construction transfers under the same readout UAmUA_m and offset bmb_m. Indeed, since 1TU=0\mathbf{1}^{\mathsf{T}}U = 0,

⟨UAm,F~m(UZ)⟩F+bm=⟨UAm,UFm(Z)+211T⟩F+bm=⟨Am,Fm(Z)⟩F+bm.\langle UA_m,\widetilde{F}_m(UZ)\rangle_F + b_m = \langle UA_m,UF_m(Z) + 2\mathbf{1}\mathbf{1}^{\mathsf{T}}\rangle_F + b_m = \langle A_m,F_m(Z)\rangle_F + b_m.

Hence the clean and adversarial inputs keep the same score margins and perturbation distance as in Theorem 4.1.

We conclude that the exponential input and parameter Lipschitz lower bounds, together with the exponentially small adversarial radius, are also attained by networks with centered batch normalization and ReLU activations.

Embedding the Construction at Larger Widths

We show that the exponential lower bounds can be achieved with constants independent of width by embedding the width-two construction into networks of arbitrary width k≥2k \ge2.

Let X0,Y0X_0,Y_0 and W1,…,WmW_1,\ldots,W_m be the witnesses from Theorem 2.3 for k=2k=2. For any k≥2k\geq2, consider the block-diagonal matrices

diag⁡(X0,Ik−2),diag⁡(Y0,Ik−2),diag⁡(Wℓ,Ik−2),ℓ∈[m],\operatorname{diag}(X_0,I_{k-2}),\qquad\operatorname{diag}(Y_0,I_{k-2}),\qquad\operatorname{diag}(W_\ell,I_{k-2}),\quad\ell\in[m],

where the identity blocks are omitted when k=2k=2. Since normalization acts columnwise and the additional columns have unit norm, the two trajectories at depth tt are precisely diag⁡(Xt,Ik−2)\operatorname{diag}(X_t,I_{k-2}) and diag⁡(Yt,Ik−2)\operatorname{diag}(Y_t,I_{k-2}). Their input distance and final output separation are therefore identical to those of the width-two construction. Consequently, the input Lipschitz lower bound

Lip⁡X(Fm)≥C2,ηeγ2,ηm.\operatorname{Lip}_{X}(F_m)\geq C_{2,\eta}e^{\gamma_{2,\eta}m}.

holds for every k≥2k\geq2, with constants independent of width.

Similarly, applying the same embedding to the parameter witnesses in Theorem 2.6, with fixed input IkI_k, preserves their parameter distance, output separation, and unit operator norms. Thus, the parameter Lipschitz lower bound also holds uniformly over width, with the constants of the width-two construction.

These bounds have width-independent exponential rates in depth. When all entries of the weight matrices are counted, however, the parameter dimension is d=mk2d=mk^2, and the input lower bound becomes

Lip⁡X(Fm)≥C2,ηexp⁡(γ2,ηk2d).\operatorname{Lip}_{X}(F_m)\geq C_{2,\eta}\exp\left(\frac{\gamma_{2,\eta}}{k^2}d\right).

Proofs and Details for Section 3

Background on Goldstein Stationarity and First-order Oracle

Notation and Background. Let f:Rd→Rf:\mathbb{R}^d\to\mathbb{R} be (locally) LL-Lipschitz with respect to ∥⋅∥2\|\cdot\|_2:

∣f(x)−f(y)∣≤L∥x−y∥2,∀x,y∈Rd.|f(x)-f(y)|\leq L\|x-y\|_2,\qquad\forall x,y\in\mathbb{R}^d.

Let ∂f(x)\partial f(x) denote the Clarke subdifferential at xx, which coincides with {∇f(x)}\{\nabla f(x)\} when ff is differentiable at xx and agrees with the usual convex subdifferential when ff is convex. The canonical solution concept studied in nonsmooth nonconvex optimization is that of a Clarke stationary point [44]. A point x∈Rdx\in\mathbb{R}^d is ϵ\epsilon-Clarke stationary if dist⁡(0,∂f(x))≤ϵ\operatorname{dist}(0,\partial f(x))\leq\epsilon, where

dist⁡(0,Q):=min⁡g∈∂f(x)∥g∥2.\operatorname{dist}(0,Q):=\min_{g\in\partial f(x)}\|g\|_2.

Zhang et al. [ [13] showed that for general nonconvex Lipschitz functions, Clarke stationarity cannot be guaranteed in a finite number of oracle queries in the worst case, even under very strong first-order oracle access. This result shows that the consideration of more relaxed solution concepts, in favor of tractability, is inevitable.

Goldstein Stationarity. A classical and now-standard relaxation is due to Goldstein [ [12]. Fix δ>0\delta>0. The key idea is to convexify the subdifferential over a neighborhood:

∂δf(x):=Conv⁡(⋃y: ∥y−x∥2<δ∂f(y)).\partial_\delta f(x):=\operatorname{Conv}\left(\bigcup_{y:\ \|y-x\|_2<\delta}\partial f(y)\right).

We say that xx is (δ,ϵ)(\delta,\epsilon)-Goldstein stationary if

dist⁡(0,∂δf(x))≤ϵ.\operatorname{dist}(0,\partial_\delta f(x))\leq\epsilon.

This definition recovers the smooth optimality condition when ff is differentiable and δ\delta is sufficiently small. The recent breakthrough of Zhang et al. [ [13] highlighted the tractability of this solution concept in nonsmooth settings, and subsequent works have further developed and studied it extensively [14, 35, 45, 15, 36, 46].

Oracle model. We consider the standard first-order oracle model used in the nonsmooth optimization literature. At each query point xtx_t, the oracle reveals the function value f(xt)f(x_t) together with subgradient information at xtx_t (e.g., a Clarke subgradient gt∈∂f(xt)g_t\in\partial f(x_t) or even the full set ∂f(xt)\partial f(x_t)). The lower bounds stated in the paper holds even when the oracle returns the entire set ∂f(xt)\partial f(x_t).

Proofs and Details for Section 3.1

Theorem C.1 ([16 Theorem 3]). For any ε<1\varepsilon< 1 and any δ≤112ε\delta\le\frac{1}{12\varepsilon}, the following holds. For every randomized first-order algorithm A\mathcal{A}, there exists a convex 1-Lipschitz function f:Rd→Rf:\mathbb{R}^{d} \to\mathbb{R}, with d=O~(1/ε6)d = \widetilde{O}(1/\varepsilon^{6}), such that f(x1)−inf⁡xf(x)≤1f(x_{1})-\inf_{x} f(x) \le1, but

Pr⁡A[∃t≤T such that xt is (δ,ε)-Goldstein stationary]<13,\Pr_{\mathcal{A}}\left[\exists t \le T\text{ such that }x_{t}\text{ is }(\delta,\varepsilon)\text{-Goldstein stationary}\right] < \frac{1}{3},

unless T=Ω(1/ε2)T = \Omega(1/\varepsilon^{2}).

We now immediately show that no randomized algorithm can avoid a polynomial dependence on LL.

Corollary 3.1 (Formal). Fix L>0L > 0 and ε∈(0,L)\varepsilon\in(0,L), and suppose that δ≤L12ε\delta\le\frac{L}{12\varepsilon}. Then, for any randomized first-order algorithm A\mathcal{A}, there exists a convex LL-Lipschitz function F:Rd→RF:\mathbb{R}^{d} \to\mathbb{R}, with d=O~((L/ε)6)d = \widetilde{O}((L/\varepsilon)^{6}), such that F(x1)−inf⁡xF(x)≤LF(x_{1})-\inf_{x} F(x) \le L, and

Pr⁡A[∃t≤T such that xt is (δ,ε)-Goldstein stationary]<23\Pr_{\mathcal{A}}\left[\exists t \le T\text{ such that }x_{t}\text{ is }(\delta,\varepsilon)\text{-Goldstein stationary}\right] < \frac{2}{3}

unless

T=Ω(L2ε2).T = \Omega\left(\frac{L^{2}}{\varepsilon^{2}}\right).

In particular, in the worst case, any randomized first-order method needs Ω(L2/ε2)\Omega(L^{2}/\varepsilon^{2}) oracle calls to find a (δ,ε)(\delta,\varepsilon)-Goldstein stationary point with probability at least 2/32/3.

Proof. Set ε′:=ε/L\varepsilon' := \varepsilon/L. Since ε∈(0,L)\varepsilon\in(0,L), we have ε′∈(0,1)\varepsilon' \in(0,1). Applying Theorem C.1 with accuracy ε′\varepsilon' gives a convex 1-Lipschitz function f:Rd→Rf:\mathbb{R}^{d} \to\mathbb{R}, with d=O~(1/ε′6)=O~((L/ε)6)d = \widetilde{O}(1/\varepsilon'^{6}) = \widetilde{O}((L/\varepsilon)^{6}), such that f(x1)−inf⁡xf(x)≤1f(x_{1})-\inf_{x} f(x) \le1, and no randomized first-order algorithm can find a (δ,ε′)(\delta,\varepsilon')-Goldstein stationary point with constant probability unless

T=Ω(1/ε′2).T = \Omega(1/\varepsilon'^{2}).

The condition on δ\delta becomes

δ≤112ε′=L12ε.\delta\le\frac{1}{12\varepsilon'} = \frac{L}{12\varepsilon}.

Now define F:=LfF := Lf. Then FF is convex and LL-Lipschitz, and

F(x1)−inf⁡xF(x)=L(f(x1)−inf⁡xf(x))≤L.F(x_{1})-\inf_{x} F(x) = L\left(f(x_{1})-\inf_{x} f(x)\right) \le L.

Moreover, by positive homogeneity of the Clarke and Goldstein subdifferentials,

∂δF(x)=L∂δf(x),\partial_{\delta}F(x) = L\partial_{\delta}f(x),

and therefore

dist⁡(0,∂δF(x))=Ldist⁡(0,∂δf(x)).\operatorname{dist}(0,\partial_{\delta}F(x)) = L\operatorname{dist}(0,\partial_{\delta}f(x)).

Thus, xx is (δ,ε)(\delta,\varepsilon)-Goldstein stationary for FF if and only if it is (δ,ε/L)(\delta,\varepsilon/L)-Goldstein stationary for ff, i.e., (δ,ε′)(\delta,\varepsilon')-Goldstein stationary for ff.

Consequently, any randomized first-order algorithm that finds a (δ,ε)(\delta,\varepsilon)-Goldstein stationary point of FF with probability at least 2/32/3 would also find a (δ,ε′)(\delta,\varepsilon')-Goldstein stationary point of ff with the same probability. By Theorem C.1, this requires

T=Ω(1/ε′2)=Ω(L2ε2).T = \Omega(1/\varepsilon'^{2}) = \Omega\left(\frac{L^{2}}{\varepsilon^{2}}\right).

This proves the claim.

Remark C.2. The regime in Corollary 3.1 is compatible with our exploding-Lipschitz setting. After rescaling the lower bound from [16], the hard instance has dimension O~((L/ε)6)\widetilde{O}((L/\varepsilon)^{6}) and requires δ≤L/(12ε)\delta\le L/(12\varepsilon). In our deep normalized networks, LL can be Ω(exp⁡(d))\Omega(\exp(d)), so these conditions are mild for fixed or polynomially scaled accuracy parameters. Thus the lower-bound regime overlaps with the regime arising in our construction, and substituting L=Ω(exp⁡(d))L = \Omega(\exp(d)) into Ω(L2/ε2)\Omega(L^{2}/\varepsilon^{2}) yields exponential dependence on the network dimension.

Algorithm 1 Probabilistic Greedy Goldstein Descent (PGGD)

Require: x0∈Rdx_0 \in\mathbb{R}^d, δ>0\delta> 0, τ>0\tau> 0, ρ∈(0,1)\rho\in(0,1), γ∈(0,1)\gamma\in(0,1), flb≤inf⁡ff_{\mathrm{lb}} \le\inf f.

  1. Δ←f(x0)−flb\Delta\gets f(x_0)-f_{\mathrm{lb}}

  2. Tmax⁡←⌈Δτδ⌉T_{\max} \gets\left\lceil\frac{\Delta}{\tau\delta} \right\rceil

  3. m←⌈1ρlog⁡Tmax⁡+1γ⌉m \gets\left\lceil\frac{1}{\rho}\log\frac{T_{\max}+1}{\gamma} \right\rceil

  4. x←x0x \gets x_0

  5. for t=0t=0 to Tmax⁡T_{\max} do

  6. &nbsp;&nbsp;&nbsp;&nbsp;Sample i.i.d. v1,…,vm∼Unif⁡(Sd−1)v_1,\ldots,v_m \sim\operatorname{Unif}(\mathbb{S}^{d-1})

  7. &nbsp;&nbsp;&nbsp;&nbsp;Query f(x−δvi)f(x-\delta v_i) for all ii and let i⋆∈arg⁡min⁡if(x−δvi)i^\star\in\arg\min_i f(x-\delta v_i)

  8. &nbsp;&nbsp;&nbsp;&nbsp;if f(x−δvi⋆)≤f(x)−τδf(x-\delta v_{i^\star}) \le f(x)-\tau\delta then

  9. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;x←x−δvi⋆x \gets x-\delta v_{i^\star}

  10. &nbsp;&nbsp;&nbsp;&nbsp;else

  11. &nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;return xx

  12. &nbsp;&nbsp;&nbsp;&nbsp;end if

  13. end for

  14. return xx

Why exploding LL undermines “dimension-free” rates. The discussion in § 2 shows that the Lipschitz constant cannot be treated as a harmless constant in deep normalized networks. Existing guarantees for finding (δ,ϵ)(\delta,\epsilon)-Goldstein stationary points scale polynomially in the objective Lipschitz constant LL, and Corollary 3.1 shows that some polynomial dependence on LL is unavoidable in the worst case. But in our setting, LL is induced by the network itself. By Theorem 2.6, it can grow exponentially with depth, and hence with the parameter dimension. Therefore, a guarantee of the form poly⁡(L,1/δ,1/ϵ)\operatorname{poly}(L,1/\delta,1/\epsilon) may be formally independent of dimension, but it can still translate into exponential dependence on the dimension for deep normalized networks.

Proofs and Details for Section 3.2

For fixed xx and δ>0\delta>0, recall

mδ(x;v):=min⁡g∈∂δf(x)⟨g,v⟩,v∈Sd−1.m_\delta(x;v) := \min_{g\in\partial_\delta f(x)} \langle g,v\rangle,\qquad v\in\mathbb{S}^{d-1}.

The following elementary identity is the dual form of Goldstein stationarity.

Lemma C.3. For any ϵ≥0\epsilon\ge0,

dist⁡(0,∂δf(x))≤ϵ⟺max⁡v∈Sd−1mδ(x;v)≤ϵ.\operatorname{dist}(0,\partial_\delta f(x))\le\epsilon\Longleftrightarrow\max_{v\in\mathbb{S}^{d-1}}m_\delta(x;v)\le\epsilon.

Moreover, if dist⁡(0,∂δf(x))>0\operatorname{dist}(0,\partial_\delta f(x))>0, then

dist⁡(0,∂δf(x))=max⁡v∈Sd−1mδ(x;v).\operatorname{dist}(0,\partial_\delta f(x))=\max_{v\in\mathbb{S}^{d-1}}m_\delta(x;v).

Proof. Let Q=∂δf(x)Q=\partial_\delta f(x) and let g⋆g^\star be the projection of 00 onto QQ. For any unit vector vv,

mδ(x;v)≤⟨g⋆,v⟩≤∥g⋆∥2.m_\delta(x;v)\le\langle g^\star,v\rangle\le\|g^\star\|_2.

Thus max⁡vmδ(x;v)≤dist⁡(0,Q)\max_v m_\delta(x;v)\le\operatorname{dist}(0,Q). If g⋆≠0g^\star\ne0, take v⋆=g⋆/∥g⋆∥2v^\star=g^\star/\|g^\star\|_2. The projection optimality condition gives ⟨g−g⋆,g⋆⟩≥0\langle g-g^\star,g^\star\rangle\ge0 for all g∈Qg\in Q, hence

⟨g,v⋆⟩≥∥g⋆∥2for all g∈Q.\langle g,v^\star\rangle\ge\|g^\star\|_2\qquad\text{for all }g\in Q.

Therefore mδ(x;v⋆)=∥g⋆∥2=dist⁡(0,Q)m_\delta(x;v^\star)=\|g^\star\|_2=\operatorname{dist}(0,Q). If g⋆=0g^\star=0, then dist⁡(0,Q)=0\operatorname{dist}(0,Q)=0, and the stated equivalence for ϵ≥0\epsilon\ge0 follows immediately.

Lemma C.4. Let f:Rd→Rf:\mathbb{R}^d\to\mathbb{R} be Lipschitz. Fix x∈Rdx\in\mathbb{R}^d and δ>0\delta>0. There exists a full-measure set Vx⊂Sd−1\mathcal{V}_x\subset\mathbb{S}^{d-1} such that, for every v∈Vxv\in\mathcal{V}_x and every τ∈R\tau\in\mathbb{R},

mδ(x;v)≥τ⟹f(x−δv)≤f(x)−τδ.m_\delta(x;v)\ge\tau\Longrightarrow f(x-\delta v)\le f(x)-\tau\delta.

Proof. Fix xx. By Davis et al. [ [14], there is a set Vx⊂Sd−1\mathcal{V}_x\subset\mathbb{S}^{d-1} of full surface measure such that, for every v∈Vxv\in\mathcal{V}_x, the function t↦f(x−tv)t\mapsto f(x-tv) is differentiable for a.e. t∈[0,δ]t\in[0,\delta] and satisfies

f(x)−f(x−δv)=∫0δ⟨∇f(x−tv),v⟩ dt.(6)f(x)-f(x-\delta v)=\int_0^\delta\langle\nabla f(x-tv),v\rangle\,dt. \tag*{(6)}

Now suppose mδ(x;v)≥τm_{\delta}(x;v)\ge\tau. For a.e. t∈[0,δ]t\in[0,\delta], whenever ff is differentiable at x−tvx-tv, we have

∇f(x−tv)∈∂f(x−tv)⊆∂δf(x),\nabla f(x-tv)\in\partial f(x-tv)\subseteq\partial_{\delta}f(x),

since x−tv∈Bδ(x)x-tv\in B_{\delta}(x) for all t<δt<\delta and the endpoint is irrelevant for the integral. Therefore,

⟨∇f(x−tv),v⟩≥τfor a.e. t∈[0,δ].\langle\nabla f(x-tv),v\rangle\ge\tau\qquad\text{for a.e. }t\in[0,\delta].

Integrating gives

f(x)−f(x−δv)≥∫0δτ dt=τδ,f(x)-f(x-\delta v)\ge\int_{0}^{\delta}\tau\,dt=\tau\delta,

which is equivalent to the desired inequality.

Theorem 3.3. Algorithm 1 finds a (δ,τ,ρ)(\delta,\tau,\rho)-PGS point with probability at least 1−γ1-\gamma using O(Δτδlog⁡Δγτδ)O\left(\frac{\Delta}{\tau\delta}\log\frac{\Delta}{\gamma\tau\delta}\right) queries. In particular, for (τ,ρ)=(ε,ε)(\tau,\rho)=(\varepsilon,\varepsilon), this becomes O(Δδε2log⁡Δγδε)O\left(\frac{\Delta}{\delta\varepsilon^{2}}\log\frac{\Delta}{\gamma\delta\varepsilon}\right), with no polynomial dependence on LL.

Proof. Each accepted step decreases ff by at least τδ\tau\delta. Since f(x0)−flb=Δf(x_{0})-f_{\mathrm{lb}}=\Delta, the algorithm can accept at most Tmax⁡≤Δ/(τδ)T_{\max}\le\Delta/(\tau\delta) steps, and hence performs at most Tmax⁡+1T_{\max}+1 stopping checks.

Call a point xx bad if

Pr⁡V∼Unif⁡(Sd−1)[f(x−δV)≤f(x)−τδ]>ρ.\Pr_{V\sim\operatorname{Unif}(\mathbb{S}^{d-1})}\left[f(x-\delta V)\le f(x)-\tau\delta\right]>\rho.

At any bad point, the probability that all mm sampled directions fail to give sufficient decrease is at most

(1−ρ)m≤e−ρm≤γTmax⁡+1.(1-\rho)^{m}\le e^{-\rho m}\le\frac{\gamma}{T_{\max}+1}.

A union bound over the stopping checks shows that, with probability at least 1−γ1-\gamma, the algorithm never stops at a bad point. Thus the returned point xoutx_{\mathrm{out}} satisfies

Pr⁡V[f(xout−δV)≤f(xout)−τδ]≤ρ.\Pr_{V}\left[f(x_{\mathrm{out}}-\delta V)\le f(x_{\mathrm{out}})-\tau\delta\right]\le\rho.

By Lemma C.4, for almost every VV,

mδ(xout;V)≥τ⟹f(xout−δV)≤f(xout)−τδ.m_{\delta}(x_{\mathrm{out}};V)\ge\tau\quad\Longrightarrow\quad f(x_{\mathrm{out}}-\delta V)\le f(x_{\mathrm{out}})-\tau\delta.

Therefore,

Pr⁡V[mδ(xout;V)≥τ]≤ρ,\Pr_{V}\left[m_{\delta}(x_{\mathrm{out}};V)\ge\tau\right]\le\rho,

so xoutx_{\mathrm{out}} is (δ,τ,ρ)(\delta,\tau,\rho)-PGS. The query bound follows from at most Tmax⁡+1T_{\max}+1 iterations and m+1m+1 function evaluations per iteration.

Proofs and Details for Section 4

This appendix makes the two informal robustness statements from § 4 precise. We reuse the rank-separated witness pair constructed in the proof of Theorem 2.3.

Theorem 4.1. Fix k≥2k\ge2 and η∈(0,1)\eta\in(0,1). There exist constants ak,η>0a_{k,\eta}>0, γk,η>0\gamma_{k,\eta}>0, and μk,η>0\mu_{k,\eta}>0 such that, for every depth m∈Nm\in\mathbb{N}, one can choose orthogonal weights W1,…,Wm∈O(k)W_{1},\ldots,W_{m}\in O(k) and inputs Xclean,Xadv∈Rk×kX_{\mathrm{clean}},X_{\mathrm{adv}}\in\mathbb{R}^{k\times k} with the following properties:

rank⁡(Xclean)=k−1,rank⁡(Xadv)=k,∥Xadv−Xclean∥F≤ak,ηe−γk,ηm.\operatorname{rank}(X_{\mathrm{clean}})=k-1,\qquad\operatorname{rank}(X_{\mathrm{adv}})=k,\qquad\lVert X_{\mathrm{adv}}-X_{\mathrm{clean}}\rVert_{F}\le a_{k,\eta}e^{-\gamma_{k,\eta}m}.

Moreover, there is a linear readout Am∈Rk×kA_{m}\in\mathbb{R}^{k\times k} and an offset bm∈Rb_{m}\in\mathbb{R} with ∥Am∥F=1\lVert A_{m}\rVert_{F}=1 such that the scalar score

sm(Z):=⟨Am,Fm(Z)⟩F+bms_{m}(Z):=\langle A_{m},F_{m}(Z)\rangle_{F}+b_{m}

satisfies

sm(Xclean)≥μk,η,sm(Xadv)≤−μk,η.s_{m}(X_{\mathrm{clean}})\ge\mu_{k,\eta},\qquad s_{m}(X_{\mathrm{adv}})\le-\mu_{k,\eta}.

Consequently, for the binary classifier Z↦sign⁡(sm(Z))Z\mapsto\operatorname{sign}(s_{m}(Z)) with clean label +1+1 at XcleanX_{\mathrm{clean}},

inf⁡{∥E∥F:sm(Xclean+E)≤0}≤ak,ηe−γk,ηm.\inf\left\{\lVert E\rVert_{F}:s_{m}(X_{\mathrm{clean}}+E)\le0\right\}\le a_{k,\eta}e^{-\gamma_{k,\eta}m}.

Proof. Use the construction in the proof of Theorem 2.3. With r=rk,η>1r=r_{k,\eta}>1 from Corollary B.5, that proof constructs inputs Y0Y_0 and X0X_0 such that

rank⁡(Y0)=k−1,rank⁡(X0)=k,∥X0−Y0∥F≤2(1−η)k2(k−1)r−mk2(k−1).\operatorname{rank}(Y_0)=k-1,\qquad\operatorname{rank}(X_0)=k,\qquad\lVert X_0-Y_0\rVert_F\leq2(1-\eta)^{\frac{k}{2(k-1)}}r^{-\frac{mk}{2(k-1)}}.

It also constructs orthogonal weights for which, writing Ym=Fm(Y0)Y_m=F_m(Y_0) and Xm=Fm(X0)X_m=F_m(X_0),

∥Xm−Ym∥F2≥δk,η,δk,η:=ck(1−η)k,\lVert X_m-Y_m\rVert_F^2\geq\delta_{k,\eta},\qquad\delta_{k,\eta}:=c_k(1-\eta)^k,

where ck=((k−1)/k)k−1c_k=((k-1)/k)^{k-1} is the constant from Lemma B.7. Set

Xclean:=Y0,Xadv:=X0,γk,η:=k2(k−1)log⁡r,ak,η:=2(1−η)k2(k−1).X_{\mathrm{clean}}:=Y_0,\qquad X_{\mathrm{adv}}:=X_0,\qquad\gamma_{k,\eta}:=\frac{k}{2(k-1)}\log r,\qquad a_{k,\eta}:=2(1-\eta)^{\frac{k}{2(k-1)}}.

Then ∥Xadv−Xclean∥F≤ak,ηe−γk,ηm\lVert X_{\mathrm{adv}}-X_{\mathrm{clean}}\rVert_F\leq a_{k,\eta}e^{-\gamma_{k,\eta}m}.

It remains to choose the readout. Let

Δm:=∥Ym−Xm∥F≥δk,η,Am:=Ym−XmΔm,bm:=−12⟨Am,Ym+Xm⟩F.\Delta_m:=\lVert Y_m-X_m\rVert_F\geq\sqrt{\delta_{k,\eta}},\qquad A_m:=\frac{Y_m-X_m}{\Delta_m},\qquad b_m:=-\frac{1}{2}\langle A_m,Y_m+X_m\rangle_F.

Then ∥Am∥F=1\lVert A_m\rVert_F=1, and direct calculation gives

sm(Xclean)=⟨Am,Ym⟩F+bm=Δm2,sm(Xadv)=⟨Am,Xm⟩F+bm=−Δm2.s_m(X_{\mathrm{clean}})=\langle A_m,Y_m\rangle_F+b_m=\frac{\Delta_m}{2},\qquad s_m(X_{\mathrm{adv}})=\langle A_m,X_m\rangle_F+b_m=-\frac{\Delta_m}{2}.

Thus both inequalities hold with

μk,η:=12δk,η.\mu_{k,\eta}:=\frac{1}{2}\sqrt{\delta_{k,\eta}}.

Finally, the perturbation E=Xadv−XcleanE=X_{\mathrm{adv}}-X_{\mathrm{clean}} is feasible for the displayed infimum because sm(Xclean+E)=sm(Xadv)≤0s_m(X_{\mathrm{clean}}+E)=s_m(X_{\mathrm{adv}})\leq0. This proves the adversarial-radius bound.

Proposition 4.2. In the construction of Theorem 4.1, the perturbation Xadv−XcleanX_{\mathrm{adv}}-X_{\mathrm{clean}} is rank-creating:

rank⁡(Xclean)=k−1,rank⁡(Xadv)=k.\operatorname{rank}(X_{\mathrm{clean}})=k-1,\qquad\operatorname{rank}(X_{\mathrm{adv}})=k.

Furthermore, if Z∈Rk×kZ\in\mathbb{R}^{k\times k} is any rank-deficient input for which the forward pass is well-defined, then Fm(Z)F_m(Z) is rank-deficient and

∥Fm(Xadv)−Fm(Z)∥F≥δk,η,δk,η=ck(1−η)k.\lVert F_m(X_{\mathrm{adv}})-F_m(Z)\rVert_F\geq\sqrt{\delta_{k,\eta}},\qquad\delta_{k,\eta}=c_k(1-\eta)^k.

Thus the full-rank adversarial branch is separated by a constant from the entire rank-deficient output branch.

Proof. The rank identities for XcleanX_{\mathrm{clean}} and XadvX_{\mathrm{adv}} were proved in Theorem 4.1. Now let ZZ be rank-deficient and suppose the forward pass is well-defined. Each layer maps

Zt−1⟼BN⁡(Zt−1Wt)=Zt−1WtD(Zt−1Wt)−1/2.Z_{t-1}\longmapsto\operatorname{BN}(Z_{t-1}W_t)=Z_{t-1}W_tD(Z_{t-1}W_t)^{-1/2}.

Here WtW_t is invertible and, because the normalization is well-defined, D(Zt−1Wt)−1/2D(Z_{t-1}W_t)^{-1/2} is an invertible diagonal matrix. Hence the rank cannot increase through a layer, so Fm(Z)F_m(Z) is rank-deficient.

On the full-rank branch of the construction, the proof of Theorem 2.3 gives

λmin⁡(Fm(Xadv)⊤Fm(Xadv))≥δk,η.\lambda_{\min}\left(F_m(X_{\mathrm{adv}})^\top F_m(X_{\mathrm{adv}})\right)\geq\delta_{k,\eta}.

Applying Lemma B.8 with X=Fm(Xadv)X=F_m(X_{\mathrm{adv}}) and Y=Fm(Z)Y=F_m(Z) yields

∥Fm(Xadv)−Fm(Z)∥F2≥λmin⁡(Fm(Xadv)⊤Fm(Xadv))≥δk,η,\lVert F_m(X_{\mathrm{adv}})-F_m(Z)\rVert_F^2\geq\lambda_{\min}\left(F_m(X_{\mathrm{adv}})^\top F_m(X_{\mathrm{adv}})\right)\geq\delta_{k,\eta},

which proves the claim.

Experiments

Low-rank structure in image datasets. To support the rank-separation mechanism studied in the main text, we examine the spectral structure of standard image datasets. As shown in Figure 3 and Figure 5, images from MNIST, Fashion-MNIST, and CIFAR-10 exhibit strong approximate low-rank structure. Their spectra decay rapidly, and most of the spectral energy is concentrated in a small number of singular components. Thus, while the images are not necessarily exactly rank-deficient, they typically lie close to a low-rank set. This observation is consistent with classical low-rank models of image data [21, 22, 23, 24, 25] and motivates our focus on perturbations that move an input away from its low-rank structure.

Perturbation model for MNIST and Fashion-MNIST. In the MNIST and Fashion-MNIST experiments, we view each image as a matrix X∈[0,1]28×28X \in[0,1]^{28\times28} and compare several perturbation families at matched pre-clipping energy. Given a perturbation Δ\Delta, the perturbed image is

X~=Π[0,1](X+Δ),\widetilde{X} = \Pi_{[0,1]}(X+\Delta),

where Π[0,1]\Pi_{[0,1]} denotes entrywise clipping to the valid pixel range. The noise level is controlled by a variance parameter σ2\sigma^2, and before clipping every sampled perturbation is rescaled so that

∥Δ∥F=σ28⋅28.\lVert\Delta\rVert_F = \sigma\sqrt{28\cdot28}.

Thus all noise families have the same root-mean-square pixel energy before clipping, making the curves directly comparable across perturbation types. We consider four perturbation structures. IID pixel noise samples a full-dimensional Gaussian direction and is full rank with probability one, so it directly tests rank-creating perturbations. Singular-value (or same-subspace) noise perturbs only the singular values of the image and therefore stays within the row and column subspaces already present in XX. Column-offset and row-offset noise broadcast one Gaussian vector across rows or columns, respectively, producing structured rank-one perturbations. Together, these perturbation families allow us to separate the effect of perturbation magnitude from the effect of perturbation geometry.

Rank effects of matched-energy perturbations. We next examine how different perturbation geometries affect the intrinsic rank structure of the images when their pre-clipping energy is matched. As shown in Figure 6 and Figure 7, i.i.d. pixel noise consistently increases effective rank, stable rank, and the number of components needed to capture most of the spectral energy. In contrast, same-subspace perturbations have a much weaker effect, while row- and column-structured perturbations often preserve or even reduce these rank-based quantities. These observations support the mechanism in the main text that perturbations that create new spectral directions are more disruptive to the low-rank structure of images than perturbations that remain aligned with it.

Figure 6. Rank structure under matched-energy perturbations on MNIST and Fashion-MNIST. For each dataset, we report numerical rank, effective rank, and stable rank as the perturbation variance σ2\sigma^{2} increases. IID pixel noise strongly inflates the intrinsic rank measures, while same-subspace, column-wise, and row-wise perturbations produce much smaller changes despite having the same pre-clipping energy.

Figure 7. Rank inflation under matched-energy perturbations on CIFAR-10 luminance images. The plots show effective rank, stable rank, and the number of components required to capture 90% and 95% of the spectral energy. IID pixel noise substantially increases all rank-based measures as σ2\sigma^{2} grows, whereas same-subspace and structured row or column perturbations have a much milder effect.

Robustness under matched-energy perturbations. We next evaluate whether the geometric differences between perturbation families translate into different prediction behavior. Figure 8 shows that, even when perturbations are matched in pre-clipping energy, their effect on BatchNorm networks can differ substantially. Across MNIST, Fashion-MNIST, and CIFAR-10, subspace-preserving perturbations tend to preserve accuracy, confidence, and margin for larger noise levels, and they produce lower prediction-flip rates. In contrast, i.i.d. pixel noise and structured row or column perturbations often cause earlier confidence collapse, margin degradation, and prediction instability. This is consistent with the rank-separation mechanism from the main text: perturbations that move images away from their dominant low-rank structure are amplified more strongly than perturbations that remain aligned with that structure.

Batch normalization versus matched controls. We next compare models with batch normalization to matched architectures without batch normalization under the same perturbation families. As shown in Figure 9, BatchNorm models often maintain high accuracy at very small noise levels but can undergo a much sharper accuracy drop once the perturbation variance crosses a moderate threshold. This effect is especially clear on MNIST under i.i.d. pixel noise, column offsets, and row offsets. In contrast, the matched NoBN models tend to degrade more gradually. Same-subspace perturbations remain comparatively less damaging, which is consistent with the rank-separation mechanism discussed in the main text.

Figure 9. Batch normalization versus matched NoBN controls under structured input perturbations. Top row shows MNIST and bottom row shows Fashion-MNIST. For each dataset, we compare accuracy as the perturbation variance σ2\sigma^2 increases under i.i.d. pixel noise, same-subspace singular perturbations, column offsets, and row offsets. BatchNorm models often remain accurate at low noise levels but exhibit sharper degradation at larger perturbation strengths, while matched NoBN controls typically degrade more gradually.

References

  1. [1]Kaiming He, Xiangyu Zhang, Shaoqing Ren, and Jian Sun. Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 770–778, 2016.arxiv.org/abs/1512.03385
  2. [2]Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya Sutskever, et al. Language models are unsupervised multitask learners. OpenAI blog, 1(8):9, 2019.
  3. [3]Sergey Ioffe and Christian Szegedy. Batch normalization: Accelerating deep network training by reducing internal covariate shift. In International conference on machine learning, pages 448–456. pmlr, 2015.
  4. [4]Peter L Bartlett, Dylan J Foster, and Matus J Telgarsky. Spectrally-normalized margin bounds for neural networks. Advances in neural information processing systems, 30, 2017.arxiv.org/abs/1706.08498
  5. [5]Sébastien Bubeck and Mark Sellke. A universal law of robustness via isoperimetry. Advances in Neural Information Processing Systems, 34:28811–28822, 2021.arxiv.org/abs/2105.12806
  6. [6]Tim Salimans and Durk P Kingma. Weight normalization: A simple reparameterization to accelerate training of deep neural networks. Advances in neural information processing systems, 29, 2016.
  7. [7]Jimmy Lei Ba, Jamie Ryan Kiros, and Geoffrey E Hinton. Layer normalization. arXiv preprint arXiv:1607.06450, 2016.
  8. [8]Sergey Ioffe. Batch renormalization: Towards reducing minibatch dependence in batch-normalized models. Advances in neural information processing systems, 30, 2017.
  9. [9]Hadi Daneshmand, Amir Joudaki, and Francis Bach. Batch normalization orthogonalizes representations in deep random networks. Advances in Neural Information Processing Systems, 34:4896–4906, 2021.
  10. [10]Susanna Lange, Kyle Helfrich, and Qiang Ye. Batch normalization preconditioning for neural network training. Journal of Machine Learning Research, 23(72):1–41, 2022.arxiv.org/abs/2108.01110
  11. [11]Yurii Nesterov. Introductory lectures on convex optimization: A basic course, volume 87. Springer Science & Business Media, 2013.
  12. [12]Allen A Goldstein. Optimization of lipschitz continuous functions. Mathematical Programming, 13 (1):14–22, 1977.DOI
  13. [13]Jingzhao Zhang, Hongzhou Lin, Stefanie Jegelka, Suvrit Sra, and Ali Jadbabaie. Complexity of finding stationary points of nonconvex nonsmooth functions. In International Conference on Machine Learning, pages 11173–11182. PMLR, 2020.
  14. [14]Damek Davis, Dmitriy Drusvyatskiy, Yin Tat Lee, Swati Padmanabhan, and Guanghao Ye. A gradient sampling method with complexity guarantees for lipschitz functions in high and low dimensions. Advances in neural information processing systems, 35:6692–6703, 2022.
  15. [15]Michael Jordan, Guy Kornowski, Tianyi Lin, Ohad Shamir, and Manolis Zampetakis. Deterministic nonsmooth nonconvex optimization. In The Thirty Sixth Annual Conference on Learning Theory, pages 4570–4597. PMLR, 2023.arxiv.org/abs/2302.08300
  16. [16]Guy Kornowski and Ohad Shamir. On the complexity of finding small subgradients in nonsmooth optimization. arXiv preprint arXiv:2209.10346, 2022.
  17. [17]Songyan Hou, Parnian Kassraie, Anastasis Kratsios, Andreas Krause, and Jonas Rothfuss. Instance-dependent generalization bounds via optimal transport. Journal of Machine Learning Research, 24 (349):1–51, 2023.
  18. [18]Angus Galloway, Anna Golubeva, Thomas Tanay, Medhat Moussa, and Graham W Taylor. Batch normalization is a cause of adversarial vulnerability. arXiv preprint arXiv:1905.02161, 2019.
  19. [19]Haotao Wang, Aston Zhang, Shuai Zheng, Xingjian Shi, Mu Li, and Zhangyang Wang. Removing batch normalization boosts adversarial training. In International Conference on Machine Learning, pages 23433–23445. PMLR, 2022.arxiv.org/abs/2207.01156
  20. [20]Noam Zeise and Tiffany Joyce Vlaar. Batchnorm layers have an outsized effect on adversarial robustness. In OPT 2025: Optimization for Machine Learning, 2025.
  21. [21]Berkant Savas and Lars Eldén. Handwritten digit classification using higher order singular value decomposition. Pattern recognition, 40(3):993–1003, 2007.
  22. [22]Yangmuzi Zhang, Zhuolin Jiang, and Larry S Davis. Learning structured low-rank representations for image classification. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 676–683, 2013.DOI
  23. [23]Shuhang Gu, Lei Zhang, Wangmeng Zuo, and Xiangchu Feng. Weighted nuclear norm minimization with application to image denoising. In Proceedings of the IEEE conference on computer vision and pattern recognition, pages 2862–2869, 2014.DOI
  24. [24]Jun Xu, Lei Zhang, Wangmeng Zuo, David Zhang, and Xiangchu Feng. Patch group based nonlocal self-similarity prior learning for image denoising. In Proceedings of the IEEE international conference on computer vision, pages 244–252, 2015.DOI
  25. [25]Hangfan Liu, Xinfeng Zhang, and Ruiqin Xiong. Content-adaptive low rank regularization for image denoising. In 2016 IEEE International Conference on Image Processing (ICIP), pages 3091–3095. IEEE, 2016.DOI
  26. [26]Amir Joudaki, Hadi Daneshmand, and Francis Bach. On the impact of activation and normalization in obtaining isometric embeddings at initialization. Advances in Neural Information Processing Systems, 36:39855–39875, 2023.arxiv.org/abs/2305.18399
  27. [27]Alexandru Meterez, Amir Joudaki, Francesco Orabona, Alexander Immer, Gunnar Ratsch, and Hadi Daneshmand. Towards training without depth limits: Batch normalization without gradient explosion. In The Twelfth International Conference on Learning Representations, 2024.arxiv.org/abs/2310.02012
  28. [28]Greg Yang, Jeffrey Pennington, Vinay Rao, Jascha Sohl-Dickstein, and Samuel S Schoenholz. A mean field theory of batch normalization. arXiv preprint arXiv:1902.08129, 2019.arxiv.org/abs/1902.08129
  29. [29]Vinod Nair and Geoffrey E Hinton. Rectified linear units improve restricted boltzmann machines. In Proceedings of the 27th international conference on machine learning (ICML-10), pages 807–814, 2010.
  30. [30]Andrew L Maas, Awni Y Hannun, Andrew Y Ng, et al. Rectifier nonlinearities improve neural network acoustic models. In Proceedings of the 30th annual international conference on machine learning. Atlanta, GA, 2013.
  31. [31]Honglak Lee, Roger Grosse, Rajesh Ranganath, and Andrew Y Ng. Convolutional deep belief networks for scalable unsupervised learning of hierarchical representations. In Proceedings of the 26th annual international conference on machine learning, pages 609–616, 2009.DOI
  32. [32]Simone Scardapane, Danilo Comminiello, Amir Hussain, and Aurelio Uncini. Group sparse regularization for deep neural networks. Neurocomputing, 241:81–89, 2017.DOI
  33. [33]Andre Martins and Ramon Astudillo. From softmax to sparsemax: A sparse model of attention and multi-label classification. In International conference on machine learning, pages 1614–1623. PMLR, 2016.arxiv.org/abs/1602.02068
  34. [34]Mathieu Blondel, Olivier Teboul, Quentin Berthet, and Josip Djolonga. Fast differentiable sorting and ranking. In International Conference on Machine Learning, pages 950–959. PMLR, 2020.arxiv.org/abs/2002.08871
  35. [35]Tianyi Lin, Zeyu Zheng, and Michael Jordan. Gradient-free methods for deterministic and stochastic nonsmooth nonconvex optimization. Advances in Neural Information Processing Systems, 35:26160–26175, 2022.arxiv.org/abs/2209.05045
  36. [36]Siyu Kong and AS Lewis. The cost of nonconvexity in deterministic nonsmooth optimization. Mathematics of Operations Research, 2023.arxiv.org/abs/2210.00652
  37. [37]Günter Klambauer, Thomas Unterthiner, Andreas Mayr, and Sepp Hochreiter. Self-normalizing neural networks. Advances in neural information processing systems, 30, 2017.arxiv.org/abs/1706.02515
  38. [38]Saeed Ghadimi and Guanghui Lan. Stochastic first-and zeroth-order methods for nonconvex stochastic programming. SIAM journal on optimization, 23(4):2341–2368, 2013.arxiv.org/abs/1309.5549
  39. [39]Rong Ge, Furong Huang, Chi Jin, and Yang Yuan. Escaping from saddle points—online stochastic gradient for tensor decomposition. In Conference on learning theory, pages 797–842. PMLR, 2015.arxiv.org/abs/1503.02101
  40. [40]Chi Jin, Rong Ge, Praneeth Netrapalli, Sham M Kakade, and Michael I Jordan. How to escape saddle points efficiently. In International conference on machine learning, pages 1724–1732. PMLR, 2017.
  41. [41]Chi Jin, Praneeth Netrapalli, Rong Ge, Sham M Kakade, and Michael I Jordan. On nonconvex optimization for machine learning: Gradients, stochasticity, and saddle points. Journal of the ACM (JACM), 68(2):1–29, 2021.
  42. [42]Max Simchowitz, Ahmed El Alaoui, and Benjamin Recht. On the gap between strict-saddles and true convexity: An omega (log d) lower bound for eigenvector approximation. arXiv preprint arXiv:1704.05448, 2017.
  43. [43]Hadi Daneshmand, Jonas Kohler, Aurelien Lucchi, and Thomas Hofmann. Escaping saddles with stochastic gradients. In International Conference on Machine Learning, pages 1155–1164. PMLR, 2018.arxiv.org/abs/1803.05999
  44. [44]Frank H Clarke. Generalized gradients and applications. Transactions of the American Mathematical Society, 205:247–262, 1975.DOI
  45. [45]Michael I Jordan, Tianyi Lin, and Manolis Zampetakis. On the complexity of deterministic nonsmooth and nonconvex optimization. arXiv preprint arXiv:2209.12463, 2022.arxiv.org/abs/2209.12463
  46. [46]Ashok Cutkosky, Harsh Mehta, and Francesco Orabona. Optimal stochastic non-smooth non-convex optimization through online-to-non-convex conversion. In International Conference on Machine Learning, pages 6643–6670. PMLR, 2023.arxiv.org/abs/2302.03775

Paper details

vlnc:W_h6dd646v
v1 · latestHistory (1)

published · Fresh — added to the Hub within the last 30 days

added ·

Usage

views · 0

downloads · 0

A view counts once per browser per day, a download each time the PDF is requested.

CC BY 4.0

Original source

  • Computer Science

Not recorded for this paper · View statement →

Looking for related papers…
Contents