中文
相关论文

相关论文: Implicit Bias of Gradient Descent for Mean Squared…

200 篇论文

Training neural networks with first order optimisation methods is at the core of the empirical success of deep learning. The scale of initialisation is a crucial factor, as small initialisations are generally associated to a feature…

机器学习 · 计算机科学 2025-09-16 Etienne Boursier , Nicolas Flammarion

We prove that two-layer (Leaky)ReLU networks initialized by e.g. the widely used method proposed by He et al. (2015) and trained using gradient descent on a least-squares loss are not universally consistent. Specifically, we describe a…

机器学习 · 统计学 2022-06-10 David Holzmüller , Ingo Steinwart

We consider networks, trained via stochastic gradient descent to minimize $\ell_2$ loss, with the training labels perturbed by independent noise at each iteration. We characterize the behavior of the training dynamics near any parameter…

机器学习 · 计算机科学 2020-07-23 Guy Blanc , Neha Gupta , Gregory Valiant , Paul Valiant

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…

机器学习 · 计算机科学 2021-11-10 Kaifeng Lyu , Zhiyuan Li , Runzhe Wang , Sanjeev Arora

Neural networks are widely used to approximate unknown functions in control. A common neural network architecture uses a single hidden layer (i.e. a shallow network), in which the input parameters are fixed in advance and only the output…

机器学习 · 计算机科学 2024-10-08 Andrew Lamperski , Siddharth Salapaka

The implicit bias induced by the training of neural networks has become a topic of rigorous study. In the limit of gradient flow and gradient descent with appropriate step size, it has been shown that when one trains a deep linear network…

机器学习 · 计算机科学 2022-04-27 Thien Le , Stefanie Jegelka

Deep neural networks are often trained in the over-parametrized regime (i.e. with far more parameters than training examples), and understanding why the training converges to solutions that generalize remains an open problem. Several…

机器学习 · 统计学 2018-03-23 Hartmut Maennel , Olivier Bousquet , Sylvain Gelly

We study the gradient descent (GD) dynamics of a depth-2 linear neural network with a single input and output. We show that GD converges at an explicit linear rate to a global minimum of the training loss, even with a large stepsize --…

机器学习 · 计算机科学 2025-01-22 Pierfrancesco Beneventano , Blake Woodworth

We prove that the gradient descent training of a two-layer neural network on empirical or population risk may not decrease population risk at an order faster than $t^{-4/(d-2)}$ under mean field scaling. Thus gradient descent training for…

机器学习 · 计算机科学 2020-05-22 Stephan Wojtowytsch , Weinan E

We study the implicit regularization imposed by gradient descent for learning multi-layer homogeneous functions including feed-forward fully connected and convolutional deep neural networks with linear, ReLU or Leaky ReLU activation. We…

机器学习 · 计算机科学 2018-11-01 Simon S. Du , Wei Hu , Jason D. Lee

Gradient descent (GD) type optimization schemes are the standard methods to train artificial neural networks (ANNs) with rectified linear unit (ReLU) activation. Such schemes can be considered as discretizations of gradient flows (GFs)…

机器学习 · 计算机科学 2022-09-27 Arnulf Jentzen , Adrian Riekert

Despite classical statistical theory predicting severe overfitting, modern massively overparameterized neural networks still generalize well. This unexpected property is attributed to the network's so-called implicit bias, which describes…

机器学习 · 计算机科学 2025-03-14 Justin Sahs , Ryan Pyle , Fabio Anselmi , Ankit Patel

We study the conjectured relationship between the implicit regularization in neural networks, trained with gradient-based methods, and rank minimization of their weight matrices. Previously, it was proved that for linear networks (of depth…

机器学习 · 计算机科学 2024-12-24 Nadav Timor , Gal Vardi , Ohad Shamir

We investigate filter level sparsity that emerges in convolutional neural networks (CNNs) which employ Batch Normalization and ReLU activation, and are trained with adaptive gradient descent techniques and L2 regularization or weight decay.…

机器学习 · 计算机科学 2019-04-08 Dushyant Mehta , Kwang In Kim , Christian Theobalt

We analyze a simple one-hidden-layer neural network with ReLU activation functions and fixed biases, with one-dimensional input and output. We study both continuous and discrete versions of the model, and we rigorously prove the convergence…

机器学习 · 计算机科学 2026-04-10 Fabricio Macià , Shu Nakamura

We study the implicit bias towards low-rank weight matrices when training neural networks (NN) with Weight Decay (WD). We prove that when a ReLU NN is sufficiently trained with Stochastic Gradient Descent (SGD) and WD, its weight matrix is…

机器学习 · 计算机科学 2024-10-04 Ke Chen , Chugang Yi , Haizhao Yang

In deep learning, it is common to use more network parameters than training points. In such scenarioof over-parameterization, there are usually multiple networks that achieve zero training error so that thetraining algorithm induces an…

机器学习 · 计算机科学 2023-08-22 Hung-Hsu Chou , Carsten Gieshoff , Johannes Maly , Holger Rauhut

The largest eigenvalue of the Hessian, or sharpness, of neural networks is a key quantity to understand their optimization dynamics. In this paper, we study the sharpness of deep linear networks for univariate regression. Minimizers can…

机器学习 · 统计学 2024-10-29 Pierre Marion , Lénaïc Chizat

We give a simple proof for the global convergence of gradient descent in training deep ReLU networks with the standard square loss, and show some of its improvements over the state-of-the-art. In particular, while prior works require all…

机器学习 · 计算机科学 2021-06-14 Quynh Nguyen

We consider the problem of learning a one-hidden-layer neural network with non-overlapping convolutional layer and ReLU activation, i.e., $f(\mathbf{Z}, \mathbf{w}, \mathbf{a}) = \sum_j a_j\sigma(\mathbf{w}^T\mathbf{Z}_j)$, in which both…

机器学习 · 计算机科学 2018-06-18 Simon S. Du , Jason D. Lee , Yuandong Tian , Barnabas Poczos , Aarti Singh