CRoS: rewarding causality on scrambled CoTs boosts faithfulness 13.8% on Gemma-3-4B

mohitban47 · x · 2026-11-03

The paper proposes CRoS (Causality Reward on Scrambled CoTs), addressing the problem that rewarding correct answers doesn't guarantee the model relies on its chain of thought, and LLM reasoning is often post-hoc rationalization.

Key idea: if reasoning causally contributes to the answer, scrambling the order of its steps should change the answer's probability; this serves as a causality reward.

Results: trained with GRPO on Gemma-3-4B, CRoS improves CoT faithfulness (+13.8%), out-of-domain accuracy (+3.0%), and CoT quality (+2.0%) over the strongest outcome and causality reward baselines, mitigating the accuracy-vs-faithfulness trade-off of outcome rewards.

Related event: CRoS Rewards Causal Chain-of-Thought, Boosting Faithfulness by 13.8%(2 posts)→

Original post →

More from Research

Research channel →