about
Categorical Deep Learning: An Algebraic Theory of Architectures (arxiv.org)
3 points by milliondreams on Apr 10, 2024 | hide | past | pdf | discuss on HN

In plain words: Earlier frameworks describe either the rules a network must obey or how it is built, not both. A single math language for how structures map to each other covers both, recovering the symmetry rules of geometric models and designs like recurrent networks.

Abstract · Position: Categorical Deep Learning is an Algebraic Theory of All Architectures

We present our position on the elusive quest for a general-purpose framework for specifying and studying deep learning architectures. Our opinion is that the key attempts made so far lack a coherent bridge between specifying constraints which models must satisfy and specifying their implementations. Focusing on building a such a bridge, we propose to apply category theory -- precisely, the universal algebra of monads valued in a 2-category of parametric maps -- as a single theory elegantly subsuming both of these flavours of neural network design. To defend our position, we show how this theory recovers constraints induced by geometric deep learning, as well as implementations of many architectures drawn from the diverse landscape of neural networks, such as RNNs. We also illustrate how the theory naturally encodes many standard constructs in computer science and automata theory.

Bruno Gavranović, Paul Lessard, Andrew Dudzik, Tamara von Glehn, João G. M. Araújo, Petar Veličković
arXiv:2402.15332 · cs.LG, cs.AI, math.CT, math.RA, stat.ML · submitted Feb 23, 2024 · updated Jun 6, 2024
abstract · pdf · html · To appear in ICML 2024. Comments welcome. More info at categoricaldeeplearning.com

add comment on HN