Full-bandwidth transformer
Not on the map. Why?
Abstract (arXiv)
Autoregressive transformers compute along two axes: horizontally across generated tokens, and vertically through model depth. Dense attention gives each token broad horizontal access to the past, but the vertical feedback channel between decoding steps remains narrow: only the sampled token returns to the bottom of the stack, while the top-layer hidden state is discarded. We introduce the full-bandwidth transformer, which widens this channel with latent feedback: at each decoding step, the previous top-layer hidden state is fused with the sampled token embedding through a gated linear unit and fed back as the next input. Latent feedback lets non-verbalized computation re-enter the stack with a renewed depth budget, while preserving the standard transformer architecture, KV cache, and language-modeling objective. To train full-bandwidth transformers without losing parallel teacher forcing, we use a scheduled multi-pass objective that introduces latent feedback late in pretraining and mixes a small fraction of deeper feedback passes for stability. We train 1B-parameter full-bandwidth transformers on up to 400B tokens and find that latent feedback improves validation loss, 5-shot language-model evaluation, math and coding generation, and instruction-tuned performance. With negligible per-token decoding overhead, full-bandwidth transformers match or approach standard transformers trained with roughly 1.5x more tokens, and manage to produce shorter reasoning when no off-policy templates are provided.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 51 of its 53 references that are in the map. First posted on arXiv 9 Aug 2026.
Builds on
51 of 53 references.
References · 51
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2022PaLM: Scaling Language Modeling with Pathwayscited by 5,304
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2023Let's Verify Step by Stepcited by 3,598
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2023Mistral 7Bcited by 3,065
- 2023Instruction-Following Evaluation for Large Language Modelscited by 1,143
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 763
38 more
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
- 2024Phi-4 Technical Reportcited by 675
- 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertaintycited by 564
- 2024Gated Delta Networks: Improving Mamba2 with Delta Rulecited by 483
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 372
- 2024Better & Faster Large Language Models via Multi-token Predictioncited by 344
- 2025Monitoring Reasoning Models for Misbehavior and the Risks of Promoting Obfuscationcited by 338
- 2024Chain of Thought Empowers Transformers to Solve Inherently Serial Problemscited by 256
- 2023Looped Transformers as Programmable Computerscited by 224
- 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
- 2025Scaling Latent Reasoning via Looped Language Modelscited by 145
- 2022Signal Propagation in Transformers: Theoretical Perspectives and the Role of Rank Collapsecited by 136
- 2025Soft Thinking: Unlocking the Reasoning Potential of LLMs in Continuous Concept Spacecited by 109
- 2025WritingBench: A Comprehensive Benchmark for Generative Writingcited by 95
- 2023Tensor Programs VI: Feature Learning in Infinite-Depth Neural Networkscited by 94
- 2025NorMuon: Making Muon more efficient and scalablecited by 74
- 2025Stabilizing MoE Reinforcement Learning by Aligning Training and Inference Routerscited by 67
- 2026MOPD: Multi-Teacher On-Policy Distillation for Capability Integration in LLM Post-Trainingcited by 60
- 2025Hybrid Latent Reasoning via Reinforcement Learningcited by 30
- 2025Next-Latent Prediction Transformers Learn Compact World Modelscited by 29
- 2025Why Gradients Rapidly Increase Near the End of Trainingcited by 21
- 2026Fixed-Point Reasoners: Stable and Adaptive Deep Looped Transformerscited by 21
- 2025EvoLM: In Search of Lost Language Model Training Dynamicscited by 16
- 2025PonderLM-2: Pretraining LLM with Latent Thoughts in Continuous Spacecited by 14
- 2025ParaRNN: Unlocking Parallel Training of Nonlinear RNNs for Large Language Modelscited by 13
- 2026Bridging the Gap Between Latent and Explicit Reasoning with Looped Transformerscited by 11
- 2025Efficient Joint Prediction of Multiple Future Tokenscited by 10
- 2026Arcee Trinity Large Technical Reportcited by 10
- 2026Next Concept Prediction in Discrete Latent Space Leads to Stronger Language Modelscited by 8
- 2026Quantifying the Necessity of Chain of Thought through Opaque Serial Depthcited by 7
- 2025Context-level Language Modeling by Learning Predictive Context Embeddingscited by 6
- 2026T^2MLR: Transformer with Temporal Middle-Layer Recurrencecited by 6
- 2026Latent Recurrent Transformer: Architecture Exploration, Training Strategies, and Scaling Behaviorcited by 5
- 2026Dense Supervision Is Not Enough: The Readout Blind Spot in Looped Language Modelscited by 5
- 2026NITP: Next Implicit Token Prediction for LLM Pre-trainingcited by 3
- 2026Maglev: Sliding Recurrent Memorycited by 3
- 2026LoopMTP: A looped transformer guided by latent multi-token predictioncited by 3
- 2026How Transparent is DiffusionGemma?cited by 2
50 most cited of 51 All 51 (JSON)
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Thinking with Looped Flowscited by 6
- 2026Write Back the Delta: Revisiting the Same Tokens with Fresh Representationscited by 0
- 2026Looped Transformers as Optimizerscited by 0
- 2026Looped Transformers as Optimizers29 Sep 2026
- 2026Write Back the Delta: Revisiting the Same Tokens with Fresh Representations26 Sep 2026
- 2026Thinking with Looped Flows10 Sep 2026
Similar, not cited
Close in wording, no citation either way
- 2026The Recurrent Transformer: Greater Effective Depth and Efficient Decoding
- 2026SNLP: Layer-Parallel Inference via Structured Newton Corrections
- 2026Revisiting the Shape Convention of Transformer Language Models
- 2026Pipeline-Native Transformers: Co-Designing Model Architecture and CPU Inference for Bandwidth-Efficient Autoregressive Decode
- 2026The Context-Ready Transformer
- 2025Latent Flow Transformer
- 2025Improving Model Representation and Reducing KV Cache via Skip Connections with First Value Heads
- 2024The Asymptotic Behavior of Attention in Transformers