Work

Neural network layers as parametric spans

Year: 2022
Type: preprint
Abstract: Properties such as composability and automatic differentiation made artificial neural networks a pervasive tool in applications. Tackling more challenging problems caused neural networks to progressively become more complex and thus difficult to define from a mathematical perspective. We present a general definition of linear layer arising from a categorical framework based on the notions of integration theory and parametric spans. This definition generalizes and encompasses classical layers (e.g., dense, convolutional), while guaranteeing existence and computability of the layer's derivatives for backpropagation.
Language: English
Open Access status: green
OpenAlex ID: W4289598139
Cites:
Cited by:
Related to: 10