English

A law of robustness for two-layer neural networks with arbitrary weights

Machine Learning 2026-07-08 v1 Probability Machine Learning

Abstract

Bubeck, Li and Nagaraj conjectured that, for generic data, any two-layer neural network with mm neurons that fits nn noisy labels must have Lipschitz constant at least of order n/m\sqrt{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 Sd1\mathbb{S}^{d-1}, d3d\ge3, or from N(0,Id/d)N(0,I_d/d), labels in [1,1][-1,1] with noise level σ2>0\sigma^2>0, and any width-mm two-layer network with arbitrary real weights, biases and affine skip connection, fitting the data ε\varepsilon below the noise floor forces Lip(f)cεn/(mˉlog(Cmˉnd/ε))\mathrm{Lip}(f)\ge c\,\varepsilon\sqrt{n/(\bar m\log(C\bar m nd/\varepsilon))}, mˉ=(K1)m+1\bar 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)nk(f)\le n distinct kink hyperplanes obeys the bound with mˉ\bar m replaced by k(f)+1k(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 B2B_2, and on Sd1\mathbb{S}^{d-1} for d3d\ge3, 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=2d=2, and an explicit two-layer ReLU interpolant with O(1)O(1) Lipschitz constant at width 2n2n matches the law at the overparameterized endpoint.

Keywords

Cite

@article{arxiv.2607.07778,
  title  = {A law of robustness for two-layer neural networks with arbitrary weights},
  author = {Yitzchak Shmalo},
  journal= {arXiv preprint arXiv:2607.07778},
  year   = {2026}
}