Shapley Values Improve Reward Attribution for LLM Parallel Reasoning

Wentao Zhang, Haoyu Zhang, Xinke Jiang, Yuxuan Cheng, Yuhan Pan, Miao Li, Zhipeng Qiao, Tao Feng, Zhen Tao, Dengji Zhao· July 22, 2026 View original

Summary

This paper introduces Parallel Shapley, a reinforcement learning framework that uses Shapley values to attribute fine-grained, path-level contributions in multi-path reasoning for Large Language Models (LLMs). This method helps distinguish individual path contributions, leading to more stable training and improved performance on mathematical reasoning benchmarks.

Researchers have developed Parallel Shapley, a novel reinforcement learning (RL) framework designed to enhance multi-step reasoning in Large Language Models (LLMs). Current parallel reasoning approaches often struggle to accurately assess the individual contributions of different reasoning paths, leading to ambiguous learning signals when uniform rewards are assigned to the overall outcome. Parallel Shapley addresses this by treating each reasoning path as a player in a cooperative game. The framework leverages Shapley values to quantify the marginal contribution of each path, using a generative reward model to evaluate path utilities and Monte Carlo sampling for efficient approximation. Experiments on mathematical reasoning benchmarks demonstrate that Parallel Shapley not only outperforms existing baselines but also provides more stable and interpretable training. By effectively "fishing out free riders" and assigning rewards proportionally, this method significantly improves the quality of multi-path reasoning in LLMs.

Why it matters

AI engineers and researchers can use this framework to develop more efficient, interpretable, and robust LLMs for complex reasoning tasks, particularly in domains requiring multi-step problem-solving like mathematics or code generation.

How to implement this in your domain

  1. 1Investigate the Parallel Shapley framework for attributing rewards in multi-path reasoning systems.
  2. 2Integrate Shapley value calculations into your reinforcement learning pipelines for LLMs engaged in complex tasks.
  3. 3Experiment with generative reward models to evaluate the utility of individual reasoning paths.
  4. 4Apply this method to improve the training stability and performance of LLMs on internal reasoning benchmarks.

Who benefits

AI EngineeringSoftware DevelopmentEducation TechnologyResearch & Development

Key takeaways

  • Parallel Shapley uses Shapley values to attribute fine-grained rewards to individual reasoning paths in LLMs.
  • This method resolves ambiguous learning signals in multi-path reasoning.
  • It leads to more stable and interpretable training for LLMs.
  • The framework significantly improves performance on mathematical reasoning benchmarks.

Original post by Wentao Zhang, Haoyu Zhang, Xinke Jiang, Yuxuan Cheng, Yuhan Pan, Miao Li, Zhipeng Qiao, Tao Feng, Zhen Tao, Dengji Zhao

"arXiv:2607.18979v1 Announce Type: new Abstract: Large Language Models (LLMs) excel at multi-step reasoning, yet current parallel reasoning approaches often fail to distinguish the contributions of individual reasoning paths. Many paths may be redundant, misleading, or even detrim…"

View on X

Originally posted by Wentao Zhang, Haoyu Zhang, Xinke Jiang, Yuxuan Cheng, Yuhan Pan, Miao Li, Zhipeng Qiao, Tao Feng, Zhen Tao, Dengji Zhao on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses