English
Related papers

Related papers: How many Neurons do we need? A refined Analysis fo…

200 papers

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 generalization mystery of overparametrized deep nets has motivated efforts to understand how gradient descent (GD) converges to low-loss solutions that generalize well. Real-life neural networks are initialized from small random values…

Machine Learning · Computer Science 2021-11-10 Kaifeng Lyu , Zhiyuan Li , Runzhe Wang , Sanjeev Arora

Recent theoretical works based on the neural tangent kernel (NTK) have shed light on the optimization and generalization of over-parameterized networks, and partially bridge the gap between their practical success and classical learning…

Machine Learning · Computer Science 2020-08-10 Kyung-Su Kim , Aurélie C. Lozano , Eunho Yang

Training neural networks is a challenging non-convex optimization problem, and backpropagation or gradient descent can get stuck in spurious local optima. We propose a novel algorithm based on tensor decomposition for guaranteed training of…

Machine Learning · Computer Science 2016-01-13 Majid Janzamin , Hanie Sedghi , Anima Anandkumar

Expressiveness and generalization of deep models was recently addressed via the connection between neural networks (NNs) and kernel learning, where first-order dynamics of NN during a gradient-descent (GD) optimization were related to…

Machine Learning · Computer Science 2020-04-21 Dmitry Kopitkov , Vadim Indelman

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

In training a neural network with gradient descent (GD), each iteration induces a linear operator that governs first-order updates to a model's internal state variables. We define this operator as the Global Empirical Neural Tangent Kernel…

Machine Learning · Computer Science 2026-05-12 James Hazelden , Laura Driscoll , Eli Shlizerman , Eric Shea-Brown

Recently, neural tangent kernel (NTK) has been used to explain the dynamics of learning parameters of neural networks, at the large width limit. Quantitative analyses of NTK give rise to network widths that are often impractical and incur…

Machine Learning · Computer Science 2022-10-11 Nir Ailon , Supratim Shit

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

Neural tangent kernels (NTKs) provide a theoretical regime to analyze the learning and generalization behavior of over-parametrized neural networks. For a supervised learning task, the association between the eigenvectors of the NTK kernel…

Machine Learning · Computer Science 2023-10-18 Shervin Khalafi , Saurabh Sihag , Alejandro Ribeiro

We consider optimizing two-layer neural networks in the mean-field regime where the learning dynamics of network weights can be approximated by the evolution in the space of probability measures over the weight parameters associated with…

Machine Learning · Computer Science 2022-10-19 Jingwei Zhang , Xunpeng Huang , Jincheng Yu

We study the complexity of training neural network models with one hidden nonlinear activation layer and an output weighted sum layer. We analyze Gradient Descent applied to learning a bounded target function on $n$ real-valued inputs. We…

Machine Learning · Computer Science 2019-05-28 Santosh Vempala , John Wilmes

The ability of learning useful features is one of the major advantages of neural networks. Although recent works show that neural network can operate in a neural tangent kernel (NTK) regime that does not allow feature learning, many works…

Machine Learning · Computer Science 2024-11-06 Mo Zhou , Rong Ge

Despite recent theoretical progress on the non-convex optimization of two-layer neural networks, it is still an open question whether gradient descent on neural networks without unnatural modifications can achieve better sample complexity…

Machine Learning · Computer Science 2023-10-10 Arvind Mahankali , Jeff Z. Haochen , Kefan Dong , Margalit Glasgow , Tengyu Ma

We study training one-hidden-layer ReLU networks in the neural tangent kernel (NTK) regime, where the networks' biases are initialized to some constant rather than zero. We prove that under such initialization, the neural network will have…

Machine Learning · Computer Science 2024-10-31 Hongru Yang , Ziyu Jiang , Ruizhe Zhang , Yingbin Liang , Zhangyang Wang

We study the training dynamics of shallow neural networks, in a two-timescale regime in which the stepsizes for the inner layer are much smaller than those for the outer layer. In this regime, we prove convergence of the gradient flow to a…

Optimization and Control · Mathematics 2023-10-26 Pierre Marion , Raphaël Berthier

We perform a study on the generalization ability of the wide two-layer ReLU neural network on $\mathbb{R}$. We first establish some spectral properties of the neural tangent kernel (NTK): $a)$ $K_{d}$, the NTK defined on $\mathbb{R}^{d}$,…

Machine Learning · Statistics 2023-02-14 Jianfa Lai , Manyun Xu , Rui Chen , Qian Lin

A biological neural network in the cortex forms a neural field. Neurons in the field have their own receptive fields, and connection weights between two neurons are random but highly correlated when they are in close proximity in receptive…

Machine Learning · Computer Science 2023-01-10 Kaito Watanabe , Kotaro Sakamoto , Ryo Karakida , Sho Sonoda , Shun-ichi Amari

Gradient descent yields zero training loss in polynomial time for deep neural networks despite non-convex nature of the objective function. The behavior of network in the infinite width limit trained by gradient descent can be described by…

Machine Learning · Computer Science 2023-05-29 Yuqing Li , Tao Luo , Nung Kwan Yip

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