Near-optimal estimates for the \ell^p-Lipschitz constants of deep random ReLU neural networks
Authors: Sjoerd Dirksen, Patrick Finke, Paul Geuchen, Dominik Stöger, Felix Voigtlaender
Abstract
This paper studies the ℓp-Lipschitz constants of ReLU neural networks Φ:Rd→R with random parameters for p∈[1,∞]. The distribution of the weights follows a variant of the He initialization. In the case of zero-bias networks, we derive high probability upper and lower bounds for wide networks that differ at most by a factor that is logarithmic in the network's depth. Remarkably, the behavior of the ℓp-Lipschitz constant varies significantly between the regimes p∈[1,2) and p∈[2,∞]. For p∈[2,∞], the ℓp-Lipschitz constant behaves similarly to ∥g∥p′, where g∈Rd is a d-dimensional standard Gaussian vector and 1/p+1/p′=1. In contrast, for p∈[1,2), the ℓp-Lipschitz constant aligns more closely to ∥g∥2. We extend our analysis to networks with possibly non-zero biases drawn from arbitrary symmetric distributions. In this case, we obtain high probability upper and lower bounds that differ at most by a factor that is logarithmic in the network's width and linear in its depth.
We prove a depth hierarchy for ReLU neural networks in which every additional ReLU layer can save exponentially many neurons. For all k≥2, we construct a globally [0,1]-valued, 1-Lipschitz function realized by a depth-(k+1) network of width O(d4), whereas any depth-k network with unrestricted weights and width at most 2d(k−1)2d has squared L2 error at least 1/24 under an absolutely continuous distribution supported at exponential distance from the origin. To the best of our knowledge, this is the first exponential hierarchy across all adjacent fixed depths, and the first exponential separation for ReLU networks between two fixed depths whose shallower network has depth at least 3. The lower bound also immediately yields the corresponding hierarchy for exact computation. Moreover, the case k=2 gives a compactly supported separation between depths 3 and 2 with unrestricted shallow-network weights, answering a question raised by Safran, Eldan, and Shamir (2019). The distribution used in our construction nevertheless has all its mass at exponential radius, placing the hierarchy outside the regularity regime in which such a separation would imply major threshold-circuit lower bounds. We also prove an exact separation for a more regular target, which is globally [0,1]-valued and O(d)-Lipschitz and maps the unit hypercube onto [0,1]. It is computed by a polynomial-width depth-4 network, whereas any depth-3 network agreeing with it on the unit hypercube requires exponentially many first-layer neurons, even with unrestricted weights.
Bubeck, Li and Nagaraj conjectured that, for generic data, any two-layer neural network with m neurons that fits n noisy labels must have Lipschitz constant at least of order n/m, with no restriction on the size of the weights. Bubeck and Sellke proved a universal version of this law for Lipschitz-parameterized classes, but under a polynomial bound on the parameters; at depth three that boundedness hypothesis is genuinely necessary. The two-layer unbounded-weight case requires a different argument. We prove the conjectured law, up to one logarithmic factor, for every continuous piecewise-linear activation, in particular for ReLU networks. For data drawn uniformly from Sd−1, d≥3, or from N(0,Id/d), labels in [−1,1] with noise level σ2>0, and any width-m two-layer network with arbitrary real weights, biases and affine skip connection, fitting the data ε below the noise floor forces Lip(f)≥cεn/(mˉlog(Cmˉnd/ε)), mˉ=(K−1)m+1, with high probability. A realized-kink-count version holds on the same event: every realized two-layer piecewise-linear function with k(f)≤n distinct kink hyperplanes obeys the bound with mˉ replaced by k(f)+1, irrespective of how many redundant hidden units parameterize it. The proof replaces parameter-space covering, impossible for unbounded weights, by a function-space covering. The central deterministic ingredient is a rigidity lemma: on B2, and on Sd−1 for d≥3, the coefficient of each canonical kink is controlled by the Lipschitz constant of the realized function, because kinks on distinct hyperplanes cannot cancel at generic points. Rigidity genuinely fails at d=2, and an explicit two-layer ReLU interpolant with O(1) Lipschitz constant at width 2n matches the law at the overparameterized endpoint.
Recent studies have shown that smooth functions can be well approximated by ReLU neural networks with path norm constraint on the weights. We extend these results from uniform approximation to approximation in Sobolev norm. Specifically, we analyze how well Sobolev functions in Wn,p can be approximated by neural networks with width W, depth L and path norm bounded by K, when the approximation error is measured in the W1,p-norm. For shallow networks with depth L=1, we derive the approximation error bound O(max{W−(n−1)/d,K−(n−1)/(s−n)}), when the smoothness index satisfies n<s=(d+3)/2 and the input is d-dimensional. For deep networks, we remove the restriction on the smoothness by showing that the approximation bound O(K−(n−1)/(d+d/p+1)) holds if the width W and depth L are sufficiently large.