Neural network layers as parametric spans
Category Theory
2022-09-07 v2 Machine Learning
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.
Cite
@article{arxiv.2208.00809,
title = {Neural network layers as parametric spans},
author = {Mattia G. Bergomi and Pietro Vertechi},
journal= {arXiv preprint arXiv:2208.00809},
year = {2022}
}
Comments
12 pages, submitted to SYCO 9