English
Related papers

Related papers: Efficiently Learning One-Hidden-Layer ReLU Network…

200 papers

We consider PAC learning of probability distributions (a.k.a. density estimation), where we are given an i.i.d. sample generated from an unknown target distribution, and want to output a distribution that is close to the target in total…

Machine Learning · Computer Science 2018-06-05 Hassan Ashtiani , Shai Ben-David , Abbas Mehrabian

Let $V$ be any vector space of multivariate degree-$d$ homogeneous polynomials with co-dimension at most $k$, and $S$ be the set of points where all polynomials in $V$ {\em nearly} vanish. We establish a qualitatively optimal upper bound on…

Machine Learning · Computer Science 2020-12-15 Ilias Diakonikolas , Daniel M. Kane

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…

Machine Learning · Computer Science 2020-10-26 Surbhi Goel , Aravind Gollakota , Zhihan Jin , Sushrut Karmalkar , Adam Klivans

We show that the class of strongly connected graphical models with treewidth at most k can be properly efficiently PAC-learnt with respect to the Kullback-Leibler Divergence. Previous approaches to this problem, such as those of Chow ([1]),…

Machine Learning · Computer Science 2012-07-19 Mukund Narasimhan , Jeff A. Bilmes

We consider the optimization problem associated with fitting two-layers ReLU networks with respect to the squared loss, where labels are generated by a target network. We leverage the rich symmetry structure to analytically characterize the…

Machine Learning · Computer Science 2020-10-19 Yossi Arjevani , Michael Field

We present a generalization of the well-known problem of learning k-juntas in R^n, and a novel tensor algorithm for unraveling the structure of high-dimensional distributions. Our algorithm can be viewed as a higher-order extension of…

Computational Complexity · Computer Science 2012-04-17 Santosh S. Vempala , Ying Xiao

We present polynomial time and sample efficient algorithms for learning an unknown depth-2 feedforward neural network with general ReLU activations, under mild non-degeneracy assumptions. In particular, we consider learning an unknown…

Machine Learning · Computer Science 2021-08-03 Pranjal Awasthi , Alex Tang , Aravindan Vijayaraghavan

We study the collaborative PAC learning problem recently proposed in Blum et al.~\cite{BHPQ17}, in which we have $k$ players and they want to learn a target function collaboratively, such that the learned function approximates the target…

Machine Learning · Computer Science 2018-10-15 Jiecao Chen , Qin Zhang , Yuan Zhou

Neural networks with REctified Linear Unit (ReLU) activation functions (a.k.a. ReLU networks) have achieved great empirical success in various domains. Nonetheless, existing results for learning ReLU networks either pose assumptions on the…

Machine Learning · Statistics 2019-05-01 Gang Wang , Georgios B. Giannakis , Jie Chen

We develop a corrective mechanism for neural network approximation: the total available non-linear units are divided into multiple groups and the first group approximates the function under consideration, the second group approximates the…

Machine Learning · Computer Science 2020-06-23 Guy Bresler , Dheeraj Nagaraj

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…

Machine Learning · Computer Science 2019-05-28 Santosh Vempala , John Wilmes

Rectified linear units, or ReLUs, have become the preferred activation function for artificial neural networks. In this paper we consider two basic learning problems assuming that the underlying data follow a generative model based on a…

Machine Learning · Statistics 2018-03-13 Arya Mazumdar , Ankit Singh Rawat

In this work, we consider the problem of learning one hidden layer ReLU neural networks with inputs from $\mathbb{R}^d$. We show that this learning problem is hard under standard cryptographic assumptions even when: (1) the size of the…

Machine Learning · Computer Science 2024-10-07 Shuchen Li , Ilias Zadik , Manolis Zampetakis

Neural networks are usually trained with different variants of gradient descent based optimization algorithms such as stochastic gradient descent or the Adam optimizer. Recent theoretical work states that the critical points (where the…

Machine Learning · Computer Science 2024-10-15 Adrian Barbu

We give the first polynomial-time algorithm for performing linear or polynomial regression resilient to adversarial corruptions in both examples and labels. Given a sufficiently large (polynomial-size) training set drawn i.i.d. from…

Machine Learning · Computer Science 2020-06-05 Adam Klivans , Pravesh K. Kothari , Raghu Meka

We analyse the convergence of one-hidden-layer ReLU networks trained by gradient flow on $n$ data points. Our main contribution leverages the high dimensionality of the ambient space, which implies low correlation of the input samples, to…

Machine Learning · Statistics 2025-12-02 Léo Dana , Francis Bach , Loucas Pillaud-Vivien

Large number of ReLU and MAC operations of Deep neural networks make them ill-suited for latency and compute-efficient private inference. In this paper, we present a model optimization method that allows a model to learn to be shallow. In…

Machine Learning · Computer Science 2023-04-27 Souvik Kundu , Yuke Zhang , Dake Chen , Peter A. Beerel

Neural networks activated by the rectified linear unit (ReLU) play a central role in the recent development of deep learning. The topic of approximating functions from H\"older spaces by these networks is crucial for understanding the…

Machine Learning · Computer Science 2023-07-25 Tong Mao , Ding-Xuan Zhou

We consider the fundamental problem of ReLU regression, where the goal is to output the best fitting ReLU with respect to square loss given access to draws from some unknown distribution. We give the first efficient, constant-factor…

Machine Learning · Computer Science 2020-09-30 Ilias Diakonikolas , Surbhi Goel , Sushrut Karmalkar , Adam R. Klivans , Mahdi Soltanolkotabi

We prove that, for the fundamental regression task of learning a single neuron, training a one-hidden layer ReLU network of any width by gradient flow from a small initialisation converges to zero loss and is implicitly biased to minimise…

Machine Learning · Computer Science 2023-10-03 Dmitry Chistikov , Matthias Englert , Ranko Lazic