中文
相关论文

相关论文: Learning One-hidden-layer Neural Networks with Lan…

200 篇论文

The clear understanding of the non-convex landscape of neural network is a complex incomplete problem. This paper studies the landscape of linear (residual) network, the simplified version of the nonlinear network. By treating the gradient…

代数几何 · 数学 2021-02-09 Xiuyi Yang

This work studies the global convergence and implicit bias of Gauss Newton's (GN) when optimizing over-parameterized one-hidden layer networks in the mean-field regime. We first establish a global convergence result for GN in the…

机器学习 · 计算机科学 2023-12-13 Michael Arbel , Romain Menegaux , Pierre Wolinski

We investigate the local spectral statistics of the loss surface Hessians of artificial neural networks, where we discover excellent agreement with Gaussian Orthogonal Ensemble statistics across several network architectures and datasets.…

机器学习 · 计算机科学 2021-12-28 Nicholas P Baskerville , Diego Granziol , Jonathan P Keating

We study the sample complexity of learning one-hidden-layer convolutional neural networks (CNNs) with non-overlapping filters. We propose a novel algorithm called approximate gradient descent for training CNNs, and show that, with high…

机器学习 · 计算机科学 2019-11-13 Yuan Cao , Quanquan Gu

One of the major concerns for neural network training is that the non-convexity of the associated loss functions may cause bad landscape. The recent success of neural networks suggests that their loss landscape is not too bad, but what…

机器学习 · 计算机科学 2023-07-19 Ruoyu Sun , Dawei Li , Shiyu Liang , Tian Ding , R Srikant

We address the structure identification and the uniform approximation of two fully nonlinear layer neural networks of the type $f(x)=1^T h(B^T g(A^T x))$ on $\mathbb R^d$ from a small number of query samples. We approach the problem by…

机器学习 · 计算机科学 2019-07-02 Massimo Fornasier , Timo Klock , Michael Rauchensteiner

For a large class of feature maps we provide a tight asymptotic characterisation of the test error associated with learning the readout layer, in the high-dimensional limit where the input dimension, hidden layer widths, and number of…

机器学习 · 统计学 2024-06-11 Dominik Schröder , Daniil Dmitriev , Hugo Cui , Bruno Loureiro

We prove the first superpolynomial lower bounds for learning one-layer neural networks with respect to the Gaussian distribution using gradient descent. We show that any classifier trained using gradient descent with respect to square-loss…

机器学习 · 计算机科学 2020-10-26 Surbhi Goel , Aravind Gollakota , Zhihan Jin , Sushrut Karmalkar , Adam Klivans

With the rise of big data analytics, multi-layer neural networks have surfaced as one of the most powerful machine learning methods. However, their theoretical mathematical properties are still not fully understood. Training a neural…

机器学习 · 计算机科学 2021-01-01 Victor Luo , Yazhen Wang , Glenn Fung

We study the problem of training a two-layer neural network (NN) of arbitrary width using stochastic gradient descent (SGD) where the input $\boldsymbol{x}\in \mathbb{R}^d$ is Gaussian and the target $y \in \mathbb{R}$ follows a…

We study the supervised learning problem under either of the following two models: (1) Feature vectors ${\boldsymbol x}_i$ are $d$-dimensional Gaussians and responses are $y_i = f_*({\boldsymbol x}_i)$ for $f_*$ an unknown quadratic…

机器学习 · 统计学 2019-06-24 Behrooz Ghorbani , Song Mei , Theodor Misiakiewicz , Andrea Montanari

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…

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

The first provably efficient algorithm for learning graph neural networks (GNNs) with one hidden layer for node information convolution is provided in this paper. Two types of GNNs are investigated, depending on whether labels are attached…

机器学习 · 计算机科学 2020-12-08 Qunwei Li , Shaofeng Zou , Wenliang Zhong

Object grasping is a crucial technology enabling robots to perceive and interact with the environment sufficiently. However, in practical applications, researchers are faced with missing or noisy ground truth while training the…

机器人学 · 计算机科学 2024-09-10 Yangfan Deng , Mengyao Zhang , Yong Zhao

Understanding the advantages of deep neural networks trained by gradient descent (GD) compared to shallow models remains an open theoretical challenge. In this paper, we introduce a class of target functions (single and multi-index Gaussian…

机器学习 · 统计学 2025-11-17 Yatin Dandi , Luca Pesce , Lenka Zdeborová , Florent Krzakala

We study the learning performance of gradient descent when the empirical risk is weakly convex, namely, the smallest negative eigenvalue of the empirical risk's Hessian is bounded in magnitude. By showing that this eigenvalue can control…

机器学习 · 统计学 2021-06-02 Dominic Richards , Mike Rabbat

We use smoothed analysis techniques to provide guarantees on the training loss of Multilayer Neural Networks (MNNs) at differentiable local minima. Specifically, we examine MNNs with piecewise linear activation functions, quadratic loss and…

机器学习 · 统计学 2016-05-31 Daniel Soudry , Yair Carmon

We give the first provably efficient algorithm for learning a one hidden layer convolutional network with respect to a general class of (potentially overlapping) patches. Additionally, our algorithm requires only mild conditions on the…

机器学习 · 计算机科学 2018-02-08 Surbhi Goel , Adam Klivans , Raghu Meka

It was conjectured that any neural network of any structure and arbitrary differentiable transfer functions at the nodes cannot learn the following problem sample efficiently when trained with gradient descent: The instances are the rows of…

机器学习 · 计算机科学 2020-10-20 Manfred K. Warmuth , Wojciech Kotłowski , Ehsan Amid