中文
相关论文

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

200 篇论文

We derive explicit equations governing the cumulative biases and weights in Deep Learning with ReLU activation function, based on gradient descent for the Euclidean cost in the input layer, and under the assumption that the weights are, in…

机器学习 · 计算机科学 2025-01-15 Thomas Chen

In this paper, we study the implicit bias of gradient descent for sparse regression. We extend results on regression with quadratic parametrization, which amounts to depth-2 diagonal linear networks, to more general depth-N networks, under…

机器学习 · 统计学 2021-10-28 Jiangyuan Li , Thanh V. Nguyen , Chinmay Hegde , Raymond K. W. Wong

Deep learning empirically achieves high performance in many applications, but its training dynamics has not been fully understood theoretically. In this paper, we explore theoretical analysis on training two-layer ReLU neural networks in a…

机器学习 · 统计学 2021-06-30 Shunta Akiyama , Taiji Suzuki

Existing analyses of optimization in deep learning are either continuous, focusing on (variants of) gradient flow, or discrete, directly treating (variants of) gradient descent. Gradient flow is amenable to theoretical analysis, but is…

机器学习 · 计算机科学 2021-12-30 Omer Elkabetz , Nadav Cohen

Deep learning models are often successfully trained using gradient descent, despite the worst case hardness of the underlying non-convex optimization problem. The key question is then under what conditions can one prove that optimization…

机器学习 · 计算机科学 2017-02-28 Alon Brutzkus , Amir Globerson

We prove linear convergence of gradient descent to a global optimum for the training of deep residual networks with constant layer width and smooth activation function. We show that if the trained weights, as a function of the layer index,…

机器学习 · 计算机科学 2023-01-26 Rama Cont , Alain Rossier , RenYuan Xu

Gradient descent (GD) type optimization methods are the standard instrument to train artificial neural networks (ANNs) with rectified linear unit (ReLU) activation. Despite the great success of GD type optimization methods in numerical…

最优化与控制 · 数学 2022-12-29 Arnulf Jentzen , Adrian Riekert

We study the overparametrization bounds required for the global convergence of stochastic gradient descent algorithm for a class of one hidden layer feed-forward neural networks, considering most of the activation functions used in…

机器学习 · 计算机科学 2022-11-17 Bartłomiej Polaczyk , Jacek Cyranka

It is important to understand how dropout, a popular regularization method, aids in achieving a good generalization solution during neural network training. In this work, we present a theoretical derivation of an implicit regularization of…

机器学习 · 计算机科学 2023-04-11 Zhongwang Zhang , Zhi-Qin John Xu

This paper studies the problem of training a two-layer ReLU network for binary classification using gradient flow with small initialization. We consider a training dataset with well-separated input vectors: Any pair of input data with the…

机器学习 · 计算机科学 2024-03-26 Hancheng Min , Enrique Mallada , René Vidal

The implicit bias of gradient-based training algorithms has been considered mostly beneficial as it leads to trained networks that often generalize well. However, Frei et al. (2023) show that such implicit bias can harm adversarial…

机器学习 · 计算机科学 2024-06-06 Hancheng Min , René Vidal

In this work, we study the implications of the implicit bias of gradient flow on generalization and adversarial robustness in ReLU networks. We focus on a setting where the data consists of clusters and the correlations between cluster…

机器学习 · 计算机科学 2023-11-01 Spencer Frei , Gal Vardi , Peter L. Bartlett , Nathan Srebro

Binary neural networks improve computationally efficiency of deep models with a large margin. However, there is still a performance gap between a successful full-precision training and binary training. We bring some insights about why this…

机器学习 · 计算机科学 2020-04-22 Xinlin Li , Vahid Partovi Nia

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…

机器学习 · 计算机科学 2019-05-28 Santosh Vempala , John Wilmes

The optimization algorithms are crucial in training physics-informed neural networks (PINNs), as unsuitable methods may lead to poor solutions. Compared to the common gradient descent (GD) algorithm, implicit gradient descent (IGD)…

机器学习 · 计算机科学 2025-08-04 Xianliang Xu , Ting Du , Wang Kong , Bin Shan , Ye Li , Zhongyi Huang

Although deep learning has shown its powerful performance in many applications, the mathematical principles behind neural networks are still mysterious. In this paper, we consider the problem of learning a one-hidden-layer neural network…

机器学习 · 计算机科学 2019-07-17 Shuhao Xia , Yuanming Shi

We show that gradient descent on full-width linear convolutional networks of depth $L$ converges to a linear predictor related to the $\ell_{2/L}$ bridge penalty in the frequency domain. This is in contrast to linearly fully connected…

机器学习 · 计算机科学 2019-01-14 Suriya Gunasekar , Jason Lee , Daniel Soudry , Nathan Srebro

Over the past years, there has been significant interest in understanding the implicit bias of gradient descent optimization and its connection to the generalization properties of overparametrized neural networks. Several works observed…

This paper shows that the implicit bias of gradient descent on linearly separable data is exactly characterized by the optimal solution of a dual optimization problem given by a smoothed margin, even for general losses. This is in contrast…

机器学习 · 计算机科学 2020-11-13 Ziwei Ji , Matus Telgarsky

We study the convergence of gradient flow for the training of deep neural networks. If Residual Neural Networks are a popular example of very deep architectures, their training constitutes a challenging optimization problem due notably to…

机器学习 · 计算机科学 2025-07-22 Raphaël Barboni , Gabriel Peyré , François-Xavier Vialard