中文
相关论文

相关论文: Learning Over-Parametrized Two-Layer ReLU Neural N…

200 篇论文

We analyze the convergence of the averaged stochastic gradient descent for overparameterized two-layer neural networks for regression problems. It was recently found that a neural tangent kernel (NTK) plays an important role in showing the…

机器学习 · 统计学 2021-06-14 Atsushi Nitanda , Taiji Suzuki

Recently, over-parameterized neural networks have been extensively analyzed in the literature. However, the previous studies cannot satisfactorily explain why fully trained neural networks are successful in practice. In this paper, we…

机器学习 · 计算机科学 2019-10-28 Cong Fang , Hanze Dong , Tong Zhang

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…

机器学习 · 计算机科学 2023-10-10 Arvind Mahankali , Jeff Z. Haochen , Kefan Dong , Margalit Glasgow , Tengyu Ma

Deep neural networks (DNNs) have demonstrated dominating performance in many fields; since AlexNet, networks used in practice are going wider and deeper. On the theoretical side, a long line of works has been focusing on training neural…

机器学习 · 计算机科学 2019-06-18 Zeyuan Allen-Zhu , Yuanzhi Li , Zhao Song

A remarkable recent discovery in machine learning has been that deep neural networks can achieve impressive performance (in terms of both lower training error and higher generalization capacity) in the regime where they are massively…

机器学习 · 计算机科学 2020-03-03 Thanh V. Nguyen , Raymond K. W. Wong , Chinmay Hegde

Recently, a spate of papers have provided positive theoretical results for training over-parameterized neural networks (where the network size is larger than what is needed to achieve low error). The key insight is that with sufficient…

机器学习 · 计算机科学 2022-03-01 Gilad Yehudai , Ohad Shamir

How can local-search methods such as stochastic gradient descent (SGD) avoid bad local minima in training multi-layer neural networks? Why can they fit random labels even given non-convex and non-smooth architectures? Most existing theory…

机器学习 · 计算机科学 2019-05-28 Zeyuan Allen-Zhu , Yuanzhi Li , Zhao Song

Consider the following fundamental learning problem: given input examples $x \in \mathbb{R}^d$ and their vector-valued labels, as defined by an underlying generative neural network, recover the weight matrices of this network. We consider…

数据结构与算法 · 计算机科学 2018-11-06 Ainesh Bakshi , Rajesh Jayaram , David P. Woodruff

We consider the problem of learning an unknown ReLU network with respect to Gaussian inputs and obtain the first nontrivial results for networks of depth more than two. We give an algorithm whose running time is a fixed polynomial in the…

机器学习 · 计算机科学 2020-09-29 Sitan Chen , Adam R. Klivans , Raghu Meka

It has been shown that gradient descent can yield the zero training loss in the over-parametrized regime (the width of the neural networks is much larger than the number of data points). In this work, combining the ideas of some existing…

最优化与控制 · 数学 2019-11-05 Lei Li

Neural networks trained to minimize the logistic (a.k.a. cross-entropy) loss with gradient-based methods are observed to perform well in many supervised classification tasks. Towards understanding this phenomenon, we analyze the training…

最优化与控制 · 数学 2020-06-23 Lenaic Chizat , Francis Bach

We revisit the problem of learning a single neuron with ReLU activation under Gaussian input with square loss. We particularly focus on the over-parameterization setting where the student network has $n\ge 2$ neurons. We prove the global…

机器学习 · 计算机科学 2023-10-11 Weihang Xu , Simon S. Du

Implicit deep learning has received increasing attention recently due to the fact that it generalizes the recursive prediction rules of many commonly used neural network architectures. Its prediction rule is provided implicitly based on the…

机器学习 · 计算机科学 2022-02-21 Tianxiang Gao , Hailiang Liu , Jia Liu , Hridesh Rajan , Hongyang Gao

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…

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…

机器学习 · 计算机科学 2023-03-09 Peizhong Ju , Xiaojun Lin , Ness B. Shroff

Over-parametrization is an important technique in training neural networks. In both theory and practice, training a larger network allows the optimization algorithm to avoid bad local optimal solutions. In this paper we study a closely…

机器学习 · 统计学 2020-10-23 Xiang Wang , Chenwei Wu , Jason D. Lee , Tengyu Ma , Rong Ge

This paper presents a comprehensive study on the convergence rates of the stochastic gradient descent (SGD) algorithm when applied to overparameterized two-layer neural networks. Our approach combines the Neural Tangent Kernel (NTK)…

机器学习 · 统计学 2024-07-11 Dinghao Cao , Zheng-Chu Guo , Lei Shi

Machine Unlearning aims to remove specific data from trained models, addressing growing privacy and ethical concerns. We provide a theoretical analysis of a simple and widely used method - gradient ascent - used to reverse the influence of…

机器学习 · 计算机科学 2025-10-17 Odelia Melamed , Gilad Yehudai , Gal Vardi

We consider the problem of learning an unknown function $f_{\star}$ on the $d$-dimensional sphere with respect to the square loss, given i.i.d. samples $\{(y_i,{\boldsymbol x}_i)\}_{i\le n}$ where ${\boldsymbol x}_i$ is a feature vector…

统计理论 · 数学 2020-02-18 Behrooz Ghorbani , Song Mei , Theodor Misiakiewicz , Andrea Montanari

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