English

Stably unactivated neurons in ReLU neural networks

Machine Learning 2024-12-18 v2 Probability Machine Learning

Abstract

The choice of architecture of a neural network influences which functions will be realizable by that neural network and, as a result, studying the expressiveness of a chosen architecture has received much attention. In ReLU neural networks, the presence of stably unactivated neurons can reduce the network's expressiveness. In this work, we investigate the probability of a neuron in the second hidden layer of such neural networks being stably unactivated when the weights and biases are initialized from symmetric probability distributions. For networks with input dimension n0n_0, we prove that if the first hidden layer has n0+1n_0+1 neurons then this probability is exactly 2n0+14n0+1\frac{2^{n_0}+1}{4^{n_0+1}}, and if the first hidden layer has n1n_1 neurons, n1n0n_1 \le n_0, then the probability is 12n1+1\frac{1}{2^{n_1+1}}. Finally, for the case when the first hidden layer has more neurons than n0+1n_0+1, a conjecture is proposed along with the rationale. Computational evidence is presented to support the conjecture.

Keywords

Cite

@article{arxiv.2412.06829,
  title  = {Stably unactivated neurons in ReLU neural networks},
  author = {Natalie Brownlowe and Christopher R. Cornwell and Ethan Montes and Gabriel Quijano and Grace Stulman and Na Zhang},
  journal= {arXiv preprint arXiv:2412.06829},
  year   = {2024}
}