How Does Reasoning Flow? Tracing Attention-Induced Information Flow for Targeted RL in LLMs
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Token-level credit assignment remains a key obstacle for reinforcement learning (RL) in large language models (LLMs), where RL recipes typically treat all tokens equally, failing to distinguish decisive reasoning steps from routine formatting or fluent filler. Recent attempts leverage model-internal signals to assign finer-grained credit, but these are often point-wise heuristics that ignore the global structure of information propagation. We propose FlowTracer, an RL framework that traces answer-targeted reasoning flow on an attention-induced directed acyclic graph in which nodes correspond to tokens and edge capacities come from aggregated attention weights and derives token credit from this global structure. The edge capacities are reweighted to retain only the influence that can reach the answer region, while enforcing local flow conservation so intermediate tokens neither lose nor gain effective mass due to path length or irrelevant branches. On this graph, FlowTracer extracts an information-flow backbone connecting the question to the answer and scores tokens by flow throughput, revealing high-impact hubs and aggregation checkpoints that mediate long-range dependencies. These derived importances are used to shape token-level rewards, enabling learning signals to focus precisely on the tokens that route information toward (or away from) correct answers and delivering consistent performance gains across a range of reasoning tasks.
Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 71 of its 82 references that are in the map. First posted on arXiv 9 Jun 2026.
Builds on
71 of 82 references.
Closest ancestors · 6
Same direction, newest first
- 2025Reasoning Palette: Modulating Reasoning via Latent Contextualization for Controllable Exploration for (V)LMs19 Dec
- 2025Attention Illuminates LLM Reasoning: The Preplan-and-Anchor Rhythm Enables Fine-Grained Policy Optimization15 Oct
- 2025Understanding Reasoning in Thinking Language Models via Steering Vectors22 Jun
- 2025Demystifying Reasoning Dynamics with Mutual Information: Thinking Tokens are Information Peaks in LLM Reasoning3 Jun
- 2025Seek in the Dark: Reasoning via Test-Time Instance-Level Policy Gradient in Latent Space19 May
- 2024Critical Tokens Matter: Token-Level Contrastive Estimation Enhances LLM's Reasoning Capability29 Nov
Other references · 65
- 2023GPT-4 Technical Reportcited by 18,396
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2021Training Verifiers to Solve Math Word Problemscited by 10,109
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
- 2017Deep reinforcement learning from human preferencescited by 3,912
- 2023Let's Verify Step by Stepcited by 3,789
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedbackcited by 3,720
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
38 more
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2019Fine-Tuning Language Models from Human Preferencescited by 2,150
- 2015High-Dimensional Continuous Control Using Generalized Advantage Estimationcited by 2,026
- 2024SWE-agent: Agent-Computer Interfaces Enable Automated Software Engineeringcited by 1,737
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
- 2025Kimi k1.5: Scaling Reinforcement Learning with LLMscited by 1,210
- 2024Qwen2.5-Coder Technical Reportcited by 1,203
- 2024OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific Problemscited by 1,136
- 2024Tulu 3: Pushing Frontiers in Open Language Model Post-Trainingcited by 1,053
- 2024SimPO: Simple Preference Optimization with a Reference-Free Rewardcited by 963
- 2023Math-Shepherd: Verify and Reinforce LLMs Step-by-step without Human Annotationscited by 928
- 2023Steering Language Models With Activation Engineeringcited by 922
- 2025Group Sequence Policy Optimizationcited by 835
- 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 767
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 623
- 2025Kimi K2: Open Agentic Intelligencecited by 567
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 559
- 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 528
- 2025The Entropy Mechanism of Reinforcement Learning for Reasoning Language Modelscited by 485
- 2022Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Taskcited by 431
- 2025Cognitive Behaviors that Enable Self-Improving Reasoners, or, Four Habits of Highly Effective STaRscited by 377
- 2025rStar-Math: Small LLMs Can Master Math Reasoning with Self-Evolved Deep Thinkingcited by 313
- 2025Reasoning with Exploration: An Entropy Perspectivecited by 281
- 2024DeepSeek-Prover-V1.5: Harnessing Proof Assistant Feedback for Reinforcement Learning and Monte-Carlo Tree Searchcited by 197
- 2023Inspecting and Editing Knowledge Representations in Language Modelscited by 141
- 2025LLMs Can Easily Learn to Reason from Demonstrations Structure, not content, is what matters!cited by 132
- 2024Attacks, Defenses and Evaluations for LLM Conversation Safety: A Surveycited by 130
- 2025Thought Anchors: Which LLM Reasoning Steps Matter?cited by 129
- 2025Process Reward Models That Thinkcited by 118
- 2025Reinforcement Learning Optimization for Large-Scale Learning: An Efficient and User-Friendly Scaling Librarycited by 101
- 2024LLM Internal States Reveal Hallucination Risk Faced With a Querycited by 82
- 2024Attention Heads of Large Language Models: A Surveycited by 67
- 2025Confidence Is All You Need: Few-Shot RL Fine-Tuning of Language Modelscited by 62
- 2024Unlocking the Capabilities of Thought: A Reasoning Boundary Framework to Quantify and Optimize Chain-of-Thoughtcited by 56
- 2024How to think step-by-step: A mechanistic understanding of chain-of-thought reasoningcited by 53
- 2025Stop Summation: Min-Form Credit Assignment Is All Process Reward Model Needs for Reasoningcited by 50
- 2024Identifying Semantic Induction Heads to Understand In-Context Learningcited by 47
- 2025Harnessing Uncertainty: Entropy-Modulated Policy Gradients for Long-Horizon LLM Agentscited by 46
50 most cited of 65 All 65 (JSON)
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026A Graph Signal Processing Perspective on Numerical Sequence Representations in LLM In-Context Learningcited by 0
- 2026Let Credit Follow Computation: Architecture-Aware Credit Transport for Large Language Model Reinforcement Learningcited by 0
- 2026Tracing the Evidence: Faithful Token Attribution Through Vision-Language Reasoningcited by 0
- 2026Tracing the Evidence: Faithful Token Attribution Through Vision-Language Reasoning29 Sep 2026
- 2026Let Credit Follow Computation: Architecture-Aware Credit Transport for Large Language Model Reinforcement Learning21 Aug 2026
- 2026A Graph Signal Processing Perspective on Numerical Sequence Representations in LLM In-Context Learning4 Aug 2026
Similar, not cited
Close in wording, no citation either way
- 2026Not All Tokens Learn Alike: Attention Entropy Reveals Heterogeneous Signals in RL Reasoning
- 2026Efficient Paths and Dense Rewards: Probabilistic Flow Reasoning for Large Language Models
- 2026Unrolled Flow Models for Reasoning
- 2026What Matters for Latent Reasoning with Flow Matching
- 2026Latent Thought Flow: Efficient Latent Reasoning in Large Language Models
- 2024Flow of Reasoning: Training LLMs for Divergent Reasoning with Minimal Examples
- 2026FlowEdit: Information-Theoretic Control of LLM Reasoning Flows for Ill-posed Problems Involving Conflicts
- 2025Interleaved Reasoning for Large Language Models via Reinforcement Learning