Saved in:
Bibliographic Details
Main Authors: Alokhina, Anastasiia, Li, Pan
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2512.12805
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909985325187072
author Alokhina, Anastasiia
Li, Pan
author_facet Alokhina, Anastasiia
Li, Pan
contents Transformers exhibit a notable property of \emph{size generalization}, demonstrating an ability to extrapolate from smaller token sets to significantly longer ones. This behavior has been documented across diverse applications, including point clouds, graphs, and natural language. Despite its empirical success, this capability still lacks some rigorous theoretical characterizations. In this paper, we develop a theoretical framework to analyze this phenomenon for geometric data, which we represent as discrete samples from a continuous source (e.g., point clouds from manifolds, graphs from graphons). Our core contribution is a bound on the error between the Transformer's output for a discrete sample and its continuous-domain equivalent. We prove that for Transformers with stable positional encodings, this bound is determined by the sampling density and the intrinsic dimensionality of the data manifold. Experiments on graphs and point clouds of various sizes confirm the tightness of our theoretical bound.
format Preprint
id arxiv_https___arxiv_org_abs_2512_12805
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle From Small to Large: Generalization Bounds for Transformers on Variable-Size Inputs
Alokhina, Anastasiia
Li, Pan
Machine Learning
Artificial Intelligence
Transformers exhibit a notable property of \emph{size generalization}, demonstrating an ability to extrapolate from smaller token sets to significantly longer ones. This behavior has been documented across diverse applications, including point clouds, graphs, and natural language. Despite its empirical success, this capability still lacks some rigorous theoretical characterizations. In this paper, we develop a theoretical framework to analyze this phenomenon for geometric data, which we represent as discrete samples from a continuous source (e.g., point clouds from manifolds, graphs from graphons). Our core contribution is a bound on the error between the Transformer's output for a discrete sample and its continuous-domain equivalent. We prove that for Transformers with stable positional encodings, this bound is determined by the sampling density and the intrinsic dimensionality of the data manifold. Experiments on graphs and point clouds of various sizes confirm the tightness of our theoretical bound.
title From Small to Large: Generalization Bounds for Transformers on Variable-Size Inputs
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2512.12805