English
Related papers

Related papers: On weight initialization in deep neural networks

200 papers

Understanding the computational complexity of training simple neural networks with rectified linear units (ReLUs) has recently been a subject of intensive research. Closing gaps and complementing results from the literature, we present…

Machine Learning · Computer Science 2022-08-24 Vincent Froese , Christoph Hertrich , Rolf Niedermeier

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…

Machine Learning · Statistics 2021-11-03 Lu Lu , Yanhui Su , George Em Karniadakis

It is notoriously difficult to train Transformers on small datasets; typically, large pre-trained models are instead used as the starting point. We explore the weights of such pre-trained Transformers (particularly for vision) to attempt to…

Computer Vision and Pattern Recognition · Computer Science 2023-05-18 Asher Trockman , J. Zico Kolter

Initialization plays a critical role in Deep Neural Network training, directly influencing convergence, stability, and generalization. Common approaches such as Glorot and He initializations rely on randomness, which can produce uneven…

Machine Learning · Computer Science 2025-12-11 Alberto Fernández-Hernández , Jose I. Mestre , Manuel F. Dolz , Jose Duato , Enrique S. Quintana-Ortí

ReLU neural networks define piecewise linear functions of their inputs. However, initializing and training a neural network is very different from fitting a linear spline. In this paper, we expand empirically upon previous theoretical work…

Machine Learning · Statistics 2016-11-30 Kevin K. Chen , Anthony Gamst , Alden Walker

Neural networks with the Rectified Linear Unit (ReLU) nonlinearity are described by a vector of parameters $\theta$, and realized as a piecewise linear continuous function $R_{\theta}: x \in \mathbb R^{d} \mapsto R_{\theta}(x) \in \mathbb…

Machine Learning · Computer Science 2022-06-08 Pierre Stock , Rémi Gribonval

A traditional approach to initialization in deep neural networks (DNNs) is to sample the network weights randomly for preserving the variance of pre-activations. On the other hand, several studies show that during the training process, the…

Machine Learning · Computer Science 2021-02-16 Mert Gurbuzbalaban , Yuanhan Hu

In this paper we study the problem of learning Rectified Linear Units (ReLUs) which are functions of the form $max(0,<w,x>)$ with $w$ denoting the weight vector. We study this problem in the high-dimensional regime where the number of…

Machine Learning · Computer Science 2017-08-03 Mahdi Soltanolkotabi

In this paper, it is demonstrated through a case study that multilayer feedforward neural networks activated by ReLU functions can in principle be trained iteratively with Mixed Integer Linear Programs (MILPs) as follows. Weights are…

Machine Learning · Computer Science 2021-04-12 Steffen Goebbels

This article derives and validates three principles for initialization and architecture selection in finite width graph neural networks (GNNs) with ReLU activations. First, we theoretically derive what is essentially the unique…

Machine Learning · Statistics 2023-06-21 Gage DeZoort , Boris Hanin

Rectified linear unit (ReLU) activations can also be thought of as 'gates', which, either pass or stop their pre-activation input when they are 'on' (when the pre-activation input is positive) or 'off' (when the pre-activation input is…

Machine Learning · Computer Science 2021-06-15 Chandrashekar Lakshminarayanan , Amit Vikram Singh

The widely used ReLU is favored for its hardware efficiency, {as the implementation at inference is a one bit sign case,} yet suffers from issues such as the ``dying ReLU'' problem, where during training, neurons fail to activate and…

Machine Learning · Computer Science 2025-10-31 Moshe Kimhi , Idan Kashani , Avi Mendelson , Chaim Baskin

Using weight decay to penalize the L2 norms of weights in neural networks has been a standard training practice to regularize the complexity of networks. In this paper, we show that a family of regularizers, including weight decay, is…

Machine Learning · Computer Science 2022-06-09 Ziquan Liu , Yufei Cui , Antoni B. Chan

In this paper, we present a novel approach for initializing deep neural networks, i.e., by turning PCA into neural layers. Usually, the initialization of the weights of a deep neural network is done in one of the three following ways: 1)…

Machine Learning · Computer Science 2018-04-25 Mathias Seuret , Michele Alberti , Rolf Ingold , Marcus Liwicki

Activation functions play a key role in neural networks so it becomes fundamental to understand their advantages and disadvantages in order to achieve better performances. This paper will first introduce common types of non linear…

Machine Learning · Computer Science 2018-04-10 Dabal Pedamonti

Current research suggests that the key factors in designing neural network architectures involve choosing number of filters for every convolution layer, number of hidden neurons for every fully connected layer, dropout and pruning. The…

Machine Learning · Computer Science 2020-09-17 Himanshu Pradeep Aswani , Amit Sethi

For most state-of-the-art architectures, Rectified Linear Unit (ReLU) becomes a standard component accompanied with each layer. Although ReLU can ease the network training to an extent, the character of blocking negative values may suppress…

Computer Vision and Pattern Recognition · Computer Science 2017-11-20 Xuanyi Dong , Guoliang Kang , Kun Zhan , Yi Yang

In many information processing systems, it may be desirable to ensure that any change of the input, whether by shifting or scaling, results in a corresponding change in the system response. While deep neural networks are gradually replacing…

Computer Vision and Pattern Recognition · Computer Science 2024-02-22 Sébastien Herbreteau , Emmanuel Moebel , Charles Kervrann

We provide a function space characterization of the inductive bias resulting from minimizing the $\ell_2$ norm of the weights in multi-channel convolutional neural networks with linear activations and empirically test our resulting…

Machine Learning · Computer Science 2022-07-12 Meena Jagadeesan , Ilya Razenshteyn , Suriya Gunasekar

In low-latency or mobile applications, lower computation complexity, lower memory footprint and better energy efficiency are desired. Many prior works address this need by removing redundant parameters. Parameter quantization replaces…

Machine Learning · Computer Science 2021-11-16 Cheng-Chou Lan