English
Related papers

Related papers: Fast Finite Width Neural Tangent Kernel

200 papers

While deep learning has achieved remarkable success across a wide range of applications, its theoretical understanding of representation learning remains limited. Deep neural kernels provide a principled framework to interpret…

Machine Learning · Computer Science 2025-11-11 Yong-Ming Tian , Shuang Liang , Shao-Qun Zhang , Feng-Lei Fan

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

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

Recent research has been focused on two different approaches to studying neural networks training in the limit of infinite width (1) a mean-field (MF) and (2) a constant neural tangent kernel (NTK) approximations. These two approaches have…

Machine Learning · Computer Science 2020-10-23 Eugene A. Golikov

We prove that a randomly initialized neural network of *any architecture* has its Tangent Kernel (NTK) converge to a deterministic limit, as the network widths tend to infinity. We demonstrate how to calculate this limit. In prior…

Machine Learning · Statistics 2020-12-01 Greg Yang

The Neural Tangent Kernel (NTK) characterizes how a model's state evolves over Gradient Descent. Computing the full NTK matrix is often infeasible, especially for recurrent architectures. Here, we introduce a matrix-free perspective, using…

Machine Learning · Computer Science 2025-11-17 James Hazelden

The Neural Tangent Kernel (NTK) has emerged as a powerful tool to provide memorization, optimization and generalization guarantees in deep neural networks. A line of work has studied the NTK spectrum for two-layer and deep networks with at…

Machine Learning · Statistics 2023-05-23 Simone Bombari , Mohammad Hossein Amani , Marco Mondelli

The Neural Tangent Kernel (NTK) characterizes the behavior of infinitely-wide neural networks trained under least squares loss by gradient descent. Recent works also report that NTK regression can outperform finitely-wide neural networks…

Machine Learning · Computer Science 2021-12-09 Amir Zandieh , Insu Han , Haim Avron , Neta Shoham , Chaewon Kim , Jinwoo Shin

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

Yang (2020a) recently showed that the Neural Tangent Kernel (NTK) at initialization has an infinite-width limit for a large class of architectures including modern staples such as ResNet and Transformers. However, their analysis does not…

Machine Learning · Computer Science 2021-05-11 Greg Yang , Etai Littwin

Motivated by both theory and practice, we study how random pruning of the weights affects a neural network's neural tangent kernel (NTK). In particular, this work establishes an equivalence of the NTKs between a fully-connected neural…

Machine Learning · Computer Science 2023-03-21 Hongru Yang , Zhangyang Wang

Common infinite-width architectures such as Neural Tangent Kernels (NTKs) have historically shown weak performance compared to finite models. This is usually attributed to the absence of feature learning. We show that this explanation is…

Machine Learning · Computer Science 2024-10-25 Luke Sernau

We perform a careful, thorough, and large scale empirical study of the correspondence between wide neural networks and kernel methods. By doing so, we resolve a variety of open questions related to the study of infinitely wide neural…

Machine Learning · Computer Science 2020-09-09 Jaehoon Lee , Samuel S. Schoenholz , Jeffrey Pennington , Ben Adlam , Lechao Xiao , Roman Novak , Jascha Sohl-Dickstein

The prevailing thinking is that orthogonal weights are crucial to enforcing dynamical isometry and speeding up training. The increase in learning speed that results from orthogonal initialization in linear networks has been well-proven.…

Machine Learning · Computer Science 2021-07-22 Wei Huang , Weitao Du , Richard Yi Da Xu

Neural Tangents is a library designed to enable research into infinite-width neural networks. It provides a high-level API for specifying complex and hierarchical neural network architectures. These networks can then be trained and…

Given the complexity of genetic risk prediction, there is a critical need for the development of novel methodologies that can effectively capture intricate genotype--phenotype relationships (e.g., nonlinear) while remaining statistically…

Applications · Statistics 2025-10-03 Heng Ge , Qing Lu

The NTK is a widely used tool in the theoretical analysis of deep learning, allowing us to look at supervised deep neural networks through the lenses of kernel regression. Recently, several works have investigated kernel models for…

Machine Learning · Computer Science 2025-05-06 Maximilian Fleissner , Gautham Govind Anil , Debarghya Ghoshdastidar

Recently, quantum neural networks or quantum-classical neural networks (qcNN) have been actively studied, as a possible alternative to the conventional classical neural network (cNN), but their practical and theoretically-guaranteed…

Quantum Physics · Physics 2023-12-12 Kouhei Nakaji , Hiroyuki Tezuka , Naoki Yamamoto

A rising trend in theoretical deep learning is to understand why deep learning works through Neural Tangent Kernel (NTK) [jgh18], a kernel method that is equivalent to using gradient descent to train a multi-layer infinitely-wide neural…

Machine Learning · Computer Science 2023-09-15 Lianke Qin , Zhao Song , Baocheng Sun

Deep neural networks have become essential for numerous applications due to their strong empirical performance such as vision, RL, and classification. Unfortunately, these networks are quite difficult to interpret, and this limits their…

Machine Learning · Computer Science 2021-10-12 Sina Alemohammad , Hossein Babaei , CJ Barberan , Naiming Liu , Lorenzo Luzi , Blake Mason , Richard G. Baraniuk