Towards Generalizable Reasoning: Group Causal Counterfactual Policy Optimization for LLM Reasoning
On the map: LLM reasoning → Causal reasoning in LLMs
Abstract (arXiv)
Large language models (LLMs) excel at complex tasks with advances in reasoning capabilities. However, existing reward mechanisms remain tightly coupled to final correctness and pay little attention to the underlying reasoning process: trajectories with sound reasoning but wrong answers receive low credit, while lucky guesses with flawed logic may be highly rewarded, affecting reasoning generalization. From a causal perspective, we interpret multi-candidate reasoning for a fixed question as a family of counterfactual experiments with theoretical supports. Building on this, we propose Group Causal Counterfactual Policy Optimization to explicitly train LLMs to learn generalizable reasoning patterns. It proposes an episodic causal counterfactual reward that jointly captures (i) robustness, encouraging the answer distribution induced by a reasoning step to remain stable under counterfactual perturbations; and (ii) effectiveness, enforcing sufficient variability so that the learned reasoning strategy can transfer across questions. We then construct token-level advantages from this reward and optimize the policy, encouraging LLMs to favor reasoning patterns that are process-valid and counterfactually robust. Extensive experiments on diverse benchmarks demonstrate its advantages.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Causal reasoning in LLMs, one of 17 directions in LLM reasoning. Builds on 37 of its 54 references that are in the map. First posted on arXiv 6 Feb 2026.
Builds on
37 of 54 references.
Closest ancestors · 2
Same direction, newest first
- 2025Executable Counterfactuals: Improving LLMs' Causal Reasoning Through Code2 Oct
- 2025CounterBench: Evaluating and Improving Counterfactual Reasoning in Large Language Models16 Feb
Other references · 35
- 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
- 2025DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learningcited by 6,146
- 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,203
- 2023SWE-bench: Can Language Models Resolve Real-World GitHub Issues?cited by 3,193
- 2024Scaling LLM Test-Time Compute Optimally can be More Effective than Scaling Model Parameterscited by 1,928
- 2025Understanding R1-Zero-Like Training: A Critical Perspectivecited by 1,276
- 2024OpenAI o1 System Cardcited by 1,220
- 2025Group Sequence Policy Optimizationcited by 788
- 2025The Lessons of Developing Process Reward Models in Mathematical Reasoningcited by 402
- 2025Demystifying Long Chain-of-Thought Reasoning in LLMscited by 323
23 more
- 2025Training Language Models to Reason Efficientlycited by 242
- 2024Free Process Rewards without Process Labelscited by 162
- 2025Thoughts Are All Over the Place: On the Underthinking of o1-Like LLMscited by 162
- 2025Pass@k Training for Adaptively Balancing Exploration and Exploitation of Large Reasoning Modelscited by 134
- 2025Optimizing Test-Time Compute via Meta Reinforcement Fine-Tuningcited by 121
- 2025Process Reward Models That Thinkcited by 118
- 2025SWEET-RL: Training Multi-Turn LLM Agents on Collaborative Reasoning Taskscited by 108
- 2025Geometric-Mean Policy Optimizationcited by 107
- 2025Beyond Reward Hacking: Causal Rewards for Large Language Model Alignmentcited by 42
- 2025VersaPRM: Multi-Domain Process Reward Model via Synthetic Reasoning Datacited by 39
- 2025Think When You Need: Self-Adaptive Chain-of-Thought Learningcited by 37
- 2025GVPO: Group Variance Policy Optimization for Large Language Model Post-Trainingcited by 32
- 2025Beyond Correctness: Harmonizing Process and Outcome Rewards through RL Trainingcited by 29
- 2025BNPO: Beta Normalization Policy Optimizationcited by 28
- 2025On the Emergence of Thinking in LLMs I: Searching for the Right Intuitioncited by 22
- 2025GTPO and GRPO-S: Token and Sequence-Level Reward Shaping with Policy Entropycited by 18
- 2025Learning to Think: Information-Theoretic Reinforcement Fine-Tuning for LLMscited by 16
- 2025Insights from Verification: Training a Verilog Generation LLM with Reinforcement Learning with Testbench Feedbackcited by 14
- 2025Can LLMs Reconcile Knowledge Conflicts in Counterfactual Reasoningcited by 7
- 2024MoreauPruner: Robust Pruning of Large Language Models against Weight Perturbationscited by 5
- 2025Benchmarking LLM for Code Smells Detection: OpenAI GPT-4.0 vs DeepSeek-V3cited by 5
- 2025CoLD: Counterfactually-Guided Length Debiasing for Process Reward Models in Mathematical Reasoningcited by 4
- 2025Group Causal Policy Optimization for Post-Training Large Language Modelscited by 4
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026Better Think Thrice: Learning to Reason Causally with Double Counterfactual Consistency
- 2025On the Eligibility of LLMs for Counterfactual Reasoning: A Decompositional Study
- 2024Reasoning Elicitation in Language Models via Counterfactual Feedback
- 2025Counterfactual Self-Questioning for Stable Policy Optimization in Language Models
- 2025Causal Consistency Regularization: Training Verifiably Sensitive Reasoning in Large Language Models
- 2024Making Reasoning Matter: Measuring and Improving Faithfulness of Chain-of-Thought Reasoning
- 2026Beyond Penalizing Mistakes: Stabilizing Efficiency Training in Large Reasoning Models via Adaptive Correct-Only Rewards
- 2025Right Question is Already Half the Answer: Fully Unsupervised LLM Reasoning Incentivization