English

Counterfactual Shapley Credit Assignment

Machine Learning 2026-07-18 v1 Artificial Intelligence

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 (ϕ\phi-value). By redistributing environmental rewards, ϕ\phi-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 ϕ\phi-values efficiently, enabling a new class of policy gradient methods, ϕ\phi-PPO, combined with Prioritized Trajectory Replay (PTR). Empirical results demonstrate that ϕ\phi-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.

Cite

@article{arxiv.2607.16999,
  title  = {Counterfactual Shapley Credit Assignment},
  author = {Mingxuan Li and Kaizhan-Lee and Elias Bareinboim},
  journal= {arXiv preprint arXiv:2607.16999},
  year   = {2026}
}

Comments

Reinforcement Learning Journal 2026 (Also RLC 2026)