mlsanityData to 14 Oct 2026

Write Back the Delta: Revisiting the Same Tokens with Fresh Representations

arXiv 2609.32457 ↗First posted 26 Sep 2026cs.LGcited by 0

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

  1. 2017Attention Is All You Needcited by 60,361
  2. 2015Deep Residual Learning for Image Recognitioncited by 52,096
  3. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
  4. 2021LoRA: Low-Rank Adaptation of Large Language Modelscited by 14,380
  5. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,459
  6. 2021Training Verifiers to Solve Math Word Problemscited by 10,109
  7. 2025Qwen3 Technical Reportcited by 8,820
  8. 2020Measuring Massive Multitask Language Understandingcited by 7,963
  9. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,886
  10. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,100
  11. 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
  12. 2022Locating and Editing Factual Associations in GPTcited by 2,691
38 more
  1. 2020The Pile: An 800GB Dataset of Diverse Text for Language Modelingcited by 2,411
  2. 2025Gemma 3 Technical Reportcited by 1,824
  3. 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
  4. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
  5. 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
  6. 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,085
  7. 2016The LAMBADA dataset: Word prediction requiring a broad discourse contextcited by 992
  8. 2020On Layer Normalization in the Transformer Architecturecited by 922
  9. 2023Steering Language Models With Activation Engineeringcited by 922
  10. 2024Training Large Language Models to Reason in a Continuous Latent Spacecited by 797
  11. 2019Compressive Transformers for Long-Range Sequence Modellingcited by 726
  12. 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
  13. 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
  14. 2016Adaptive Computation Time for Recurrent Neural Networkscited by 506
  15. 2025Scaling up Test-Time Compute with Latent Reasoning: A Recurrent Depth Approachcited by 415
  16. 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
  17. 2021Efficient Attentions for Long Document Summarizationcited by 345
  18. 2024Quiet-STaR: Language Models Can Teach Themselves to Think Before Speakingcited by 301
  19. 2018Newsroom: A Dataset of 1.3 Million Summaries with Diverse Extractive Strategiescited by 294
  20. 2024Massive Activations in Large Language Modelscited by 294
  21. 2024Mixture-of-Depths: Dynamically allocating compute in transformer-based language modelscited by 258
  22. 2023Looped Transformers as Programmable Computerscited by 237
  23. 2024Patchscopes: A Unifying Framework for Inspecting Hidden Representations of Language Modelscited by 235
  24. 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 226
  25. 2024Let's Think Dot by Dot: Hidden Computation in Transformer Language Modelscited by 191
  26. 2021BookSum: A Collection of Datasets for Long-form Narrative Summarizationcited by 184
  27. 2023Implicit Chain of Thought Reasoning via Knowledge Distillationcited by 176
  28. 2021PonderNet: Learning to Pondercited by 168
  29. 2019BIGPATENT: A Large-Scale Dataset for Abstractive and Coherent Summarizationcited by 156
  30. 2023Looped Transformers are Better at Learning Learning Algorithmscited by 149
  31. 2025Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computationcited by 133
  32. 2024LaCo: Large Language Model Pruning via Layer Collapsecited by 126
  33. 2019BillSum: A Corpus for Automatic Summarization of US Legislationcited by 117
  34. 2024Steering Language Model Refusal with Sparse Autoencoderscited by 95
  35. 2026Attention Residualscited by 65
  36. 2025The Curse of Depth in Large Language Modelscited by 62
  37. 2025Understanding (Un)Reliability of Steering Vectors in Language Modelscited by 47
  38. 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

  1. 2026Multi-Stream Residual Computation in Diffusion Language Models
  2. 2026Trading Depth for Time in Recurrent Transformers
  3. 2025Change of Thought: Adaptive Test-Time Computation
  4. 2026AdaPonderLM: Gated Pondering Language Models with Token-Wise Adaptive Depth
  5. 2026BackPlay: Head-Only Look-Back Self-Correction for Diffusion Language Models
  6. 2025Compress, Gather, and Recompute: REFORMing Long-Context Processing in Transformers
  7. 2026Residual Streams Read, Recurrent States Remember: The Global Workspace in Mamba Models
  8. 2026DreamingGoose: Staged Distillation from Autoregressive Transformers to Bidirectional Recurrent Diffusion Language Models