English

Graph Convolutional Neural Networks as Parametric CoKleisli morphisms

Category Theory 2022-12-02 v1 Machine Learning

Abstract

We define the bicategory of Graph Convolutional Neural Networks GCNNn\mathbf{GCNN}_n for an arbitrary graph with nn nodes. We show it can be factored through the already existing categorical constructions for deep learning called Para\mathbf{Para} and Lens\mathbf{Lens} with the base category set to the CoKleisli category of the product comonad. We prove that there exists an injective-on-objects, faithful 2-functor GCNNnPara(CoKl(Rn×n×))\mathbf{GCNN}_n \to \mathbf{Para}(\mathsf{CoKl}(\mathbb{R}^{n \times n} \times -)). We show that this construction allows us to treat the adjacency matrix of a GCNN as a global parameter instead of a a local, layer-wise one. This gives us a high-level categorical characterisation of a particular kind of inductive bias GCNNs possess. Lastly, we hypothesize about possible generalisations of GCNNs to general message-passing graph neural networks, connections to equivariant learning, and the (lack of) functoriality of activation functions.

Keywords

Cite

@article{arxiv.2212.00542,
  title  = {Graph Convolutional Neural Networks as Parametric CoKleisli morphisms},
  author = {Bruno Gavranović and Mattia Villani},
  journal= {arXiv preprint arXiv:2212.00542},
  year   = {2022}
}

Comments

21 pages

R2 v1 2026-06-28T07:19:27.911Z