Abstract: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.
| Subjects: | Machine Learning (cs.LG) |
| Cite as: | arXiv:2602.06475 [cs.LG] |
| (or arXiv:2602.06475v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2602.06475 arXiv-issued DOI via DataCite |
Submission history
From: Jingyao Wang [view email]
[v1]
Fri, 6 Feb 2026 08:03:11 UTC (3,956 KB)
[v2]
Wed, 13 May 2026 09:12:10 UTC (3,959 KB)