cs.LGJul 18, 2026

Counterfactual Shapley Credit Assignment

Authors: Mingxuan LiKai-Zhan LeeElias Bareinboim

Organizations: Department of Computer Science, Columbia University, New York, NY, USA

Abstract

The Credit Assignment Problem (CAP) is fundamental to developing efficient and explainable Reinforcement Learning (RL) agents. Existing frameworks, whether relying on temporal contiguity or hindsight-conditioned reward reweighting, frequently fail to attribute properly between an agent's policy (skill) and environmental stochasticity (luck). A principled approach to CAP must isolate the true causal drivers of observed outcomes from spurious correlations and environmental randomness. We introduce Counterfactual Shapley Credit Assignment, a novel framework grounded in causal theory that attributes credit and blame via the Counterfactual Shapley Value (φφ-value). By redistributing environmental rewards, φφ-values enhance temporal credit assignment across three critical dimensions: sparse causality, high stochasticity, and delayed rewards, all while preserving the optimal policy. We derive a consistent estimator that computes φφ-values efficiently, enabling a new class of policy gradient methods, φφ-PPO, combined with Prioritized Trajectory Replay (PTR). Empirical results demonstrate that φφ-values align precisely to the ground truth causes of task rewards with superior sample efficiency in challenging environments where prior state-of-the-art methods fail to converge.

Explore similar work

CardsList