An Approximation Theory for Metric Space-Valued Functions With A View Towards Transformers

31.01.2024 16:00 - 17:00

Anastasis Kratsios (McMaster University and Vector Institute)

Abstract:

We build universal approximators of continuous maps between arbitrary Polish metric spaces X and Y using universal approximators between Euclidean spaces as building blocks. Earlier results assume that the output space Y is a topological vector space. We overcome this limitation by "randomization": our approximators output discrete probability measures over Y. When X and Y are Polish without additional structure, we prove very general qualitative guarantees; when they have suitable combinatorial structure, we prove quantitative guarantees for Hölder-like maps, including maps between finite graphs, solution operators to rough differential equations between certain Carnot groups, and continuous non-linear operators between Banach spaces arising in inverse problems. In particular, we show that the required number of Dirac measures is determined by the combinatorial structure of X and Y. For barycentric Y, including Banach spaces, R-trees, Hadamard manifolds, or Wasserstein spaces on Polish metric spaces, our approximators reduce to Y-valued functions. When the Euclidean approximators are neural networks, our constructions generalize transformer networks, providing a new probabilistic viewpoint of geometric deep learning.
As an application, we show that the solution operator to an RDE can be approximated within our framework.

Zoom link and passcode can be found at the Mathematics of Machine Learning and Data Science seminar (https://math-ml.univie.ac.at/seminar/).

Organiser:
M. Neuman, S. Schmutzhard-Hoefler
Location:
Zoom