Write Back the Delta: Revisiting the Same Tokens with Fresh Representations
Not on the map. Why?
Abstract (arXiv)
Transformers process information strictly forward through depth, preventing deeper computation from revisiting and refining earlier representations. To augment the standard forward pass, existing approaches either re-execute depth, incurring additional computation, or modify the residual stream using predefined directions, limiting their instance-level adaptation. Recently, inference-time feedback offers a direct mechanism for recycling endogenously produced computation by writing deeper residual states back to earlier layers, yet what should be fed back remains unclear. We argue that the depth increment Delta, capturing newly accumulated computation between two layers, provides a more effective, composable, and scalable feedback signal than the full state. Building on this observation, we introduce ReFlux, a learnable feedback graph that dynamically selects and composes increment-carrying routes. ReFlux supports synchronous feedback to the same token and streaming feedback to subsequent tokens. Extensive experiments across various models, corpora, and benchmarks show that synchronous ReFlux consistently reduces perplexity across ten language-modeling corpora, and improves accuracy by 2.1-2.3 points, with gains reaching 4.7 points on multi-hop reasoning. Streaming ReFlux further retains most of these gains while preserving the base model's 1x theoretical backbone FLOPs. These results establish ReFlux as an efficient paradigm for unlocking the latent computational potential of LLMs, allowing them to revisit the same tokens with fresh representations. Code implementation can be found at https://github.com/gooogleshanghai/reflux.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 64 of its 73 references that are in the map. First posted on arXiv 26 Sep 2026.
Builds on
64 of 73 references.
References · 64
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
- 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2025Qwen3 Technical Reportcited by 8,820
- 2020Measuring Massive Multitask Language Understandingcited by 7,963
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
38 more
- 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
- 2025Gemma 3 Technical Reportcited by 1,824
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
- 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,085
- 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
- 2020On Layer Normalization in the Transformer Architecturecited by 922
- 2023Steering Language Models With Activation Engineeringcited by 922
- 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
- 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
- 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
- 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
- 2021Efficient Attentions for Long Document Summarizationcited by 345
- 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 301
- 2018Newsroom: A Dataset of 1.3 Million Summaries with Diverse Extractive Strategiescited by 294
- 2024Massive Activations in Large Language Modelscited by 294
- 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 258
- 2023Looped Transformers as Programmable Computerscited by 237
- 2024Patchscopes: A Unifying Framework for Inspecting Hidden Representations of Language Modelscited by 235
- 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 226
- 2024Let's Think Dot by Dot: Hidden Computation in Transformer Language Modelscited by 191
- 2021BookSum: A Collection of Datasets for Long-form Narrative Summarizationcited by 184
- 2023Implicit Chain of Thought Reasoning via Knowledge Distillationcited by 176
- 2021PonderNet: Learning to Pondercited by 168
- 2019BIGPATENT: A Large-Scale Dataset for Abstractive and Coherent Summarizationcited by 156
- 2023Looped Transformers are Better at Learning Learning Algorithmscited by 149
- 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 133
- 2024LaCo: Large Language Model Pruning via Layer Collapsecited by 126
- 2019BillSum: A Corpus for Automatic Summarization of US Legislationcited by 117
- 2024Steering Language Model Refusal with Sparse Autoencoderscited by 95
- 2026Attention Residualscited by 65
- 2025The Curse of Depth in Large Language Modelscited by 62
- 2025Understanding (Un)Reliability of Steering Vectors in Language Modelscited by 47
- 2024Steering Without Side Effects: Improving Post-Deployment Control of Language Modelscited by 46
50 most cited of 64 All 64 (JSON)
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Multi-Stream Residual Computation in Diffusion Language Models
- 2026Trading Depth for Time in Recurrent Transformers
- 2025Change of Thought: Adaptive Test-Time Computation
- 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depth
- 2026BackPlay: Head-Only Look-Back Self-Correction for Diffusion Language Models
- 2025Compress, Gather, and Recompute: REFORMing Long-Context Processing in Transformers
- 2026Residual Streams Read, Recurrent States Remember: The Global Workspace in Mamba Models
- 2026DreamingGoose: Staged Distillation from Autoregressive Transformers to Bidirectional Recurrent Diffusion Language Models