中文
相关论文

相关论文: Learning One-hidden-layer ReLU Networks via Gradie…

200 篇论文

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

We develop exact representations of training two-layer neural networks with rectified linear units (ReLUs) in terms of a single convex program with number of variables polynomial in the number of training samples and the number of hidden…

机器学习 · 计算机科学 2020-08-18 Mert Pilanci , Tolga Ergen

ReLU (rectified linear units) neural network has received significant attention since its emergence. In this paper, a univariate ReLU (UReLU) neural network is proposed to both modelling the nonlinear dynamic system and revealing insights…

系统与控制 · 电气工程与系统科学 2020-03-06 Xinglong Liang , Jun Xu

In the effort to learn from extensive collections of distributed data, federated learning has emerged as a promising approach for preserving privacy by using a gradient-sharing mechanism instead of exchanging raw data. However, recent…

We study feature learning in two-layer neural networks within the linear-width regime, where the number of hidden neurons, sample size, and input dimension scale proportionally. While recent work has analyzed feature learning via a single…

机器学习 · 统计学 2026-05-25 Behrad Moniri , Hamed Hassani

Recent theoretical work has demonstrated that deep neural networks have superior performance over shallow networks, but their training is more difficult, e.g., they suffer from the vanishing gradient problem. This problem can be typically…

机器学习 · 统计学 2021-11-03 Lu Lu , Yanhui Su , George Em Karniadakis

While deep learning has outperformed other methods for various tasks, theoretical frameworks that explain its reason have not been fully established. To address this issue, we investigate the excess risk of two-layer ReLU neural networks in…

机器学习 · 统计学 2022-06-07 Shunta Akiyama , Taiji Suzuki

In many numerical simulations stochastic gradient descent (SGD) type optimization methods perform very effectively in the training of deep neural networks (DNNs) but till this day it remains an open problem of research to provide a…

机器学习 · 计算机科学 2023-06-26 Martin Hutzenthaler , Arnulf Jentzen , Katharina Pohl , Adrian Riekert , Luca Scarpa

Understanding the implicit regularization (or implicit bias) of gradient descent has recently been a very active research area. However, the implicit regularization in nonlinear neural networks is still poorly understood, especially for…

机器学习 · 计算机科学 2021-06-09 Gal Vardi , Ohad Shamir

In this article we study fully-connected feedforward deep ReLU ANNs with an arbitrarily large number of hidden layers and we prove convergence of the risk of the GD optimization method with random initializations in the training of such…

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

Recent works have empirically analyzed in-context learning and shown that transformers trained on synthetic linear regression tasks can learn to implement ridge regression, which is the Bayes-optimal predictor, given sufficient capacity…

机器学习 · 计算机科学 2023-07-10 Arvind Mahankali , Tatsunori B. Hashimoto , Tengyu Ma

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

In this paper, we explore some basic questions on the complexity of training neural networks with ReLU activation function. We show that it is NP-hard to train a two-hidden layer feedforward ReLU neural network. If dimension of the input…

计算复杂性 · 计算机科学 2020-11-05 Digvijay Boob , Santanu S. Dey , Guanghui Lan

Very deep convolutional neural networks introduced new problems like vanishing gradient and degradation. The recent successful contributions towards solving these problems are Residual and Highway Networks. These networks introduce skip…

计算机视觉与模式识别 · 计算机科学 2016-10-06 Anish Shah , Eashan Kadam , Hena Shah , Sameer Shinde , Sandip Shingade

In recent years, stochastic gradient descent (SGD) based techniques has become the standard tools for training neural networks. However, formal theoretical understanding of why SGD can train neural networks in practice is largely missing.…

机器学习 · 计算机科学 2017-11-03 Yuanzhi Li , Yang Yuan

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

Overparametrized neural networks trained by gradient descent (GD) can provably overfit any training data. However, the generalization guarantee may not hold for noisy data. From a nonparametric perspective, this paper studies how well…

机器学习 · 统计学 2021-09-28 Tianyang Hu , Wenjia Wang , Cong Lin , Guang Cheng

We study the learning ability of linear recurrent neural networks with Gradient Descent. We prove the first theoretical guarantee on linear RNNs to learn any stable linear dynamic system using any a large type of loss functions. For an…

机器学习 · 计算机科学 2023-10-24 Lifu Wang , Tianyu Wang , Shengwei Yi , Bo Shen , Bo Hu , Xing Cao

Understanding the generalization properties of neural networks on simple input-output distributions is key to explaining their performance on real datasets. The classical teacher-student setting, where a network is trained on data generated…

Despite their prevalence in neural networks we still lack a thorough theoretical characterization of ReLU layers. This paper aims to further our understanding of ReLU layers by studying how the activation function ReLU interacts with the…

机器学习 · 计算机科学 2019-08-13 Sören Dittmer , Emily J. King , Peter Maass