A Unified Representation of Neural Networks Architectures
This provides a theoretical framework for unifying neural network representations, which is incremental as it builds on prior work like neural ODEs and continuous neural networks.
The paper tackles the problem of representing neural network architectures in a unified way by deriving a continuum limit as neurons and layers go to infinity, resulting in a Distributed Parameter neural Network (DiPaNet) that generalizes existing finite and infinite-dimensional architectures through homogenization and discretization techniques.
In this paper we consider the limiting case of neural networks (NNs) architectures when the number of neurons in each hidden layer and the number of hidden layers tend to infinity thus forming a continuum, and we derive approximation errors as a function of the number of neurons and/or hidden layers. Firstly, we consider the case of neural networks with a single hidden layer and we derive an integral infinite width neural representation that generalizes existing continuous neural networks (CNNs) representations. Then we extend this to deep residual CNNs that have a finite number of integral hidden layers and residual connections. Secondly, we revisit the relation between neural ODEs and deep residual NNs and we formalize approximation errors via discretization techniques. Then, we merge these two approaches into a unified homogeneous representation of NNs as a Distributed Parameter neural Network (DiPaNet) and we show that most of the existing finite and infinite-dimensional NNs architectures are related via homogenization/discretization with the DiPaNet representation. Our approach is purely deterministic and applies to general, uniformly continuous matrix weight functions. Relations with neural fields and other neural integro-differential equations are discussed along with further possible generalizations and applications of the DiPaNet framework.