English
Related papers

Related papers: Tensor Programs II: Neural Tangent Kernel for Any …

200 papers

We establish explicit dynamics for neural networks whose training objective has a regularising term that constrains the parameters to remain close to their initial value. This keeps the network in a lazy training regime, where the dynamics…

Machine Learning · Statistics 2023-12-21 Eugenio Clerico , Benjamin Guedj

It is well understood that neural networks with carefully hand-picked weights provide powerful function approximation and that they can be successfully trained in over-parametrized regimes. Since over-parametrization ensures zero training…

Machine Learning · Computer Science 2024-05-21 G. Welper

In this paper, we study the generalization performance of min $\ell_2$-norm overfitting solutions for the neural tangent kernel (NTK) model of a two-layer neural network with ReLU activation that has no bias term. We show that, depending on…

Machine Learning · Computer Science 2023-03-09 Peizhong Ju , Xiaojun Lin , Ness B. Shroff

Recent theoretical work has established connections between over-parametrized neural networks and linearized models governed by he Neural Tangent Kernels (NTKs). NTK theory leads to concrete convergence and generalization results, yet the…

Machine Learning · Computer Science 2020-02-18 Yu Bai , Jason D. Lee

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…

Neural and Evolutionary Computing · Computer Science 2021-05-11 Greg Yang

A recent trend in explainable AI research has focused on surrogate modeling, where neural networks are approximated as simpler ML algorithms such as kernel machines. A second trend has been to utilize kernel functions in various…

Machine Learning · Computer Science 2024-03-13 Andrew Engel , Zhichao Wang , Natalie S. Frank , Ioana Dumitriu , Sutanay Choudhury , Anand Sarwate , Tony Chiang

In wireless communications, estimation of channels in OFDM systems spans frequency and time, which relies on sparse collections of pilot data, posing an ill-posed inverse problem. Moreover, deep learning estimators require large amounts of…

Machine Learning · Computer Science 2025-04-14 Mohammed Mallik , Guillaume Villemaud

We are interested in estimating the uncertainties of deep neural networks, which play an important role in many scientific and engineering problems. In this paper, we present a striking new finding that an ensemble of neural networks with…

Machine Learning · Computer Science 2022-12-05 Jayaraman J. Thiagarajan , Rushil Anirudh , Vivek Narayanaswamy , Peer-Timo Bremer

There has been a recent surge of interest in understanding the convergence of gradient descent (GD) and stochastic gradient descent (SGD) in overparameterized neural networks. Most previous works assume that the training data is provided a…

Machine Learning · Statistics 2021-05-04 Jiaming Xu , Hanjing Zhu

Recently, it was proved that group equivariance emerges in ensembles of neural networks as the result of full augmentation in the limit of infinitely wide neural networks (neural tangent kernel limit). In this paper, we extend this result…

Machine Learning · Computer Science 2025-12-19 Oskar Nordenfors , Axel Flinth

To better understand the theoretical behavior of large neural networks, several works have analyzed the case where a network's width tends to infinity. In this regime, the effect of random initialization and the process of training a neural…

Machine Learning · Computer Science 2022-01-14 Florian Juengermann , Maxime Laasri , Marius Merkle

A longstanding goal in the theory of deep learning is to characterize the conditions under which a given neural network architecture will be trainable, and if so, how well it might generalize to unseen data. In this work, we provide such a…

Machine Learning · Computer Science 2020-07-14 Lechao Xiao , Jeffrey Pennington , Samuel S. Schoenholz

Modern deep neural networks (DNNs) are extremely powerful; however, this comes at the price of increased depth and having more parameters per layer, making their training and inference more computationally challenging. In an attempt to…

Machine Learning · Statistics 2024-03-04 Lingyu Gu , Yongqi Du , Yuan Zhang , Di Xie , Shiliang Pu , Robert C. Qiu , Zhenyu Liao

We provide quantitative bounds measuring the $L^2$ difference in function space between the trajectory of a finite-width network trained on finitely many samples from the idealized kernel dynamics of infinite width and infinite data. An…

Machine Learning · Statistics 2022-10-18 Benjamin Bowman , Guido Montufar

Natural gradients have been widely studied from both theoretical and empirical perspectives, and it is commonly believed that natural gradients have advantages over standard (Euclidean) gradients in capturing the intrinsic geometric…

Machine Learning · Computer Science 2025-09-30 Qinxun Bai , Steven Rosenberg , Wei Xu

In this paper, we provide the first precise distributional characterization of gradient descent iterates for general multi-layer neural networks under the canonical single-index regression model, in the `finite-width proportional regime'…

Machine Learning · Computer Science 2025-05-09 Qiyang Han , Masaaki Imaizumi

Deep neural networks (NNs) encounter scalability limitations when confronted with a vast array of neurons, thereby constraining their achievable network depth. To address this challenge, we propose an integration of tensor networks (TN)…

Disordered Systems and Neural Networks · Physics 2024-08-20 Saeed S. Jahromi , Roman Orus

Recently, theoretical analyses of deep neural networks have broadly focused on two directions: 1) Providing insight into neural network training by SGD in the limit of infinite hidden-layer width and infinitesimally small learning rate…

Machine Learning · Computer Science 2023-09-27 Rajat Vadiraj Dwaraknath , Tolga Ergen , Mert Pilanci

The paper contains approximation guarantees for neural networks that are trained with gradient flow, with error measured in the continuous $L_2(\mathbb{S}^{d-1})$-norm on the $d$-dimensional unit sphere and targets that are Sobolev smooth.…

Machine Learning · Computer Science 2023-09-12 G. Welper

A theory of neural networks (NNs) built upon collective variables would provide scientists with the tools to better understand the learning process at every stage. In this work, we introduce two such variables, the entropy and the trace of…

Machine Learning · Computer Science 2023-05-03 Samuel Tovey , Sven Krippendorf , Konstantin Nikolaou , Christian Holm