Numerical influence of ReLU'(0) on backpropagation
David Bertoin, Jérôme Bolte, Sébastien Gerchinovitz, Edouard Pauwels
Abstract
In theory, the choice of ReLU (0) in [0, 1] for a neural network has a negligible influence both on backpropagation and training. Yet, in the real world, 32 bits default precision combined with the size of deep learning problems makes it a hyperparameter of training methods. We investigate the importance of the value of ReLU (0) for several precision levels (16, 32, 64 bits), on various networks (fully connected, VGG, ResNet) and datasets (MNIST, CIFAR10, SVHN, Ima-geNet). We observe considerable variations of backpropagation outputs which occur around half of the time in 32 bits precision. The effect disappears with double precision, while it is systematic at 16 bits. For vanilla SGD training, the choice ReLU (0) = 0 seems to be the most efficient. For our experiments on ImageNet the gain in test accuracy over ReLU (0) = 1 was more than 10 points (two runs). We also evidence that reconditioning approaches as batch-norm or ADAM tend to buffer the influence of ReLU (0)'s value. Overall, the message we convey is that algorithmic differentiation of nonsmooth problems potentially hides parameters that could be tuned advantageously. 1 Definition in Section 2. This can be coded explicitly or cheaply emulated by backprop 0 . Indeed, considering fs : x → ReLU(x) + s(ReLU(-x) -ReLU(x) + x) = ReLU(x), we have backprop 0 fs = s.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext eabd0f04-99aa-4dee-94b0-c8730492fa62Cited by top-tier papers9
- Learning single-index models with shallow neural networksAlberto Bietti, Joan Bruna, Clayton Sanford, Min Jae SongNeurIPS 2022 · 119 citations
- Gradient flow dynamics of shallow ReLU networks for square loss and orthogonal inputsEtienne Boursier, Loucas Pillaud-Vivien, Nicolas FlammarionNeurIPS 2022 · 92 citations
- On the Correctness of Automatic Differentiation for Neural Networks with Machine-Representable ParametersWonyeol Lee, Sejun Park, Alex AikenICML 2023 · 6 citations
- Hamiltonian Monte Carlo on ReLU Neural Networks is InefficientVu C. Dinh, Lam Si Tung Ho, Cuong V. NguyenNeurIPS 2024 · 3 citations
- Differentiable Decision Tree via "ReLU+Argmin" ReformulationQiangqiang Mao, Jiayang Ren, Yixiu Wang, Chenxuanyin Zou et al.NeurIPS 2025 · 2 citations
Builds on1
Related papers
- Any-Precision Deep Neural NetworksHaichao Yu, Haoxiang Li, Humphrey Shi, Thomas S. Huang et al.AAAI 2021 · 79 citations
- S: Sign-Sparse-Shift Reparametrization for Effective Training of Low-bit Shift NetworksXinlin Li, Bang Liu, Yaoliang Yu, Wulong Liu et al.NeurIPS 2021 · 12 citations
- The Effect of Weight Precision on the Neuron Count in Deep ReLU NetworksSonghua He, Periklis A. PapakonstantinouICML 2024
- Extreme Memorization via Scale of InitializationHarsh Mehta, Ashok Cutkosky, Behnam NeyshaburICLR 2021 · 22 citations
- Accurate Neural Training with 4-bit Matrix Multiplications at Standard FormatsBrian Chmiel, Ron Banner, Elad Hoffer, Hilla Ben-Yaacov et al.ICLR 2023 · 6 citations
