Why ReLU? A Bit-Model Dichotomy for Deep Network Training
Abstract
Lay Summary
Many theoretical results about neural-network training assume an idealized world where computers can store real numbers with infinite precision. This makes the mathematics cleaner, but it does not match real training systems, where every number is stored using only finitely many bits. We study how this finite-precision constraint changes the computational difficulty of training neural networks. Our main finding is that the answer depends sharply on the activation function, the rule that determines how each neuron transforms its input. For networks with polynomial activations of degree at least two, finite-precision training remains computationally intractable: even deciding whether good parameters exist, or extracting certain information from the training gradients, can encode extremely hard counting problems. By contrast, for piecewise-linear activations such as ReLU, finite precision behaves much more benignly: the relevant numbers can be represented with only polynomially many bits, and standard procedures such as backpropagation can be carried out efficiently. This shows that the numerical model is not just an implementation detail. It can change the complexity landscape of neural-network training and separate activation functions that look similar in idealized mathematical models but behave very differently on real computers.