Tensor Programs III: Neural Matrix Laws
- URL: http://arxiv.org/abs/2009.10685v3
- Date: Sat, 8 May 2021 12:59:35 GMT
- Title: Tensor Programs III: Neural Matrix Laws
- Abstract summary: We show that (pre-activations) of a randomly NN become *independent* from the weights as the NN's widths tend to infinity, in the sense of freeness in random matrix theory.
We show FIP by proving a Master Theorem for any Kernel Program, as introduced in Yang [50,51], generalizing the Master Theorems proved in those works.
- Score: 23.166601000371728
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: In a neural network (NN), *weight matrices* linearly transform inputs into
*preactivations* that are then transformed nonlinearly into *activations*. A
typical NN interleaves multitudes of such linear and nonlinear transforms to
express complex functions. Thus, the (pre-)activations depend on the weights in
an intricate manner. We show that, surprisingly, (pre-)activations of a
randomly initialized NN become *independent* from the weights as the NN's
widths tend to infinity, in the sense of asymptotic freeness in random matrix
theory. We call this the Free Independence Principle (FIP), which has these
consequences: 1) It rigorously justifies the calculation of asymptotic Jacobian
singular value distribution of an NN in Pennington et al. [36,37], essential
for training ultra-deep NNs [48]. 2) It gives a new justification of gradient
independence assumption used for calculating the Neural Tangent Kernel of a
neural network. FIP and these results hold for any neural architecture. We show
FIP by proving a Master Theorem for any Tensor Program, as introduced in Yang
[50,51], generalizing the Master Theorems proved in those works. As warmup
demonstrations of this new Master Theorem, we give new proofs of the semicircle
and Marchenko-Pastur laws, which benchmarks our framework against these
fundamental mathematical results.
Related papers
- Finite-Width Neural Tangent Kernels from Feynman Diagrams [3.6731536660959985]
We introduce Feynman diagrams for computing finite-width corrections to NTK statistics.<n>We demonstrate the feasibility of our framework by extending stability results for deep networks from preactivations to NTKs.
arXiv Detail & Related papers (2025-08-15T15:02:40Z) - FACT: the Features At Convergence Theorem for neural networks [11.578851413831247]
We prove the Features at Convergence Theorem (FACT), which gives a self-consistency equation that neural network weights satisfy at convergence when trained with nonzero weight decay.<n>We validate this relation empirically, showing that neural features indeed satisfy the FACT at convergence.<n>By modifying the "Recursive Feature Machines" of Radhakrishnan et al. 2024 so that they obey the FACT, we arrive at a new learning algorithm, FACT-RFM.
arXiv Detail & Related papers (2025-07-08T03:52:48Z) - Genus expansion for non-linear random matrix ensembles with applications to neural networks [3.801509221714223]
We present a unified approach to studying certain non-linear random matrix ensembles and associated random neural networks.<n>We use a novel series expansion for neural networks which generalizes Fa'a di Bruno's formula to an arbitrary number of compositions.<n>As an application, we prove several results about neural networks with random weights.
arXiv Detail & Related papers (2024-07-11T12:58:07Z) - Neural Network Verification with Branch-and-Bound for General Nonlinearities [63.39918329535165]
Branch-and-bound (BaB) is among the most effective techniques for neural network (NN) verification.<n>We develop a general framework, named GenBaB, to conduct BaB on general nonlinearities to verify NNs with general architectures.<n>Our framework also allows the verification of general nonlinear graphs and enables verification applications beyond simple NNs.
arXiv Detail & Related papers (2024-05-31T17:51:07Z) - Multi-layer random features and the approximation power of neural networks [4.178980693837599]
We prove that a reproducing kernel Hilbert space contains only functions that can be approximated by the architecture.
We show that if eigenvalues of the integral operator of the NNGP decay slower than $k-n-frac23$ where $k$ is an order of an eigenvalue, our theorem guarantees a more succinct neural network approximation than Barron's theorem.
arXiv Detail & Related papers (2024-04-26T14:57:56Z) - Benign Overfitting in Deep Neural Networks under Lazy Training [72.28294823115502]
We show that when the data distribution is well-separated, DNNs can achieve Bayes-optimal test error for classification.
Our results indicate that interpolating with smoother functions leads to better generalization.
arXiv Detail & Related papers (2023-05-30T19:37:44Z) - A Functional-Space Mean-Field Theory of Partially-Trained Three-Layer
Neural Networks [49.870593940818715]
We study the infinite-width limit of a type of three-layer NN model whose first layer is random and fixed.
Our theory accommodates different scaling choices of the model, resulting in two regimes of the MF limit that demonstrate distinctive behaviors.
arXiv Detail & Related papers (2022-10-28T17:26:27Z) - Extrapolation and Spectral Bias of Neural Nets with Hadamard Product: a
Polynomial Net Study [55.12108376616355]
The study on NTK has been devoted to typical neural network architectures, but is incomplete for neural networks with Hadamard products (NNs-Hp)
In this work, we derive the finite-width-K formulation for a special class of NNs-Hp, i.e., neural networks.
We prove their equivalence to the kernel regression predictor with the associated NTK, which expands the application scope of NTK.
arXiv Detail & Related papers (2022-09-16T06:36:06Z) - On the Neural Tangent Kernel Analysis of Randomly Pruned Neural Networks [91.3755431537592]
We study how random pruning of the weights affects a neural network's neural kernel (NTK)
In particular, this work establishes an equivalence of the NTKs between a fully-connected neural network and its randomly pruned version.
arXiv Detail & Related papers (2022-03-27T15:22:19Z) - On the Equivalence between Neural Network and Support Vector Machine [23.174679357972984]
The dynamics of an infinitely wide neural network (NN) trained by gradient descent can be characterized by Tangent Neural Kernel (NTK)
We establish the equivalence between NN and support vector machine (SVM)
Our main theoretical results include establishing the equivalence between NN and a broad family of $ell$ regularized KMs with finite-width bounds.
arXiv Detail & Related papers (2021-11-11T06:05:00Z) - Nonperturbative renormalization for the neural network-QFT
correspondence [0.0]
We study the concepts of locality and power-counting in this context.
We provide an analysis in terms of the nonperturbative renormalization group using the Wetterich-Morris equation.
Our aim is to provide a useful formalism to investigate neural networks behavior beyond the large-width limit.
arXiv Detail & Related papers (2021-08-03T10:36:04Z) - How Neural Networks Extrapolate: From Feedforward to Graph Neural
Networks [80.55378250013496]
We study how neural networks trained by gradient descent extrapolate what they learn outside the support of the training distribution.
Graph Neural Networks (GNNs) have shown some success in more complex tasks.
arXiv Detail & Related papers (2020-09-24T17:48:59Z) - Bayesian Perceptron: Towards fully Bayesian Neural Networks [5.5510642465908715]
Training and predictions of a perceptron are performed within the Bayesian inference framework in closed-form.
The weights and the predictions of the perceptron are considered Gaussian random variables.
This approach requires no computationally expensive gradient calculations and further allows sequential learning.
arXiv Detail & Related papers (2020-09-03T15:08:49Z) - Provably Efficient Neural Estimation of Structural Equation Model: An
Adversarial Approach [144.21892195917758]
We study estimation in a class of generalized Structural equation models (SEMs)
We formulate the linear operator equation as a min-max game, where both players are parameterized by neural networks (NNs), and learn the parameters of these neural networks using a gradient descent.
For the first time we provide a tractable estimation procedure for SEMs based on NNs with provable convergence and without the need for sample splitting.
arXiv Detail & Related papers (2020-07-02T17:55:47Z) - Banach Space Representer Theorems for Neural Networks and Ridge Splines [17.12783792226575]
We develop a variational framework to understand the properties of the functions learned by neural networks fit to data.
We derive a representer theorem showing that finite-width, single-hidden layer neural networks are solutions to inverse problems.
arXiv Detail & Related papers (2020-06-10T02:57:37Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.