More Than Memory: Task-Conditioned Signed FFN Writes in Long-Context Retrieval
Not on the map. Why?
Abstract (arXiv)
FFNs are often treated as parametric memories. In long-context retrieval, however, the sharper question is not only what they store, but whether their native residual writes push the current retrieval state toward or away from the correct answer. We test this by scaling the model's own FFN write one layer at a time, without editing weights or injecting external steering vectors. Across controlled literal and semantic retrieval suites, native FFN response surfaces are signed, layer-specific, and task-conditioned: the final FFN is a suppressor in 7 of 8 model-suite cases, and 60% of layers switch role between retrieval modes (95% CI [50%, 69%]). A local directional derivative along the native write separates the two monotone roles: suppressors have negative derivative in 34/35 cases, and amplifiers have positive derivative in 18/18 cases, so the roles are not reducible to write size. On a safety-filtered LongBench retrieval-QA probe, the same diagnostic predicts attenuation damage with raw R^2=0.796 on Qwen2.5-7B and 0.791 on Qwen3.5-9B; a held-out suppressor-attenuation policy improves retrieval margins over random and norm-matched controls. These results show that native FFN scaling exposes a signed, task-conditioned residual-write structure in retrieval, and that write-gradient alignment is a compact diagnostic for the two monotone roles.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 23 of its 30 references that are in the map. First posted on arXiv 27 Jun 2026.
Builds on
23 of 30 references.
References · 23
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2023LongBench: A Bilingual, Multitask Benchmark for Long Context Understandingcited by 1,417
- 2024RULER: What's the Real Context Size of Your Long-Context Language Models?cited by 1,346
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,336
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
- 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,085
- 2023Steering Language Models With Activation Engineeringcited by 922
- 2021Knowledge Neurons in Pretrained Transformerscited by 601
- 2022Transformer Feed-Forward Layers Build Predictions by Promoting Concepts in the Vocabulary Spacecited by 532
- 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 528
11 more
- 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
- 2024∞Bench: Extending Long Context Evaluation Beyond 100K Tokenscited by 367
- 2022Extracting Latent Steering Vectors from Pretrained Language Modelscited by 225
- 2025NoLiMa: Long-Context Evaluation Beyond Literal Matchingcited by 126
- 2026Conditional Memory via Scalable Lookup: A New Axis of Sparsity for Large Language Modelscited by 86
- 2024Memory Layers at Scalecited by 48
- 2023One Wide Feedforward is All You Needcited by 15
- 2026STEM: Scaling Transformers with Embedding Modulescited by 12
- 2025Layerwise Importance Analysis of Feed-Forward Networks in Transformer-based Language Modelscited by 6
- 2025What Can We Actually Steer? A Multi-Behavior Study of Activation Controlcited by 6
- 2026MemoryLLM: Plug-n-Play Interpretable Feed-Forward Memory for Transformerscited by 5
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026From Interpretability to Performance: Optimizing Retrieval Heads for Long-Context Language Models
- 2026Logit-Contribution Scoring Identifies Non-Literal Retrieval Heads
- 2025Memory in Large Language Models: Mechanisms, Evaluation and Evolution
- 2025Demystifying the Roles of LLM Layers in Retrieval, Knowledge, and Reasoning
- 2026Memory Depth, Not Memory Access: Selective Parametric Consolidation for Long-Running Language Agents
- 2026History Replays Itself: Unleashing Long-Context Capability in Linear Attention
- 2026Diagnosing Retrieval vs. Utilization Bottlenecks in LLM Agent Memory
- 2024Writing in the Margins: Better Inference Pattern for Long Context Retrieval