English
Related papers

Related papers: Tight conditions for when the NTK approximation is…

200 papers

The neural tangent kernel (NTK) has garnered significant attention as a theoretical framework for describing the behavior of large-scale neural networks. Kernel methods are theoretically well-understood and as a result enjoy algorithmic…

Machine Learning · Computer Science 2024-05-30 Jonathan Wenger , Felix Dangel , Agustinus Kristiadi

We introduce the neural tangent kernel (NTK) regime for two-layer neural operators and analyze their generalization properties. For early-stopped gradient descent (GD), we derive fast convergence rates that are known to be minimax optimal…

Machine Learning · Statistics 2024-12-24 Mike Nguyen , Nicole Mücke

A recent line of work has analyzed the theoretical properties of deep neural networks via the Neural Tangent Kernel (NTK). In particular, the smallest eigenvalue of the NTK has been related to the memorization capacity, the global…

Machine Learning · Statistics 2022-08-23 Quynh Nguyen , Marco Mondelli , Guido Montufar

A recent breakthrough in deep learning theory shows that the training of over-parameterized deep neural networks can be characterized by a kernel function called \textit{neural tangent kernel} (NTK). However, it is known that this type of…

Machine Learning · Computer Science 2020-10-07 Zixiang Chen , Yuan Cao , Quanquan Gu , Tong Zhang

The Neural Tangent Kernel (NTK) is the wide-network limit of a kernel defined using neural networks at initialization, whose embedding is the gradient of the output of the network with respect to its parameters. We study the "after kernel",…

Machine Learning · Computer Science 2021-12-14 Philip M. Long

Recent research shows that the following two models are equivalent: (a) infinitely wide neural networks (NNs) trained under l2 loss by gradient descent with infinitesimally small learning rate (b) kernel regression with respect to so-called…

Machine Learning · Computer Science 2019-10-29 Sanjeev Arora , Simon S. Du , Zhiyuan Li , Ruslan Salakhutdinov , Ruosong Wang , Dingli Yu

The Neural Tangent Kernel (NTK), defined as $\Theta_\theta^f(x_1, x_2) = \left[\partial f(\theta, x_1)\big/\partial \theta\right] \left[\partial f(\theta, x_2)\big/\partial \theta\right]^T$ where $\left[\partial f(\theta,…

Machine Learning · Computer Science 2022-06-20 Roman Novak , Jascha Sohl-Dickstein , Samuel S. Schoenholz

The Neural Tangent Kernel (NTK) has recently attracted intense study, as it describes the evolution of an over-parameterized Neural Network (NN) trained by gradient descent. However, it is now well-known that gradient descent is not always…

Machine Learning · Computer Science 2021-03-23 Lei Tan , Shutong Wu , Xiaolin Huang

Recent works have partly attributed the generalization ability of over-parameterized neural networks to frequency bias -- networks trained with gradient descent on data drawn from a uniform distribution find a low frequency fit before high…

Machine Learning · Computer Science 2020-03-11 Ronen Basri , Meirav Galun , Amnon Geifman , David Jacobs , Yoni Kasten , Shira Kritchman

Small generalization errors of over-parameterized neural networks (NNs) can be partially explained by the frequency biasing phenomenon, where gradient-based algorithms minimize the low-frequency misfit before reducing the high-frequency…

Machine Learning · Computer Science 2022-09-27 Annan Yu , Yunan Yang , Alex Townsend

At initialization, artificial neural networks (ANNs) are equivalent to Gaussian processes in the infinite-width limit, thus connecting them to kernel methods. We prove that the evolution of an ANN during training can also be described by a…

Machine Learning · Computer Science 2020-02-11 Arthur Jacot , Franck Gabriel , Clément Hongler

We analyze the generalization properties of two-layer neural networks in the neural tangent kernel (NTK) regime, trained with gradient descent (GD). For early stopped GD we derive fast rates of convergence that are known to be minimax…

Machine Learning · Statistics 2023-09-18 Mike Nguyen , Nicole Mücke

Two key challenges facing modern deep learning are mitigating deep networks' vulnerability to adversarial attacks and understanding deep learning's generalization capabilities. Towards the first issue, many defense strategies have been…

Machine Learning · Computer Science 2022-10-24 Noel Loo , Ramin Hasani , Alexander Amini , Daniela Rus

It has become standard to solve NLP tasks by fine-tuning pre-trained language models (LMs), especially in low-data settings. There is minimal theoretical understanding of empirical success, e.g., why fine-tuning a model with $10^8$ or more…

Machine Learning · Computer Science 2023-06-07 Sadhika Malladi , Alexander Wettig , Dingli Yu , Danqi Chen , Sanjeev Arora

Recent works have examined theoretical and empirical properties of wide neural networks trained in the Neural Tangent Kernel (NTK) regime. Given that biological neural networks are much wider than their artificial counterparts, we consider…

Machine Learning · Computer Science 2022-07-14 Akhilan Boopathy , Ila Fiete

There are currently two parameterizations used to derive fixed kernels corresponding to infinite width neural networks, the NTK (Neural Tangent Kernel) parameterization and the naive standard parameterization. However, the extrapolation of…

Machine Learning · Computer Science 2020-04-21 Jascha Sohl-Dickstein , Roman Novak , Samuel S. Schoenholz , Jaehoon Lee

$Q$-learning with function approximation is one of the most empirically successful while theoretically mysterious reinforcement learning (RL) algorithms, and was identified in Sutton (1999) as one of the most important theoretical open…

Machine Learning · Computer Science 2022-05-04 Zaiwei Chen , John Paul Clarke , Siva Theja Maguluri

The performance of the data-dependent neural tangent kernel (NTK; Jacot et al. (2018)) associated with a trained deep neural network (DNN) often matches or exceeds that of the full network. This implies that DNN training via gradient…

Machine Learning · Computer Science 2025-05-22 Johannes Schwab , Bryan Kelly , Semyon Malamud , Teng Andrea Xu

We prove the precise scaling, at finite depth and width, for the mean and variance of the neural tangent kernel (NTK) in a randomly initialized ReLU network. The standard deviation is exponential in the ratio of network depth to width.…

Machine Learning · Computer Science 2019-09-16 Boris Hanin , Mihai Nica

This work studies the neural tangent kernel (NTK) of the deep equilibrium (DEQ) model, a practical ``infinite-depth'' architecture which directly computes the infinite-depth limit of a weight-tied network via root-finding. Even though the…

Machine Learning · Computer Science 2023-10-24 Zhili Feng , J. Zico Kolter