arXiv · 2506.23551
A unified framework for establishing the universal approximation of transformer-type architectures
Abstract
We investigate the universal approximation property (UAP) of transformer-type architectures, providing a unified theoretical framework that extends prior results on residual networks to models incorporating attention mechanisms. Our work identifies token distinguishability as a fundamental requirement for UAP and introduces a general sufficient condition that applies to a broad class of architectures. Leveraging an analyticity assumption on the attention layer, we can significantly simplify the verification of this condition, providing a non-constructive approach in establishing UAP for such architectures. We demonstrate the applicability of our framework by proving UAP for transformers with various attention mechanisms, including kernel-based and sparse attention mechanisms. The corollaries of our results either generalize prior works or establish UAP for architectures not previously covered. Furthermore, our framework offers a principled foundation for designing novel transformer architectures with inherent UAP guarantees, including those with specific functional symmetries. We propose examples to illustrate these insights.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Jingpu Cheng, Ting Lin, Zuowei Shen, Qianxiao Li. 2025-06-30. A unified framework for establishing the universal approximation of transformer-type architectures. https://arxiv.org/abs/2506.23551
Cite the original work for its findings. Save a collection to share your selection of sources.