Universal Transformers
Not on the map. Why?
Cited by 98 in LLM reasoning, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
Recurrent neural networks (RNNs) sequentially process data by updating their state with each new data point, and have long been the de facto choice for sequence modeling tasks. However, their inherently sequential computation makes them slow to train. Feed-forward and convolutional architectures have recently been shown to achieve superior results on some sequence modeling tasks such as machine translation, with the added advantage that they concurrently process all inputs in the sequence, leading to easy parallelization and faster training times. Despite these successes, however, popular feed-forward sequence models like the Transformer fail to generalize in many simple tasks that recurrent models handle with ease, e.g. copying strings or even simple logical inference when the string or formula lengths exceed those observed at training time. We propose the Universal Transformer (UT), a parallel-in-time self-attentive recurrent sequence model which can be cast as a generalization of the Transformer model and which addresses these issues. UTs combine the parallelizability and global receptive field of feed-forward sequence models like the Transformer with the recurrent inductive bias of RNNs. We also add a dynamic per-position halting mechanism and find that it improves accuracy on several tasks. In contrast to the standard Transformer, under certain assumptions, UTs can be shown to be Turing-complete. Our experiments show that UTs outperform standard Transformers on a wide range of algorithmic and language understanding tasks, including the challenging LAMBADA language modeling task where UTs achieve a new state of the art, and machine translation where UTs achieve a 0.9 BLEU improvement over Transformers on the WMT14 En-De dataset.
Key facts. Cited by 647 papers in the map (data through 14 Oct 2026). First posted on arXiv 10 Jul 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 99; 2022: 78; 2023: 80; 2024: 61; 2025: 84; 2026: 245.
647 papers.
Sort:
By area:
- 2023GPT-4 Technical Reportcited by 18,396
- 2022Emergent Abilities of Large Language Modelscited by 2,512
- 2021ViViT: A Video Vision Transformercited by 1,620
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,104
- 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2023LightGlue: Local Feature Matching at Light Speedcited by 590
- 2022Transformers in Time Series: A Surveycited by 497
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2021Dynamic Neural Networks: A Surveycited by 408
38 more
- 2023Scaling Data-Constrained Language Modelscited by 403
- 2021A Survey of Transformerscited by 401
- 2021Random Feature Attentioncited by 354
- 2021Choose a Transformer: Fourier or Galerkincited by 343
- 2023Trained Transformers Learn Linear Models In-Contextcited by 337
- 2022Confident Adaptive Language Modelingcited by 310
- 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
- 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 294
- 2021TransVG: End-to-End Visual Grounding with Transformerscited by 271
- 2022Transformers Learn Shortcuts to Automatacited by 268
- 2021I-BERT: Integer-only BERT Quantizationcited by 255
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 240
- 2023Looped Transformers as Programmable Computerscited by 237
- 2022Exploring Length Generalization in Large Language Modelscited by 206
- 2022The Parallelism Tradeoff: Limitations of Log-Precision Transformerscited by 193
- 2021Thinking Like Transformerscited by 184
- 2022Outlier Suppression: Pushing the Limit of Low-bit Transformer Language Modelscited by 181
- 2025Scaling Latent Reasoning via Looped Language Modelscited by 165
- 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 146
- 2021Deep Learning Through the Lens of Example Difficultycited by 141
- 2021Understanding and Overcoming the Challenges of Efficient Transformer Quantizationcited by 141
- 2021Investigating the Limitations of Transformers with Simple Arithmetic Taskscited by 139
- 2021The Benchmark Lotterycited by 136
- 2022Language Models as Agent Modelscited by 132
- 2021AMMUS : A Survey of Transformer-based Pretrained Models in Natural Language Processingcited by 131
- 2025Advances and Challenges in Foundation Agents: From Brain-Inspired Intelligence to Evolutionary, Collaborative, and Safe Systemscited by 126
- 2024Zamba: A Compact 7B SSM Hybrid Modelcited by 124
- 2022RSVG: Exploring Data and Models for Visual Grounding on Remote Sensing Datacited by 123
- 2021The Devil is in the Detail: Simple Tricks Improve Systematic Generalization of Transformerscited by 117
- 2023Randomized Positional Encodings Boost Length Generalization of Transformerscited by 117
- 2025Towards System 2 Reasoning in LLMs: Learning How to Think With Meta Chain-of-Thoughtcited by 109
- 2021Self-Attention Networks Can Process Bounded Hierarchical Languagescited by 105
- 2021Position Information in Transformers: An Overviewcited by 103
- 2022Scaling Laws vs Model Architectures: How does Inductive Bias Influence Scaling?cited by 103
- 2023Scan and Snap: Understanding Training Dynamics and Token Composition in 1-layer Transformercited by 102
- 2021Finetuning Pretrained Transformers into RNNscited by 96
- 2021Luna: Linear Unified Nested Attentioncited by 96
- 2023To Repeat or Not To Repeat: Insights from Scaling LLM under Token-Crisiscited by 92
50 most cited of 647 All 647 (JSON)
- 2026When Does Depth Help? Understanding Layerwise Inference in Tabular Foundation Models14 Oct 2026
- 2026Learning to Compute Is Not Learning to Retain or Reuse in Looped Transformers14 Oct 2026
- 2026DAGFormer: Global Topology Prediction for Transformer Cross-Layer Connections14 Oct 2026
- 2026Looped Reasoning Finishes Earlier Than You Think13 Oct 2026
- 2026LoOPSD: On-Policy Self-Distillation with Looped Teachers13 Oct 2026
- 2026LoopREG: Representation Supervision Enables Test-Time Recurrent Depth in Diffusion Transformers13 Oct 2026
- 2026Resample the Answer: Where to Spend Test-Time Compute in Recursive Reasoning12 Oct 2026
- 2026Running Fewer Loops Removes Refusal Behaviour in Recurrent-Depth Language Models10 Oct 2026
- 2026Recurrent In-Context Learning for Tabular Foundation Models10 Oct 2026
- 2026Learning to Extrapolate in Depth with Looped Transformers10 Oct 2026
- 2026Scaling RNA Foundation Models through Iterative Refinement9 Oct 2026
- 2026One Model to Loop Them All: Elastic Latent Recurrence for Language Models8 Oct 2026
38 more
- 2026Think Before You Paint: Recursive Latent Reasoning for Diffusion Models7 Oct 2026
- 2026LayerRoPE: Dynamic Depth-wise Magnitude & Angular Superposition6 Oct 2026
- 2026Enabling Dynamic Computation in Looped LMs6 Oct 2026
- 2026GeneICL: A Tabular Foundation Model for Bulk Transcriptomics6 Oct 2026
- 2026SanSi: A Looped Typed Decision Model for System 1.5 Thinking6 Oct 2026
- 2026Recurrent Looped Transformer6 Oct 2026
- 2026SPECTRUM: Proximal Spectral Modulation for Looped Self-Distillation5 Oct 2026
- 2026Towards Looped Models Done Right, Part II: Rethinking at Fixed Points5 Oct 2026
- 2026LiFT: Loop Flow Transformers4 Oct 2026
- 2026Universal Test-Time Training4 Oct 2026
- 2026Loopy: Low-Bit Quantization Framework for Looped Language Models4 Oct 2026
- 2026LoopMoEVR: Loop-Based Degradation-Aware Mixture-of-Experts for Unified UHD Video Restoration4 Oct 2026
- 2026AdaLoop: Adaptive-Depth Latent Reasoning for Audio Language Models3 Oct 2026
- 2026Dynamic Routing as a New Dimension for Test-time Versatility of LLMs3 Oct 2026
- 2026ALoDLM: Adaptively Looped Diffusion Language Models3 Oct 2026
- 2026Beyond the Parameter Monolith: Reconstructive Memories, Executable Skills, and Residual Assembly for Language Models2 Oct 2026
- 2026Depth as Time in One-Step Generative Models2 Oct 2026
- 2026Refinement Buys Intelligibility, Search Buys Identity: What Test-Time Compute Buys in Masked-Diffusion TTS2 Oct 2026
- 2026The Surprising Effectiveness of Shared Memory in Looped Transformers1 Oct 2026
- 2026Decoding Looped Transformers Better for (Almost) Free1 Oct 2026
- 2026Architectural Sampling: Test-Time Scaling via Computational Diversity in Frozen Vision-Language Models1 Oct 2026
- 2026Looping Beyond Twice: A Scalable Recipe for Looped Mixture-of-Experts1 Oct 2026
- 2026Rotate the Loop: Blockwise Conformality for Looped Transformers30 Sep 2026
- 2026Closing the Loop: Practical Training Recipes for Looped Language Models30 Sep 2026
- 2026DeepJEPA: Scaling World Models from Within30 Sep 2026
- 2026Shared Weights, Selected Computations: How Looped Transformers Route What Each Loop Does30 Sep 2026
- 2026LatentHarness: Learning Latent Actions for Memory and Reasoning via Counterfactual Policy Distillation30 Sep 2026
- 2026Towards Better Exploration in Sequential Test-Time Scaling30 Sep 2026
- 2026A Tilted Bowl Is Not a Slippery Slope: Compressing Looped Models30 Sep 2026
- 2026LoopVL: Recurrent Visual Intelligence29 Sep 2026
- 2026Function-Space Transformer with Adaptive Anchors29 Sep 2026
- 2026Pretraining Latent Information Feedback Transformers with Teacher Supervision29 Sep 2026
- 2026Pattern Formation in Transformers29 Sep 2026
- 2026Procedural Core: A Compact Recurrent Initialization for Vision Transformers29 Sep 2026
- 2026Looped Transformers as Optimizers29 Sep 2026
- 2026Transformers Stop Thinking Too Early, and a Tiny LoRA Fixes It29 Sep 2026
- 2026LoopICL: Looping a single transformer block to solve tabular tasks28 Sep 2026
- 2026How to Loop MoE: Flatten the Experts, Untie the Attention28 Sep 2026
50 newest of 647 All 647 (JSON)
- 2021Show Your Work: Scratchpads for Intermediate Computation with Language Modelscited by 886
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2023Towards Revealing the Mystery behind Chain of Thought: A Theoretical Perspectivecited by 294
- 2025Reasoning with Latent Thoughts: On the Power of Looped Transformerscited by 240
- 2025Scaling Latent Reasoning via Looped Language Modelscited by 165
- 2025Towards System 2 Reasoning in LLMs: Learning How to Think With Meta Chain-of-Thoughtcited by 109
- 2025A Survey on Latent Reasoningcited by 81
- 2024Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalizationcited by 79
- 2024Learning How Hard to Think: Input-Adaptive Allocation of LM Computationcited by 70
- 2022StepGame: A New Benchmark for Robust Multi-Hop Spatial Reasoning in Textscited by 68
- 2023Coupling Large Language Models with Logic Programming for Robust and General Reasoning from Textcited by 59
- 2025Inner Thinking Transformer: Leveraging Dynamic Depth Scaling to Foster Adaptive Internal Thinkingcited by 57
12 most cited of 98 All 98 (JSON)
- 2026Resample the Answer: Where to Spend Test-Time Compute in Recursive Reasoning12 Oct 2026
- 2026Think Before You Paint: Recursive Latent Reasoning for Diffusion Models7 Oct 2026
- 2026SanSi: A Looped Typed Decision Model for System 1.5 Thinking6 Oct 2026
- 2026Towards Looped Models Done Right, Part II: Rethinking at Fixed Points5 Oct 2026
- 2026AdaLoop: Adaptive-Depth Latent Reasoning for Audio Language Models3 Oct 2026
- 2026Dynamic Routing as a New Dimension for Test-time Versatility of LLMs3 Oct 2026
- 2026Refinement Buys Intelligibility, Search Buys Identity: What Test-Time Compute Buys in Masked-Diffusion TTS2 Oct 2026
- 2026Architectural Sampling: Test-Time Scaling via Computational Diversity in Frozen Vision-Language Models1 Oct 2026
- 2026LatentHarness: Learning Latent Actions for Memory and Reasoning via Counterfactual Policy Distillation30 Sep 2026
- 2026Towards Better Exploration in Sequential Test-Time Scaling30 Sep 2026
- 2026MoR-MLLM: Mixture of Recursions for Efficient Multimodal Large Language Models27 Sep 2026
- 2026Beyond the Training Horizon: Mechanisms and Limits of Length Generalization in Looped Transformers27 Sep 2026
12 newest of 98 All 98 (JSON)
- 2024OLMoE: Open Mixture-of-Experts Language Modelscited by 301
- 2023To Repeat or Not To Repeat: Insights from Scaling LLM under Token-Crisiscited by 92
- 2024MoEUT: Mixture-of-Experts Universal Transformerscited by 59
- 2021Go Wider Instead of Deepercited by 57
- 2025Parameters vs FLOPs: Scaling Laws for Optimal Sparsity for Mixture-of-Experts Language Modelscited by 50
- 2023Sparse Universal Transformercited by 32
- 2024Shortcut-connected Expert Parallelism for Accelerating Mixture-of-Expertscited by 32
- 2024Mixture of Nested Experts: Adaptive Processing of Visual Tokenscited by 25
- 2026SMELT: Scaling Laws for Compute-Matched MoE Looped Transformerscited by 18
- 2026Training-Free Looped Transformerscited by 15
- 2025Chain-of-Experts: Unlocking the Communication Power of Mixture-of-Experts Modelscited by 9
- 2026Mixture of Universal Experts: Scaling Virtual Width via Depth-Width Transformationcited by 8
12 most cited of 28 All 28 (JSON)
- 2026One Model to Loop Them All: Elastic Latent Recurrence for Language Models8 Oct 2026
- 2026Universal Test-Time Training4 Oct 2026
- 2026LoopMoEVR: Loop-Based Degradation-Aware Mixture-of-Experts for Unified UHD Video Restoration4 Oct 2026
- 2026Looping Beyond Twice: A Scalable Recipe for Looped Mixture-of-Experts1 Oct 2026
- 2026How to Loop MoE: Flatten the Experts, Untie the Attention28 Sep 2026
- 2026MoRE: Mixture of Reused Experts16 Sep 2026
- 2026SMELT: Scaling Laws for Compute-Matched MoE Looped Transformers1 Sep 2026
- 2026Improving Parameter Utilization by Sharing Neural Experts Across Layers in Transformers31 Aug 2026
- 2026Loop the Loopies!17 Jul 2026
- 2026TriRoute: Unified Learned Routing for Joint Adaptive Attention, Experts, and KV-Cache Allocation7 Jul 2026
- 2026Tying the Loop -- Tied Expert Layers in Mixture-of-Experts Language Models15 Jun 2026
- 2026Training-Free Looped Transformers22 May 2026
12 newest of 28 All 28 (JSON)
- 2022Confident Adaptive Language Modelingcited by 310
- 2022Token Dropping for Efficient BERT Pretrainingcited by 42
- 2021TR-BERT: Dynamic Token Reduction for Accelerating BERT Inferencecited by 40
- 2022A Simple Hash-Based Early Exiting Approach For Language Understanding and Generationcited by 32
- 2024Block Transformer: Global-to-Local Language Modeling for Fast Inferencecited by 32
- 2026Memory-Efficient Looped Transformer: Decoupling Compute from Memory in Looped Language Modelscited by 14
- 2023Revisiting Token Dropping Strategy in Efficient BERT Pretrainingcited by 11
- 2024Performance Control in Early Exiting to Deploy Large Models at the Same Cost of Smaller Onescited by 4
- 2026LoopCoder-v2: Only Loop Once for Efficient Test-Time Computation Scalingcited by 4
- 2026Scratchpad Patching: Decoupling Compute from Patch Size in Byte-Level Language Modelscited by 1
- 2025Subjective Depth and Timescale Transformers: Learning Where and When to Computecited by 0
- 2026TIDE: Token-Informed Depth Execution for Per-Token Early Exit in LLM Inferencecited by 0
12 most cited of 16 All 16 (JSON)
- 2026Enabling Dynamic Computation in Looped LMs6 Oct 2026
- 2026WaveFront Decoding: Parallelized Self-Speculative Decoding for Looped Language Models19 Sep 2026
- 2026MergeOver: Post-Training Token Merging for Recursive Vision Transformers13 Aug 2026
- 2026LoopCoder-v2: Only Loop Once for Efficient Test-Time Computation Scaling16 Jun 2026
- 2026Scratchpad Patching: Decoupling Compute from Patch Size in Byte-Level Language Models10 May 2026
- 2026Memory-Efficient Looped Transformer: Decoupling Compute from Memory in Looped Language Models8 May 2026
- 2026LEAP: Layer-wise Exit-Aware Pretraining for Efficient Transformer Inference1 May 2026
- 2026TIDE: Token-Informed Depth Execution for Per-Token Early Exit in LLM Inference22 Mar 2026
- 2025Subjective Depth and Timescale Transformers: Learning Where and When to Compute26 Nov 2025
- 2024Performance Control in Early Exiting to Deploy Large Models at the Same Cost of Smaller Ones26 Dec 2024
- 2024Block Transformer: Global-to-Local Language Modeling for Fast Inference4 Jun 2024
- 2023Revisiting Token Dropping Strategy in Efficient BERT Pretraining24 May 2023
12 newest of 16 All 16 (JSON)
- 2025Efficient Parallel Samplers for Recurrent-Depth Models and Their Connection to Diffusion Language Modelscited by 7
- 2026Forward-Free Diffusion Language Models with BPTT-Free Looped Refinementcited by 2
- 2026Recursive Scaling in Masked Diffusion Modelscited by 2
- 2026Don't Commit Alone: Joint Token Commitment in Diffusion Language Modelscited by 2
- 2026Looped Diffusion Language Modelscited by 1
- 2026Depth through recurrence: Looped transformers for flow-matching TTScited by 1
- 2026LiFT: Loop Flow Transformerscited by 1
- 2022Changes from Classical Statistics to Modern Statistics and Data Sciencecited by 0
- 2026A one-step generation model with a Single-Layer Transformer: Layer number re-distillation of FreeFlowcited by 0
- 2026Training-Free Hidden-State Refinement for Flow-Matching Image Generatorscited by 0
- 2026ALoDLM: Adaptively Looped Diffusion Language Modelscited by 0
- 2026LoopREG: Representation Supervision Enables Test-Time Recurrent Depth in Diffusion Transformerscited by 0
- 2026LoopREG: Representation Supervision Enables Test-Time Recurrent Depth in Diffusion Transformers13 Oct 2026
- 2026LiFT: Loop Flow Transformers4 Oct 2026
- 2026ALoDLM: Adaptively Looped Diffusion Language Models3 Oct 2026
- 2026Depth through recurrence: Looped transformers for flow-matching TTS24 Sep 2026
- 2026Training-Free Hidden-State Refinement for Flow-Matching Image Generators29 Aug 2026
- 2026Don't Commit Alone: Joint Token Commitment in Diffusion Language Models5 Jul 2026
- 2026Recursive Scaling in Masked Diffusion Models16 Jun 2026
- 2026Forward-Free Diffusion Language Models with BPTT-Free Looped Refinement6 Jun 2026
- 2026Looped Diffusion Language Models25 May 2026
- 2026A one-step generation model with a Single-Layer Transformer: Layer number re-distillation of FreeFlow14 Jan 2026
- 2025Efficient Parallel Samplers for Recurrent-Depth Models and Their Connection to Diffusion Language Models16 Oct 2025
- 2022Changes from Classical Statistics to Modern Statistics and Data Science30 Oct 2022
- 2026Looped World Modelscited by 3
- 2026Can Video World Models Track Unobserved World States?cited by 2
- 2026Adaptive Compute in Latent World Models: When Depth Helps, Hurts, or Doesn't Mattercited by 1
- 2024When in Doubt, Think Slow: Iterative Reasoning with Latent Imaginationcited by 0
- 2026DeepJEPA: Scaling World Models from Withincited by 0
- 2026DeepJEPA: Scaling World Models from Within30 Sep 2026
- 2026Can Video World Models Track Unobserved World States?31 Aug 2026
- 2026Adaptive Compute in Latent World Models: When Depth Helps, Hurts, or Doesn't Matter11 Jul 2026
- 2026Looped World Models16 Jun 2026
- 2024When in Doubt, Think Slow: Iterative Reasoning with Latent Imagination23 Feb 2024
- 2025Mamba Modulation: On the Length Generalization of Mambacited by 16
- 2025Fixed-Point RNNs: Interpolating from Diagonal to Densecited by 5
- 2026Why Depth Matters in Parallelizable Sequence Models: A Lie Algebraic Viewcited by 1
- 2026Loop-Mamba: A Loop Mamba with Degradation-Aware and Shared Memory for Old Photo Restorationcited by 1
- 2026Loop-Mamba: A Loop Mamba with Degradation-Aware and Shared Memory for Old Photo Restoration3 Aug 2026
- 2026Why Depth Matters in Parallelizable Sequence Models: A Lie Algebraic View5 Mar 2026
- 2025Mamba Modulation: On the Length Generalization of Mamba23 Sep 2025
- 2025Fixed-Point RNNs: Interpolating from Diagonal to Dense13 Mar 2025
- 2023GPT-4 Technical Reportcited by 18,396
- 2022Emergent Abilities of Large Language Modelscited by 2,512
- 2021ViViT: A Video Vision Transformercited by 1,620
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,104
- 2022What Can Transformers Learn In-Context? A Case Study of Simple Function Classescited by 715
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
- 2023LightGlue: Local Feature Matching at Light Speedcited by 590
- 2022Transformers in Time Series: A Surveycited by 497
- 2021Attention is Not All You Need: Pure Attention Loses Rank Doubly Exponentially with Depthcited by 419
- 2021Dynamic Neural Networks: A Surveycited by 408
- 2023Scaling Data-Constrained Language Modelscited by 403
- 2021A Survey of Transformerscited by 401
12 most cited of 468 All 468 (JSON)
- 2026When Does Depth Help? Understanding Layerwise Inference in Tabular Foundation Models14 Oct 2026
- 2026Learning to Compute Is Not Learning to Retain or Reuse in Looped Transformers14 Oct 2026
- 2026DAGFormer: Global Topology Prediction for Transformer Cross-Layer Connections14 Oct 2026
- 2026Looped Reasoning Finishes Earlier Than You Think13 Oct 2026
- 2026Recurrent In-Context Learning for Tabular Foundation Models10 Oct 2026
- 2026Learning to Extrapolate in Depth with Looped Transformers10 Oct 2026
- 2026Scaling RNA Foundation Models through Iterative Refinement9 Oct 2026
- 2026LayerRoPE: Dynamic Depth-wise Magnitude & Angular Superposition6 Oct 2026
- 2026GeneICL: A Tabular Foundation Model for Bulk Transcriptomics6 Oct 2026
- 2026Recurrent Looped Transformer6 Oct 2026
- 2026Loopy: Low-Bit Quantization Framework for Looped Language Models4 Oct 2026
- 2026Beyond the Parameter Monolith: Reconstructive Memories, Executable Skills, and Residual Assembly for Language Models2 Oct 2026
12 newest of 468 All 468 (JSON)