atlas

Research map / Large language models / Reasoning and chain of thought

Two Experts Are All You Need for Steering Thinking: Reinforcing Cognitive Effort in MoE Reasoning Models Without Additional Training

NeurIPS 2025 · Large language models › Reasoning and chain of thought · approach: Efficient and adaptive thinking · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Identifies MoE "cognitive experts" in DeepSeek-R1 and Qwen3-235B via nPMI between expert activations and thinking tokens, then amplifies two of them at inference time (no training) to improve AIME accuracy and token efficiency.

Problem

Large reasoning models suffer from overthinking and underthinking; it is unclear whether specific MoE experts govern meta-level reasoning behavior and whether steering them can improve reasoning without additional training.

Approach

Uses normalized pointwise mutual information between expert activations and thinking markers (<think>, </think>, "Alternatively") to find cognitive experts from a single forward pass, then multiplies their routing weights by a reinforcement factor at inference time.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
reinforce top-2 math-identified cognitive experts (multiplier 64)DeepSeek-R1 · AIME24 / AIME25accuracy: +10.0 on AIME24 (73.3 to 83.3); +10.0 on AIME25 (63.3 to 73.3)Helped
reinforce top-2 cognitive experts while reducing thinking-token countDeepSeek-R1 · AIME24thoughts and tokens: Thoughts 12.0 to 10.2; tokens 9,219 to 8,317Helped
reinforce two cognitive experts identified on AIME24 (layers 70, 23)Qwen3-235B · AIME25accuracy: +6.6 (66.7 to 73.3)Helped
reinforce two cognitive experts identified on AIME24Qwen3-235B · AIME24accuracy: 0 (86.7 to 86.7)No clear effect
reinforce two cognitive experts identified on AIME24 (Qwen3)Qwen3-235B · pass@8 on AIME25 / AIME24 (sampled)pass@1 / pass@8: AIME24 pass@1 +1.0; AIME25 pass@1 -0.6; AIME25 pass@8 +1.4Mixed
reinforce top-2 cognitive experts with sampling (pass@k, temperature 0.6)DeepSeek-R1 · AIME24 / AIME25 pass@1pass@1: AIME24 +1.2 (74.8 to 76.0); AIME25 -0.8 (68.5 to 67.7)Mixed
reinforce top-2 cognitive experts (vs. prompt before think, prompt after think, Tip_t decoding constraint)DeepSeek-R1 · AIME24+AIME25 average accuracyaccuracy: avg 78.7 vs 75.0 (prompt before), 71.7 (prompt after), 76.7 (Tip_t), 68.3 (baseline)Helped
prompt placed before <think> tokenDeepSeek-R1 · AIME24+AIME25 average accuracyaccuracy: +6.7 average (68.3 to 75.0)Helped
prompt placed after <think> tokenDeepSeek-R1 · AIME24+AIME25 average accuracyaccuracy: +3.4 average (68.3 to 71.7)Helped
reinforce top-2 math experts with multiplier 512 (overly large multiplier) (ablation)DeepSeek-R1 · AIME24accuracy: 63.3 at multiplier 512 (vs 73.3 baseline); near-zero for lower-ranked experts at multiplier 256+Hurt
reinforce two random experts across multipliers 2 to 512 (ablation)DeepSeek-R1 · AIME24accuracy: minimal variation (about 63.3 to 76.7)No clear effect
reinforce math-domain experts (general, all-domain identification) (ablation)DeepSeek-R1 · ArenaHard (50 queries)accuracy: 91.0 to 92.0Helped

Surprising findings

Limitations

Experimental setup

Datasets: AIME2024 GPQA Diamond

Benchmarks: AIME2024 AIME2025 GPQA Diamond ArenaHard

Metrics: accuracy token count underthinking score (Thoughts) pass@1 pass@8

Baselines: Prompt before <think> Prompt after <think> Tip_t vanilla DeepSeek-R1 vanilla Qwen3-235B

Builds on: normalized pointwise mutual information Mixture-of-Experts Tip (decoding constraints for efficient reasoning)

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy