English
Related papers

Related papers: Learning ReLUs via Gradient Descent

200 papers

Understanding implicit bias of gradient descent for generalization capability of ReLU networks has been an important research topic in machine learning research. Unfortunately, even for a single ReLU neuron trained with the square loss, it…

Machine Learning · Computer Science 2022-06-14 Sangmin Lee , Byeongsu Sim , Jong Chul Ye

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 provide a characterization of the feature-learning process in two-layer ReLU networks trained by gradient descent on the logistic loss following random initialization. We consider data with binary labels that are generated…

Machine Learning · Computer Science 2023-09-15 Spencer Frei , Niladri S. Chatterji , Peter L. Bartlett

Rectified linear unit (ReLU) is a widely used activation function for deep convolutional neural networks. However, because of the zero-hard rectification, ReLU networks miss the benefits from negative values. In this paper, we propose a…

Computer Vision and Pattern Recognition · Computer Science 2018-01-30 Suo Qiu , Xiangmin Xu , Bolun Cai

We introduce the "exponential linear unit" (ELU) which speeds up learning in deep neural networks and leads to higher classification accuracies. Like rectified linear units (ReLUs), leaky ReLUs (LReLUs) and parametrized ReLUs (PReLUs), ELUs…

Machine Learning · Computer Science 2016-02-23 Djork-Arné Clevert , Thomas Unterthiner , Sepp Hochreiter

We study the problem of inverting a deep generative model with ReLU activations. Inversion corresponds to finding a latent code vector that explains observed measurements as much as possible. In most prior works this is performed by…

Machine Learning · Computer Science 2019-06-20 Qi Lei , Ajil Jalal , Inderjit S. Dhillon , Alexandros G. Dimakis

Rectified linear activation units are important components for state-of-the-art deep convolutional networks. In this paper, we propose a novel S-shaped rectified linear activation unit (SReLU) to learn both convex and non-convex functions,…

Computer Vision and Pattern Recognition · Computer Science 2015-12-23 Xiaojie Jin , Chunyan Xu , Jiashi Feng , Yunchao Wei , Junjun Xiong , Shuicheng Yan

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,…

Machine Learning · Computer Science 2023-01-26 Rama Cont , Alain Rossier , RenYuan Xu

We study the implicit regularization imposed by gradient descent for learning multi-layer homogeneous functions including feed-forward fully connected and convolutional deep neural networks with linear, ReLU or Leaky ReLU activation. We…

Machine Learning · Computer Science 2018-11-01 Simon S. Du , Wei Hu , Jason D. Lee

The training of neural networks by gradient descent methods is a cornerstone of the deep learning revolution. Yet, despite some recent progress, a complete theory explaining its success is still missing. This article presents, for…

Machine Learning · Statistics 2026-04-15 Etienne Boursier , Loucas Pillaud-Vivien , Nicolas Flammarion

In this work, we propose to train a deep neural network by distributed optimization over a graph. Two nonlinear functions are considered: the rectified linear unit (ReLU) and a linear unit with both lower and upper cutoffs (DCutLU). The…

Machine Learning · Computer Science 2017-06-20 Guoqiang Zhang , W. Bastiaan Kleijn

We introduce a novel framework for learning in neural networks by decomposing each neuron's weight vector into two distinct parts, $W_1$ and $W_2$, thereby modeling contrastive information directly at the neuron level. Traditional gradient…

Machine Learning · Computer Science 2025-03-19 Xi Wang

Motivated by the growing theoretical understanding of neural networks that employ the Rectified Linear Unit (ReLU) as their activation function, we revisit the use of ReLU activation functions for learning implicit neural representations…

Image and Video Processing · Electrical Eng. & Systems 2024-08-05 Joseph Shenouda , Yamin Zhou , Robert D. Nowak

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…

Machine Learning · Computer Science 2020-09-29 Sitan Chen , Adam R. Klivans , Raghu Meka

Neural networks are a powerful class of functions that can be trained with simple gradient descent to achieve state-of-the-art performance on a variety of applications. Despite their practical success, there is a paucity of results that…

Machine Learning · Computer Science 2017-03-06 Bo Xie , Yingyu Liang , Le Song

A key challenge in modern deep learning theory is to explain the remarkable success of gradient-based optimization methods when training large-scale, complex deep neural networks. Though linear convergence of such methods has been proved…

Machine Learning · Computer Science 2025-09-30 Yash Jakhmola

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…

Optimization and Control · Mathematics 2020-06-23 Lenaic Chizat , Francis Bach

In this paper, we consider one dimensional (shallow) ReLU neural networks in which weights are chosen randomly and only the terminal layer is trained. First, we mathematically show that for such networks L2-regularized regression…

Machine Learning · Computer Science 2023-10-05 Jakob Heiss , Josef Teichmann , Hanna Wutte

We examine gradient descent on unregularized logistic regression problems, with homogeneous linear predictors on linearly separable datasets. We show the predictor converges to the direction of the max-margin (hard margin SVM) solution. The…

Machine Learning · Statistics 2024-10-29 Daniel Soudry , Elad Hoffer , Mor Shpigel Nacson , Suriya Gunasekar , Nathan Srebro

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…

Machine Learning · Computer Science 2020-08-18 Mert Pilanci , Tolga Ergen