Backprop as Functor: A compositional perspective on supervised learning
Category Theory
2019-05-02 v3 Artificial Intelligence
Machine Learning
Abstract
A supervised learning algorithm searches over a set of functions parametrised by a space to find the best approximation to some ideal function . It does this by taking examples , and updating the parameter according to some rule. We define a category where these update rules may be composed, and show that gradient descent---with respect to a fixed step size and an error function satisfying a certain property---defines a monoidal functor from a category of parametrised functions to this category of update rules. This provides a structural perspective on backpropagation, as well as a broad generalisation of neural networks.
Cite
@article{arxiv.1711.10455,
title = {Backprop as Functor: A compositional perspective on supervised learning},
author = {Brendan Fong and David I. Spivak and Rémy Tuyéras},
journal= {arXiv preprint arXiv:1711.10455},
year = {2019}
}
Comments
13 pages + 4 page appendix