English

Function graph transformers universally approximate operators between function spaces

Machine Learning 2026-05-19 v1

Abstract

We study the approximation of nonlinear operators between function spaces by transformers. Our approach is to lift functions to measures supported on their graphs and leverage a recently introduced measure-theoretic view of transformers. A function hh is represented by its graph measure γh\gamma_h, with finite tokens {(xj,h(xj))}j=1N\{(x_j,h(x_j))\}_{j=1}^N being its empirical approximations. We show that this framework elegantly models discretization refinement via convergence of measures and provides a natural setting for operator learning. Within this framework, we introduce function graph transformers, a graph-preserving subclass of measure-theoretic transformers that maps graph measures to graph measures, which is to say that outputs remain single-valued functions. Crucially, this additional structure does not reduce generality: we prove that the resulting graph-preserving maps can be approximated by finite compositions of standard softmax self-attention layers and pointwise MLPs, yielding universal approximation results for broad classes of nonlinear operators. Unlike existing theoretical approaches to operator learning with transformers, the measure-theoretic framework also accommodates regularized negative-order Sobolev inputs for which discretization invariance is particularly challenging, as well as query points on different output domains. Overall, function graph transformers provide a continuum viewpoint and mathematical toolkit for transformer-based operator learning, clarifying the roles of positional encodings, graph structure, regularization, and ensuring consistency across discretizations.

Keywords

Cite

@article{arxiv.2605.17968,
  title  = {Function graph transformers universally approximate operators between function spaces},
  author = {Takashi Furuya and David Mis and Ivan Dokmanić and Maarten V. de Hoop and Matti Lassas},
  journal= {arXiv preprint arXiv:2605.17968},
  year   = {2026}
}