On the Impact of the Activation Function on Deep Neural Networks Training
Not on the map. Why?
Abstract (arXiv)
The weight initialization and the activation function of deep neural networks have a crucial impact on the performance of the training procedure. An inappropriate selection can lead to the loss of information of the input during forward propagation and the exponential vanishing/exploding of gradients during back-propagation. Understanding the theoretical properties of untrained random networks is key to identifying which deep networks may be trained successfully as recently demonstrated by Samuel et al (2017) who showed that for deep feedforward neural networks only a specific choice of hyperparameters known as the `Edge of Chaos' can lead to good performance. While the work by Samuel et al (2017) discuss trainability issues, we focus here on training acceleration and overall performance. We give a comprehensive theoretical analysis of the Edge of Chaos and show that we can indeed tune the initialization parameters and the activation function in order to accelerate the training and improve the performance.
Key facts. Cited by 101 papers in the map (data through 7 Oct 2026). First posted on arXiv 19 Feb 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 23; 2022: 15; 2023: 18; 2024: 13; 2025: 18; 2026: 14.
101 papers.
Sort:
By area:
- 2024LoRA+: Efficient Low Rank Adaptation of Large Modelscited by 365
- 2021Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspectivecited by 168
- 2021Gradients are Not All You Needcited by 82
- 2024Disentangling the Causes of Plasticity Loss in Neural Networkscited by 82
- 2022Evaluating the Susceptibility of Pre-Trained Language Models via Handcrafted Adversarial Examplescited by 65
- 2024The Impact of Initialization on LoRA Finetuning Dynamicscited by 53
- 2024Riemannian Preconditioned LoRA for Fine-Tuning Foundation Modelscited by 51
- 2023The Shaped Transformer: Attention Models in the Infinite Depth-and-Width Limitcited by 47
- 2023Deep Transformers without Shortcuts: Modifying Self-attention for Faithful Signal Propagationcited by 44
- 2022The Neural Covariance SDE: Shaped Infinite Depth-and-Width Networks at Initializationcited by 41
- 2023Simplifying Transformer Blockscited by 37
- 2022Deep Learning without Shortcuts: Shaping the Kernel with Tailored Rectifierscited by 28
38 more
- 2023A Primer on Bayesian Neural Networks: Review and Debatescited by 26
- 2022On the infinite-depth limit of finite-width neural networkscited by 22
- 2023Width and Depth Limits Commute in Residual Networkscited by 21
- 2023Leave-one-out Distinguishability in Machine Learningcited by 20
- 2022Understanding the Role of Nonlinearity in Training Dynamics of Contrastive Learningcited by 19
- 2023Quantitative CLTs in Deep Neural Networkscited by 19
- 2021Feature Learning and Signal Propagation in Deep Neural Networkscited by 18
- 2025Beyond Zero Initialization: Investigating the Impact of Non-Zero Initialization on LoRA Fine-Tuning Dynamicscited by 16
- 2021Truly Sparse Neural Networks at Scalecited by 15
- 2021Towards Deepening Graph Neural Networks: A GNTK-based Optimization Perspectivecited by 14
- 2023Wide Neural Networks as Gaussian Processes: Lessons from Deep Equilibrium Modelscited by 14
- 2021Activation function design for deep networks: linearity and effective initialisationcited by 13
- 2025PLoP: Precise LoRA Placement for Efficient Finetuning of Large Modelscited by 13
- 2025The Hidden Width of Deep ResNets: Tight Error Bounds and Phase Diagramcited by 13
- 2021Finite volume method network for acceleration of unsteady computational fluid dynamics: non-reacting and reacting flowscited by 12
- 2022Wide and Deep Neural Networks Achieve Optimality for Classificationcited by 12
- 2023The Feature Speed Formula: a flexible approach to scale hyper-parameters of deep neural networkscited by 11
- 2021Wide stochastic networks: Gaussian limit and PAC-Bayesian trainingcited by 10
- 2024Understanding and Minimising Outlier Features in Neural Network Trainingcited by 10
- 2021Automatic Fault Detection for Deep Learning Programs Using Graph Transformationscited by 9
- 2021Neuron Campaign for Initialization Guided by Information Bottleneck Theorycited by 9
- 2022Gaussian Pre-Activations in Neural Networks: Myth or Reality?cited by 9
- 2024Theoretical Insights into Fine-Tuning Attention Mechanism: Generalization and Optimizationcited by 9
- 2026The Myth of Expert Specialization in MoEs: Why Routing Reflects Geometry, Not Necessarily Domain Expertisecited by 9
- 2021Deep Stable neural networks: large-width asymptotics and convergence ratescited by 8
- 2025Physics of Skill Learningcited by 8
- 2025AltLoRA: Towards Better Gradient Approximation in Low-Rank Adaptation with Alternating Projectionscited by 8
- 2026Learning Rate Scaling across LoRA Ranks and Transfer to Full Finetuningcited by 8
- 2021Using activation histograms to bound the number of affine regions in ReLU feed-forward neural networkscited by 7
- 2021Regularization in ResNet with Stochastic Depthcited by 7
- 2021Understanding and Accelerating Neural Architecture Search with Training-Free and Theory-Grounded Metricscited by 7
- 2022Analysis of Convolutions, Non-linearity and Depth in Graph Neural Networks using Neural Tangent Kernelcited by 7
- 2021Bayesian neural network unit priors and generalized Weibull-tail propertycited by 5
- 2022Self-scalable Tanh (Stan): Faster Convergence and Better Generalization in Physics-informed Neural Networkscited by 5
- 2023On the Initialisation of Wide Low-Rank Feedforward Neural Networkscited by 5
- 2024Task structure and nonlinearity jointly determine learned representational geometrycited by 5
- 2024Kolmogorov-Arnold Networks in Low-Data Regimes: A Comparative Study with Multilayer Perceptronscited by 5
- 2021Universal scaling laws in the gradient descent training of neural networkscited by 4
50 most cited of 101 All 101 (JSON)
- 2026Stable initialization without the CLT24 Sep 2026
- 2026Critical initialization destabilizes higher input derivatives in wide scalar-input networks8 Sep 2026
- 2026Q-TriM: Question-Guided Tri-Modal Attention for Audio-Visual Question Answering4 Jul 2026
- 2026Rethinking Neural Nonlinearity as Gating3 Jul 2026
- 2026How Deep Are Deep GPs, Really? A Sharp Threshold and a Non-Gaussian Limit for Compositional GPs6 Jun 2026
- 2026Dropout Universality: Scaling Laws and Optimal Scheduling at the Edge-of-Chaos20 May 2026
- 2026A Composite Activation Function for Learning Stable Binary Representations12 May 2026
- 2026Universal Smoothness via Bernstein Polynomials: A Constructive Approximation Approach for Activation Functions4 May 2026
- 2026Inter-Layer Hessian Analysis of Neural Networks with DAG Architectures13 Apr 2026
- 2026The Myth of Expert Specialization in MoEs: Why Routing Reflects Geometry, Not Necessarily Domain Expertise10 Apr 2026
- 2026When Does Sparsity Mitigate the Curse of Depth in LLMs16 Mar 2026
- 2026Stable-LoRA: Stabilizing Feature Learning of Low-Rank Adaptation5 Mar 2026
38 more
- 2026Optimal Initialization in Depth: Lyapunov Initialization and Limit Theorems for Deep Leaky ReLU Networks11 Feb 2026
- 2026Learning Rate Scaling across LoRA Ranks and Transfer to Full Finetuning5 Feb 2026
- 2025A new initialisation to Control Gradients in Sinusoidal Neural network6 Dec 2025
- 2025A Proof of Learning Rate Transfer under μP3 Nov 2025
- 2025Beyond Gaussian Initializations: Signal Preserving Weight Initialization for Odd-Sigmoid Activations27 Sep 2025
- 2025The Hidden Width of Deep ResNets: Tight Error Bounds and Phase Diagram12 Sep 2025
- 2025SingLoRA: Low Rank Adaptation Using a Single Matrix8 Jul 2025
- 2025A parametric activation function based on Wendland RBF28 Jun 2025
- 2025PLoP: Precise LoRA Placement for Efficient Finetuning of Large Models25 Jun 2025
- 2025Optimal Embedding Learning Rate in LLMs: The Effect of Vocabulary Size17 Jun 2025
- 2025On the Stability of the Jacobian Matrix in Deep Neural Networks10 Jun 2025
- 2025Impact of Tuning Parameters in Deep Convolutional Neural Network Using a Crack Image Dataset30 May 2025
- 2025Beyond Zero Initialization: Investigating the Impact of Non-Zero Initialization on LoRA Fine-Tuning Dynamics29 May 2025
- 2025Approach to Finding a Robust Deep Learning Model22 May 2025
- 2025AltLoRA: Towards Better Gradient Approximation in Low-Rank Adaptation with Alternating Projections18 May 2025
- 2025Revisiting Residual Connections: Orthogonal Updates for Stable and Efficient Deep Networks17 May 2025
- 2025How good is PAC-Bayes at explaining generalisation?11 Mar 2025
- 2025REAct: Rational Exponential Activation for Better Learning and Generalization in PINNs4 Mar 2025
- 2025Feature Learning Beyond the Edge of Stability18 Feb 2025
- 2025Physics of Skill Learning21 Jan 2025
- 2024Emergence of Globally Attracting Fixed Points in Deep Neural Networks With Nonlinear Activations26 Oct 2024
- 2024Generalized Probabilistic Attention Mechanism in Transformers21 Oct 2024
- 2024Theoretical Insights into Fine-Tuning Attention Mechanism: Generalization and Optimization3 Oct 2024
- 2024Kolmogorov-Arnold Networks in Low-Data Regimes: A Comparative Study with Multilayer Perceptrons16 Sep 2024
- 2024The Impact of Initialization on LoRA Finetuning Dynamics12 Jun 2024
- 2024Understanding and Minimising Outlier Features in Neural Network Training29 May 2024
- 2024Spectral complexity of deep neural networks15 May 2024
- 2024Rolling the dice for better deep learning performance: A study of randomness techniques in deep neural networks5 Apr 2024
- 2024Disentangling the Causes of Plasticity Loss in Neural Networks29 Feb 2024
- 2024LoRA+: Efficient Low Rank Adaptation of Large Models19 Feb 2024
- 2024Adaptive Activation Functions for Predictive Modeling with Sparse Experimental Data8 Feb 2024
- 2024Riemannian Preconditioned LoRA for Fine-Tuning Foundation Models4 Feb 2024
- 2024Task structure and nonlinearity jointly determine learned representational geometry24 Jan 2024
- 2023The Feature Speed Formula: a flexible approach to scale hyper-parameters of deep neural networks30 Nov 2023
- 2023Simplifying Transformer Blocks3 Nov 2023
- 2023From Alexnet to Transformers: Measuring the Non-linearity of Deep Neural Networks with Affine Optimal Transport17 Oct 2023
- 2023Wide Neural Networks as Gaussian Processes: Lessons from Deep Equilibrium Models16 Oct 2023
- 2023Commutative Width and Depth Scaling in Deep Neural Networks2 Oct 2023
50 newest of 101 All 101 (JSON)
- 2024LoRA+: Efficient Low Rank Adaptation of Large Modelscited by 365
- 2021Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspectivecited by 168
- 2021Gradients are Not All You Needcited by 82
- 2024Disentangling the Causes of Plasticity Loss in Neural Networkscited by 82
- 2022Evaluating the Susceptibility of Pre-Trained Language Models via Handcrafted Adversarial Examplescited by 65
- 2024The Impact of Initialization on LoRA Finetuning Dynamicscited by 53
- 2024Riemannian Preconditioned LoRA for Fine-Tuning Foundation Modelscited by 51
- 2023The Shaped Transformer: Attention Models in the Infinite Depth-and-Width Limitcited by 47
- 2023Deep Transformers without Shortcuts: Modifying Self-attention for Faithful Signal Propagationcited by 44
- 2022The Neural Covariance SDE: Shaped Infinite Depth-and-Width Networks at Initializationcited by 41
- 2023Simplifying Transformer Blockscited by 37
- 2022Deep Learning without Shortcuts: Shaping the Kernel with Tailored Rectifierscited by 28
12 most cited of 100 All 100 (JSON)
- 2026Stable initialization without the CLT24 Sep 2026
- 2026Critical initialization destabilizes higher input derivatives in wide scalar-input networks8 Sep 2026
- 2026Q-TriM: Question-Guided Tri-Modal Attention for Audio-Visual Question Answering4 Jul 2026
- 2026Rethinking Neural Nonlinearity as Gating3 Jul 2026
- 2026How Deep Are Deep GPs, Really? A Sharp Threshold and a Non-Gaussian Limit for Compositional GPs6 Jun 2026
- 2026Dropout Universality: Scaling Laws and Optimal Scheduling at the Edge-of-Chaos20 May 2026
- 2026A Composite Activation Function for Learning Stable Binary Representations12 May 2026
- 2026Universal Smoothness via Bernstein Polynomials: A Constructive Approximation Approach for Activation Functions4 May 2026
- 2026Inter-Layer Hessian Analysis of Neural Networks with DAG Architectures13 Apr 2026
- 2026When Does Sparsity Mitigate the Curse of Depth in LLMs16 Mar 2026
- 2026Stable-LoRA: Stabilizing Feature Learning of Low-Rank Adaptation5 Mar 2026
- 2026Optimal Initialization in Depth: Lyapunov Initialization and Limit Theorems for Deep Leaky ReLU Networks11 Feb 2026
12 newest of 100 All 100 (JSON)