Transformer Injectivity & Geometric Robustness - Analytic Margins and Bi-Lipschitz Uniformity of Sequence-Level Hidden States
Not on the map. Why?
Abstract (arXiv)
Under real-analytic assumptions on decoder-only Transformers, recent work shows that the map from discrete prompts to last-token hidden states is generically injective on finite prompt sets. We refine this picture: for each layer ℓ we define a collision discriminant Deltaℓ subset Theta and injective stratum Uℓ = Theta setminus Deltaℓ, and prove a dichotomy -- either the model is nowhere injective on the set, or Uℓ is open and dense and every Fℓθ is injective. Under mild non-singularity assumptions on the optimizer and an absolutely continuous initialization, generic injectivity persists along smooth training trajectories over any fixed horizon. We also treat symmetry groups G, showing that discriminants and injective strata descend to the quotient Theta/G, so injectivity is naturally a property of functional equivalence classes. We complement these results with an empirical study of layerwise geometric diagnostics. We define a separation margin and a co-Lipschitz (lower Lipschitz) constant between prompt space and last-token representation space, estimated via nearest-neighbor statistics on large prompt sets. Applying these diagnostics to pretrained LLaMA-3 and Qwen models, we study behavior across layers, sequence lengths, model scales, and 8- and 4-bit activation quantization. On our sampled prompts we see no collisions in full precision or at 8 bits, while 4-bit quantization induces a small number of collisions and markedly shrinks co-Lipschitz estimates. For a small GPT-2 trained from scratch, normalized metrics remain stable over training. Overall, the results suggest that Transformer representations are generically and persistently injective in the continuous-parameter idealization, while their practical invertibility can be probed using simple geometric diagnostics.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 16 of its 19 references that are in the map. First posted on arXiv 17 Nov 2025.
Builds on
16 of 19 references.
References · 16
- 2016Layer Normalizationcited by 5,669
- 2016Density estimation using Real NVPcited by 2,042
- 2018Glow: Generative Flow with Invertible 1x1 Convolutionscited by 1,545
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
- 2022RLPrompt: Optimizing Discrete Text Prompts with Reinforcement Learningcited by 366
- 2023Hard Prompts Made Easy: Gradient-Based Discrete Optimization for Prompt Tuning and Discoverycited by 336
- 2017The Reversible Residual Network: Backpropagation Without Storing Activationscited by 217
- 2023EvoPrompt: Connecting LLMs with Evolutionary Algorithms Yields Powerful Prompt Optimizerscited by 193
- 2023Text Embeddings Reveal (Almost) As Much As Textcited by 192
- 2017Breaking the Softmax Bottleneck: A High-Rank RNN Language Modelcited by 152
- 2023Language Model Inversioncited by 68
- 2022BBTv2: Towards a Gradient-Free Future with Large Language Modelscited by 53
4 more
- 2025The Non-Linear Representation Dilemma: Is Causal Abstraction Enough for Mechanistic Interpretability?cited by 37
- 2025Language Models are Injective and Hence Invertiblecited by 35
- 2025Better Language Model Inversion by Compactly Representing Next-Token Distributionscited by 9
- 2025On Surjectivity of Neural Networks: Can you elicit any behavior from your model?cited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Latent Object Permanence: Topological Phase Transitions, Free-Energy Principles, and Renormalization Group Flows in Deep Transformer Manifolds
- 2024Theoretical limitations of multi-layer Transformer
- 2025Exact Expressive Power of Transformers with Padding
- 2025Dynamical Properties of Tokens in Self-Attention and Effects of Positional Encoding
- 2026The Expressive Limits of Diagonal SSMs for State-Tracking
- 2026Every Bit Counts: A Theoretical Study of Precision-Expressivity Tradeoffs in Quantized Transformers
- 2026A Sharper Picture of Generalization in Transformers
- 2026Fully Interpretable Minimal Transformers: From Geometry to Algorithm