Identifying and Transferring Reasoning-Critical Neurons: Improving LLM Inference Reliability via Activation Steering
On the map: LLM reasoning → Test-time compute and efficient thinking
Abstract (arXiv)
Despite the strong reasoning capabilities of recent large language models (LLMs), achieving reliable performance on challenging tasks often requires post-training or computationally expensive sampling strategies, limiting their practical efficiency. In this work, we first show that a small subset of neurons in LLMs exhibits strong predictive correlations with reasoning correctness. Based on this observation, we propose AdaRAS (Adaptive Reasoning Activation Steering), a lightweight test-time framework that improves reasoning reliability by selectively intervening on neuron activations. AdaRAS identifies Reasoning-Critical Neurons (RCNs) via a polarity-aware mean-difference criterion and adaptively steers their activations during inference, enhancing incorrect reasoning traces while avoiding degradation on already-correct cases. Experiments on 10 mathematics and coding benchmarks demonstrate consistent improvements, including over 13% gains on AIME-24 and AIME-25. Moreover, AdaRAS exhibits strong transferability across datasets and scalability to stronger models, outperforming post-training methods without additional training or sampling cost.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Placed in Test-time compute and efficient thinking, one of 17 directions in LLM reasoning. Builds on 47 of its 49 references that are in the map. First posted on arXiv 27 Jan 2026.
Builds on
47 of 49 references.
Closest ancestors · 5
Same direction, newest first
- 2025Demystifying Reasoning Dynamics with Mutual Information: Thinking Tokens are Information Peaks in LLM Reasoning3 Jun
- 2025START: Self-taught Reasoner with Tools6 Mar
- 2024TreeBoN: Enhancing Inference-Time Alignment with Speculative Tree-Search and Best-of-N Sampling18 Oct
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameters6 Aug
- 2024Toward Self-Improvement of LLMs via Imagination, Searching, and Criticizing18 Apr
Other references · 42
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023Let's Verify Step by Stepcited by 3,598
- 2021Program Synthesis with Large Language Modelscited by 3,509
- 2022Locating and Editing Factual Associations in GPTcited by 2,609
- 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 1,906
- 2020GLU Variants Improve Transformercited by 1,724
- 2023Is Your Code Generated by ChatGPT Really Correct? Rigorous Evaluation of Large Language Models for Code Generationcited by 1,332
- 2020Transformer Feed-Forward Layers Are Key-Value Memoriescited by 1,307
30 more
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,259
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,239
- 2023Steering Llama 2 via Contrastive Activation Additioncited by 1,009
- 2024Refusal in Language Models Is Mediated by a Single Directioncited by 960
- 2024Large Language Monkeys: Scaling Inference Compute with Repeated Samplingcited by 946
- 2021Probing Classifiers: Promises, Shortcomings, and Advancescited by 831
- 2022Discovering Latent Knowledge in Language Models Without Supervisioncited by 733
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 673
- 2023Eliciting Latent Predictions from Transformers with the Tuned Lenscited by 579
- 2025Beyond the 80/20 Rule: High-Entropy Minority Tokens Drive Effective Reinforcement Learning for LLM Reasoningcited by 540
- 2022Emergent World Representations: Exploring a Sequence Model Trained on a Synthetic Taskcited by 417
- 2022Self-critiquing models for assisting human evaluatorscited by 372
- 2023Finding Neurons in a Haystack: Case Studies with Sparse Probingcited by 328
- 2025rStar-Math: Small LLMs Can Master Math Reasoning with Self-Evolved Deep Thinkingcited by 308
- 2025Persona Vectors: Monitoring and Controlling Character Traits in Language Modelscited by 303
- 2023In-context Vectors: Making In Context Learning More Effective and Controllable Through Latent Space Steeringcited by 240
- 2024Programming Refusal with Conditional Activation Steeringcited by 191
- 2024Scene-LLM: Extending Language Model for 3D Visual Understanding and Reasoningcited by 170
- 2023Linear Representations of Sentiment in Large Language Modelscited by 157
- 2024Improving Instruction-Following in Language Models through Activation Steeringcited by 149
- 2023Inspecting and Editing Knowledge Representations in Language Modelscited by 140
- 2025OpenCodeReasoning: Advancing Data Distillation for Competitive Codingcited by 106
- 2024Hopping Too Late: Exploring the Limitations of Large Language Models on Multi-Hop Queriescited by 78
- 2025Learning to Plan & Reason for Evaluation with Thinking-LLM-as-a-Judgecited by 75
- 2024Can Language Models Solve Olympiad Programming?cited by 70
- 2024How Large Language Models Encode Context Knowledge? A Layer-Wise Probing Studycited by 64
- 2024Confidence Regulation Neurons in Language Modelscited by 57
- 2024Latent Space Chain-of-Embedding Enables Output-free LLM Self-Evaluationcited by 56
- 2025I Have Covered All the Bases Here: Interpreting Reasoning Features in Large Language Models via Sparse Autoencoderscited by 43
- 2024Expanded Gating Ranges Improve Activation Functionscited by 4
Built on by
Citing papers per year: 2026: 4.
4 papers.
- 2026LLM Parameters for Math Across Languages: Shared or Separate?cited by 1
- 2026TALAN: Task-Aligned Latent Adaptation Networks for Targeted Post-Training of Large Language Modelscited by 0
- 2026Recognition-Refusal Misalignment in LLMs: Why Models Answer Structurally Unanswerable Questionscited by 0
- 2026Metacognitive Steering: Learning the Structure of Scientific Judgmentcited by 0
- 2026Metacognitive Steering: Learning the Structure of Scientific Judgment14 Sep 2026
- 2026Recognition-Refusal Misalignment in LLMs: Why Models Answer Structurally Unanswerable Questions29 Aug 2026
- 2026LLM Parameters for Math Across Languages: Shared or Separate?16 Jun 2026
- 2026TALAN: Task-Aligned Latent Adaptation Networks for Targeted Post-Training of Large Language Models5 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026ATLAS: Verifier-Guided Adaptive Latent Activation Steering for Efficient LLM Reasoning
- 2025Rank-1 LoRAs Encode Interpretable Reasoning Signals
- 2026Reforming the Mechanism: Editing Reasoning Patterns in LLMs with Circuit Reshaping
- 2026RISER: Orchestrating Latent Reasoning Skills for Adaptive Activation Steering
- 2024Language Models are Hidden Reasoners: Unlocking Latent Reasoning Capabilities via Self-Rewarding
- 2026Batch-wise Adaptive Pruning: Periodic Neuron Activation-Aware Weight Pruning for Language Reasoning Model
- 2026NeuroProlog: Multi-Task Fine-Tuning for Neurosymbolic Mathematical Reasoning via the Cocktail Effect
- 2026COMPASS: Finding Where Reasoning Lives in Language Models