WorksComputer Science
On Lipschitz Explosion in Deep Neural Networks with Normalization: Consequences for Optimization and Adversarial Robustness
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 , 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 for the network map with parameters (and using Frobenius / Euclidean norms for concreteness), we consider
The parameter Lipschitz constant 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 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 and can grow exponentially with depth , and consequently exponentially with the ambient parameter dimension . This occurs even under strong per-layer norm control (e.g., , and in the input-Lipschitz construction even 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 of the objective [13, 14, 15]. In our setting, is not an external constant; instead, it is induced by the network and our lower bound shows it can be for deep normalized architectures. Consequently, Lipschitz-based “dimension-free” guarantees of the form can unfortunately translate into exponential dependence on parameter for this model class. Moreover, lower bounds in nonsmooth optimization show that some polynomial dependence on 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 . 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 . 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 time and hence in time.
Implications of input-Lipschitz explosion for generalization and adversarial robustness
The same phenomenon impacts learning-theoretic uses of Lipschitz control. Since (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 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 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).

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 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 . 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 .
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 . 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 , is the Frobenius norm and the operator norm. For a square matrix , denotes its smallest eigenvalue. We write for the -th row of and for the -th column of . We denote the set by .
Batch normalization (Column-wise normalization on ). Let . Define the diagonal matrix of column-squared norms
When every column of is nonzero (equivalently ), the column-wise batch normalization map is . Equivalently, . 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 , then every column of has unit norm, hence
Remark 2.1 (Regularization at zero columns). If one wants to be defined on all of , one may work with the standard -regularized variant
and (when desired) take at the end. Our lower bound proof below is written for on the open set , and remains valid for with the same qualitative conclusion.
A deep network with normalization layers and linear activation. Fix an integer , i.e., the batch size. Let be positive integers (layer widths / feature dimensions). For each layer let be a weight matrix. Given an input matrix , define the forward iterates
whenever the normalization is well-defined. The depth- network map is
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 and an input domain on which every normalization step is defined. View the network as a map from inputs to outputs. Its input Lipschitz constant is
Second, fix an input batch and view the network output as a function of the parameters. For a weight tuple , we use the Euclidean norm obtained by vectorizing all weights, . The corresponding parameter-to-output map and its Lipschitz constant are
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 for every . Our exponential lower bounds are established on compact admissible domains with scales bounded away from zero uniformly in depth. Specifically, the input domains satisfy , while every weight matrix in the parameter witnesses has operator norm one.
Explosion of the Input Lipschitz Constant
We first show that the input Lipschitz constant 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:
Here each weight matrix is orthogonal , so and therefore . 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, . 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 . Moreover, this growth is witnessed by uncountably many input pairs.
Theorem 2.3 (Exponential lower bound on ). Fix and constant . There exist constants and such that, for every depth , one can choose orthogonal weights with for all , for which the depth- map defined in (2)–(3) satisfies , where constants and depend on . 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, and , that are exponentially close, with rank-deficient and only slightly perturbed away from being rank-deficient. After carefully chosen normalization steps, the trajectory starting from becomes uniformly well-conditioned, while the trajectory starting from remains rank-deficient. Thus the two outputs are separated by at least 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 and set . There exist constants and such that, for every integer , there is a depth- network with at most scalar weight parameters whose input–output map satisfies .
Remark 2.5. Although the constants in our general bounds in Theorem 2.3 depend on , the exponential lower bounds hold uniformly over all widths . Indeed, the width-two construction embeds into any larger width via block-diagonal matrices, preserving the same exponential rate and constants independently of ; see Appendix § 8.10.
Explosion of the Parameter Lipschitz Constant
In Section 2.2, we showed that the input–output map can have an input Lipschitz constant 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 .
Theorem 2.6 (Exponential lower bound on ). Fix and constant . There exist constants and depending on and such that, for every depth , there is a depth- network of the form (2), with for all , for which the fixed-input parameter-to-output map with satisfies .
The proof is a direct reduction from the input-Lipschitz construction in Theorem 2.3. We fix the input to be 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 .
Corollary 2.7. Fix any and set . There exist constants and such that, for every integer , one can construct a depth- network with at most scalar weight parameters for which .
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 and define
Here plays the role of the observed label generated by the ground-truth network. The objective function is equivalent to the mean-squared loss evaluated on the outputs of the neural network.
The next corollary shows that the exponential growth of transfers to the Lipschitz constant of this induced objective.
Corollary 2.8. Let be the parameter-to-output map constructed in Theorem 2.6. There exists a realizable target parameter such that the objective defined in (4) satisfies
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 . Since the proof of Theorem 2.3 establishes existence through a nonconstructive probabilistic argument over the weights , we use a constructive heuristic to select the weights in our experiments. Specifically, at each layer , we choose 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 and its random rotations 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 with the predicted exponential growth. The same plot also shows witnesses obtained by applying random rotations to the input witness pair, yielding pairs of the form . 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 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 , but if it depends polynomially on the objective Lipschitz constant , then Corollary 2.8 implies that the same guarantee can still scale exponentially with . 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 -approximate Goldstein stationary points have oracle complexity that depends polynomially on the objective Lipschitz constant [13, 14, 35, 15, 36]. Such a dependence is acceptable when is viewed as a fixed property of the optimization problem. In deep normalized networks, however, is not an external constant; it is determined by the architecture and can grow with depth. Our results in § 2 show that can be as large as . Consequently, a guarantee of the form can translate into exponential dependence on the parameter dimension , even when the stated rate has no explicit dependence on .
This raises a natural question: can one design algorithms for Goldstein stationarity whose dependence on is only logarithmic, for example with oracle complexity polynomial in , , , and ?
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 , , , and only logarithmic in , 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 -Goldstein stationary points of 1-Lipschitz functions.
Corollary 3.1 (Informal). In the worst case, polynomial dependence on the Lipschitz constant is unavoidable for finding -Goldstein stationary points. In particular, no randomized first-order algorithm can have oracle complexity only 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 is taken into account. This suggests a natural quest: to identify weaker, but still meaningful, stationarity notions whose complexity is polynomial in , , and , and polynomial only in . 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 .
Goldstein stationarity is a worst-case directional condition. To see this, fix and let denote the Goldstein subdifferential. For a direction , consider the support function of the set of Goldstein subdifferentials . By Lemma C.3, Goldstein stationarity can be written in the equivalent form . Thus, a -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 , a margin threshold , and a probability level . Let . We say that is -PGS if . We write is -PGS as shorthand for .
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 -PGS point with probability at least using queries, with no polynomial dependence on .
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 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 , 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 , but the clean input can be misclassified after a perturbation of Frobenius norm at most .
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 . 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 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 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 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 -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 , the radius parameter , and the accuracy . More recently, Davis et al. [ [14] developed a gradient-sampling approach under a standard oracle model, improving the dependence on and in certain regimes, at the cost of an additional dependence on the dimension. While these guarantees are polynomial for a fixed Lipschitz constant , this distinction is important in deep learning, where 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 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 , the linear maps are non-expansive, so any growth must come from the normalization. Our argument exploits a tension between (i) the ability of 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 in Definition B.1. For normalized matrices we have by (1), hence . A key input is the one-step lift inequality of Meterez et al. [ [27]: Theorem B.3 shows that for and Haar-uniform ,
where is the spectral deviation from Definition B.2. Lemma B.4 provides a uniform gap whenever . Combining these yields a deterministic multiplicative lift: Corollary B.5 guarantees that, as long as , we can choose so that with . Iterating this choice gives the greedy geometric increase Corollary B.6, so after layers we can enforce provided the initial isotropy is at least .
We realize such an initialization while keeping it exponentially close to rank deficiency. As in the appendix proof of Theorem 2.3, we set and choose with . Then , so , and . Choosing
ensures , hence (by Corollary B.6) we obtain along the resulting trajectory. Near-isotropy then forces uniform conditioning: Lemma B.7 yields .
To turn this into a Lipschitz lower bound, we compare to a rank- neighbor. Let be an SVD with singular values and define the matrix . Then and . Let under the same weights (one may, if desired, interpret BN via the standard -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 for all . Applying Lemma B.8 at depth gives , and therefore
with and as in Theorem 2.3. Finally, the orthogonal equivariance for implies , so the same lower bound is witnessed by the uncountable family of pairs , 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 and assume .
Definition B.1 (Isotropy functional). Define
The quantity is scale-invariant and satisfies if and only if is a scalar multiple of the identity. In this case, we say that we have “perfect isotropy”. If (so that ), then .
Definition B.2 (Spectral deviation). Suppose and let be the eigenvalues of (so ). Define
Theorem B.3 (One-step isotropy lift [27 Theorem A.4]). Let and let be Haar-uniform on . Then
A Uniform Multiplicative Lift Below an Isotropy Cap
Fix and define the compact set of eigenvalue vectors
Consider from Definition B.2.
We now prove a uniform spectral gap away from .
Lemma B.4. Let and set . Then . Moreover, if and , then .
Proof. is nonempty, closed, and bounded, hence compact, and is continuous, so the minimum is attained. Also iff , but that vector is not in since its geometric mean is . Thus . The final statement follows because and, for , .
In turn, we can convert this result into a deterministic multiplicative lift for the isotropy functional , as long as it is below .
Corollary B.5. Fix and define
If and , then there exists an orthogonal matrix such that
Proof. By Lemma B.4, . Theorem B.3 then gives . Since , there must exist a realization of achieving at least the expectation.
Consequently, we can show that the isotropy functional increases geometrically until reaches near-isotropy. In the proof, we choose the weight matrices greedily according to Corollary B.5.
Corollary B.6. Fix and let be as above. Let and choose greedily as follows: for each , if choose satisfying Corollary B.5, and set . Then for every ,
Proof. If , then by construction . Once we do not require any further increase. An induction yields the claimed bound.
Near-isotropy Forces a Floor on
A useful basic result is the inequality that connects the minimum eigenvalue in the spectrum of to the isotropy functional .
Lemma B.7. Let and let be the eigenvalues of (so and ). Then
Proof. Set and note . By AM–GM, for fixed sum the product of the first eigenvalues is maximized when they are equal, hence
Therefore
which rearranges to .
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 and suppose that . Then
Proof. Since there exists a unit vector . Then
Finally, .
Exponential growth of with depth
Theorem 2.3 (Exponential lower bound on ). Fix and constant . There exist constants and such that, for every depth , one can choose orthogonal weights with for all , for which the depth- map defined in (2)–(3) satisfies
where constants and depend on . Moreover, for this same fixed choice of weights, the lower bound is witnessed by an uncountable family of input pairs.
Proof. Fix and let from Corollary B.5. Define
Step 1: choose an initialization with controlled anisotropy. Let
Then and the eigenvalues of are and . Choose any such that (e.g. take where is a Cholesky factorization). Since , every column of has norm , hence .
Moreover,
because for .
Step 2: pick so that greedy steps reach near-isotropy. Choose
Then .
Now choose greedily as in Corollary B.6 and set . By Corollary B.6, .
Step 3: eigenvalue floor after reaching near-isotropy. Lemma B.7 yields
Step 4: a rank-deficient companion and distance growth. Let be an SVD with singular values . Define the rank- matrix
Then and
Let . Since each layer is multiplication by an invertible matrix () followed by multiplication by a diagonal matrix (coming from BN), the rank cannot increase, hence for all .
Applying Lemma B.8 at depth gives
Therefore
Uncountably many witness pairs for the same weights. For any orthogonal , we have for all and all orthogonal ,
because . Hence by induction. Consequently, the ratio equals the ratio for for every . Since is uncountable, this yields uncountably many witness pairs.
Corollary 2.4. Fix and set . There exist constants and such that, for every integer , there is a depth- network with at most scalar weight parameters whose input–output map satisfies
Proof. When , each layer contains 4 scalar parameters if the weight matrix is counted as an unconstrained matrix. Thus, a depth- network has scalar parameters. Given a budget , choose , so that . Applying Theorem 2.3 with gives
The claim follows by setting
Exponential growth of with depth
Theorem 2.6 (Exponential lower bound on ). Fix and constant . There exist constants and depending on and such that, for every depth , there is a depth- network of the form (2), with for all , for which the fixed-input parameter-to-output map with satisfies
Proof. Apply Theorem 2.3 with depth to obtain orthogonal matrices and matrices with such that the corresponding depth- map
satisfies
where is exponentially small in as in the proof of Theorem 2.3.
Now consider a depth- network with input and weights
where is chosen so that . Since column-wise batch normalization is scale-invariant,
we have and . Moreover, .
Let and . Then
The second trajectory remains rank-deficient at every depth, so Lemma B.8 (applied at depth ) and the conditioning lower bound from the proof of Theorem 2.3 yield
On the other hand,
Therefore
since is exponentially small in (same as in Theorem 2.3).
Corollary 2.8. Let be the parameter-to-output map constructed in Theorem 2.6. There exists a realizable target parameter such that the objective defined in (4) satisfies
up to the constants in Corollary 2.7.
Proof. In the construction of Theorem 2.6, let be one of the two parameter choices witnessing the lower bound on , and let be the other one. Since the target is realizable, . Therefore,
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.
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 , by
whenever every centered column is nonzero. This corresponds to the standard batch-variance normalization with a fixed affine scale of .
We now try to embed our hard instances into the centered subspace. Suppose , and choose satisfying
Such a matrix exists since the orthogonal complement of in has dimension . Left multiplication by is an isometry and maps every matrix into the subspace of column-centered matrices. In particular, for every ,
so the embedding preserves rank, Gram eigenvalues, column norms, and Frobenius distances.
Moreover, since , mean subtraction leaves unchanged. Consequently, for every and for which normalization is defined,
Thus, centered normalization reproduces the original normalization dynamics exactly on the embedded subspace.
Let denote the depth- network obtained by replacing each normalization layer of with , while keeping the same weight matrices. Applying (5) inductively yields
In particular, applying this identity to the witness pair from Theorem 2.3 gives
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 in Theorem 2.6 with , and using the same two weight tuples, preserves both their parameter distance and the corresponding output separation. Since the weight matrices remain , the parameter count is unchanged.
Finally, the robustness construction of Theorem 4.1 extends directly to the centered network. Using the readout and the same offset , we have and
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
where ReLU acts entrywise and denotes the all-ones matrix. Due to normalization layers, each column of has unit Euclidean norm, so every entry lies in . 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
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 , induction yields
In particular, for the witness pair ,
Since , 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 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 and offset . Indeed, since ,
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 .
Let and be the witnesses from Theorem 2.3 for . For any , consider the block-diagonal matrices
where the identity blocks are omitted when . Since normalization acts columnwise and the additional columns have unit norm, the two trajectories at depth are precisely and . Their input distance and final output separation are therefore identical to those of the width-two construction. Consequently, the input Lipschitz lower bound
holds for every , with constants independent of width.
Similarly, applying the same embedding to the parameter witnesses in Theorem 2.6, with fixed input , 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 , and the input lower bound becomes
Proofs and Details for Section 3
Background on Goldstein Stationarity and First-order Oracle
Notation and Background. Let be (locally) -Lipschitz with respect to :
Let denote the Clarke subdifferential at , which coincides with when is differentiable at and agrees with the usual convex subdifferential when is convex. The canonical solution concept studied in nonsmooth nonconvex optimization is that of a Clarke stationary point [44]. A point is -Clarke stationary if , where
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 . The key idea is to convexify the subdifferential over a neighborhood:
We say that is -Goldstein stationary if
This definition recovers the smooth optimality condition when is differentiable and 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 , the oracle reveals the function value together with subgradient information at (e.g., a Clarke subgradient or even the full set ). The lower bounds stated in the paper holds even when the oracle returns the entire set .
Proofs and Details for Section 3.1
Theorem C.1 ([16 Theorem 3]). For any and any , the following holds. For every randomized first-order algorithm , there exists a convex 1-Lipschitz function , with , such that , but
unless .
We now immediately show that no randomized algorithm can avoid a polynomial dependence on .
Corollary 3.1 (Formal). Fix and , and suppose that . Then, for any randomized first-order algorithm , there exists a convex -Lipschitz function , with , such that , and
unless
In particular, in the worst case, any randomized first-order method needs oracle calls to find a -Goldstein stationary point with probability at least .
Proof. Set . Since , we have . Applying Theorem C.1 with accuracy gives a convex 1-Lipschitz function , with , such that , and no randomized first-order algorithm can find a -Goldstein stationary point with constant probability unless
The condition on becomes
Now define . Then is convex and -Lipschitz, and
Moreover, by positive homogeneity of the Clarke and Goldstein subdifferentials,
and therefore
Thus, is -Goldstein stationary for if and only if it is -Goldstein stationary for , i.e., -Goldstein stationary for .
Consequently, any randomized first-order algorithm that finds a -Goldstein stationary point of with probability at least would also find a -Goldstein stationary point of with the same probability. By Theorem C.1, this requires
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 and requires . In our deep normalized networks, can be , 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 into yields exponential dependence on the network dimension.
Algorithm 1 Probabilistic Greedy Goldstein Descent (PGGD)
Require: , , , , , .
for to do
Sample i.i.d.
Query for all and let
if then
else
return
end if
end for
return
Why exploding 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 -Goldstein stationary points scale polynomially in the objective Lipschitz constant , and Corollary 3.1 shows that some polynomial dependence on is unavoidable in the worst case. But in our setting, 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 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 and , recall
The following elementary identity is the dual form of Goldstein stationarity.
Lemma C.3. For any ,
Moreover, if , then
Proof. Let and let be the projection of onto . For any unit vector ,
Thus . If , take . The projection optimality condition gives for all , hence
Therefore . If , then , and the stated equivalence for follows immediately.
Lemma C.4. Let be Lipschitz. Fix and . There exists a full-measure set such that, for every and every ,
Proof. Fix . By Davis et al. [ [14], there is a set of full surface measure such that, for every , the function is differentiable for a.e. and satisfies
Now suppose . For a.e. , whenever is differentiable at , we have
since for all and the endpoint is irrelevant for the integral. Therefore,
Integrating gives
which is equivalent to the desired inequality.
Theorem 3.3. Algorithm 1 finds a -PGS point with probability at least using queries. In particular, for , this becomes , with no polynomial dependence on .
Proof. Each accepted step decreases by at least . Since , the algorithm can accept at most steps, and hence performs at most stopping checks.
Call a point bad if
At any bad point, the probability that all sampled directions fail to give sufficient decrease is at most
A union bound over the stopping checks shows that, with probability at least , the algorithm never stops at a bad point. Thus the returned point satisfies
By Lemma C.4, for almost every ,
Therefore,
so is -PGS. The query bound follows from at most iterations and 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 and . There exist constants , , and such that, for every depth , one can choose orthogonal weights and inputs with the following properties:
Moreover, there is a linear readout and an offset with such that the scalar score
satisfies
Consequently, for the binary classifier with clean label at ,
Proof. Use the construction in the proof of Theorem 2.3. With from Corollary B.5, that proof constructs inputs and such that
It also constructs orthogonal weights for which, writing and ,
where is the constant from Lemma B.7. Set
Then .
It remains to choose the readout. Let
Then , and direct calculation gives
Thus both inequalities hold with
Finally, the perturbation is feasible for the displayed infimum because . This proves the adversarial-radius bound.
Proposition 4.2. In the construction of Theorem 4.1, the perturbation is rank-creating:
Furthermore, if is any rank-deficient input for which the forward pass is well-defined, then is rank-deficient and
Thus the full-rank adversarial branch is separated by a constant from the entire rank-deficient output branch.
Proof. The rank identities for and were proved in Theorem 4.1. Now let be rank-deficient and suppose the forward pass is well-defined. Each layer maps
Here is invertible and, because the normalization is well-defined, is an invertible diagonal matrix. Hence the rank cannot increase through a layer, so is rank-deficient.
On the full-rank branch of the construction, the proof of Theorem 2.3 gives
Applying Lemma B.8 with and yields
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 and compare several perturbation families at matched pre-clipping energy. Given a perturbation , the perturbed image is
where denotes entrywise clipping to the valid pixel range. The noise level is controlled by a variance parameter , and before clipping every sampled perturbation is rescaled so that
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 . 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 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 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 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]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]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]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]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]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]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]Jimmy Lei Ba, Jamie Ryan Kiros, and Geoffrey E Hinton. Layer normalization. arXiv preprint arXiv:1607.06450, 2016.
- [8]Sergey Ioffe. Batch renormalization: Towards reducing minibatch dependence in batch-normalized models. Advances in neural information processing systems, 30, 2017.
- [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]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]Yurii Nesterov. Introductory lectures on convex optimization: A basic course, volume 87. Springer Science & Business Media, 2013.
- [12]Allen A Goldstein. Optimization of lipschitz continuous functions. Mathematical Programming, 13 (1):14–22, 1977.DOI
- [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]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]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]Guy Kornowski and Ohad Shamir. On the complexity of finding small subgradients in nonsmooth optimization. arXiv preprint arXiv:2209.10346, 2022.
- [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]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]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]Noam Zeise and Tiffany Joyce Vlaar. Batchnorm layers have an outsized effect on adversarial robustness. In OPT 2025: Optimization for Machine Learning, 2025.
- [21]Berkant Savas and Lars Eldén. Handwritten digit classification using higher order singular value decomposition. Pattern recognition, 40(3):993–1003, 2007.
- [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]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]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]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]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]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]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]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]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]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]Simone Scardapane, Danilo Comminiello, Amir Hussain, and Aurelio Uncini. Group sparse regularization for deep neural networks. Neurocomputing, 241:81–89, 2017.DOI
- [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]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]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]Siyu Kong and AS Lewis. The cost of nonconvexity in deterministic nonsmooth optimization. Mathematics of Operations Research, 2023.arxiv.org/abs/2210.00652
- [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]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]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]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]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]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]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]Frank H Clarke. Generalized gradients and applications. Transactions of the American Mathematical Society, 205:247–262, 1975.DOI
- [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]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







