ACPO Improves LLM Reasoning with Adaptive Credit Assignment

Zijun Xie, Yuyang You, Yongzhi Li, Enlei Gong, Zeyu Chen, Quan Chen, Yanhua Cheng, Peng Jiang, Yadong Mu· July 7, 2026 View original

Summary

This paper introduces Adaptive Credit Policy Optimization (ACPO), a novel token-level credit assignment framework for Reinforcement Learning (RL) in Large Language Models (LLMs). ACPO uses a mode-local surrogate entropy to modulate policy updates, emphasizing uncertain decisions in successful rollouts and overconfident tokens in failed ones, leading to consistent improvements in mathematical reasoning and coding benchmarks.

Reinforcement Learning (RL) has significantly boosted the reasoning capabilities of Large Language Models (LLMs). However, the challenge of assigning credit at the token level, especially with sparse outcome rewards, remains a major hurdle. Traditional scalable RL methods often apply trajectory-level rewards uniformly, while existing entropy-aware approaches either use detached heuristics or directly optimize true entropy, which can lead to misaligned gradients. This research proposes Adaptive Credit Policy Optimization (ACPO), a new framework designed for fine-grained, token-level credit assignment. ACPO leverages a mode-local surrogate entropy to adaptively adjust policy updates. It strategically emphasizes uncertain decisions in successful sequences and, conversely, targets overconfident tokens in failed sequences. The authors demonstrate that this surrogate entropy allows for deterministic bounds and, under specific conditions, preserves the policy-gradient direction. Experiments on demanding benchmarks like AIME 2025 for mathematical reasoning and HumanEvalPro for coding show that ACPO consistently outperforms strong RL baselines such as DAPO, GTPO, and SAPO, indicating a significant step forward in making LLMs more robust and intelligent.

Why it matters

ACPO offers a more effective way to train LLMs for complex reasoning tasks, leading to more robust and accurate models, which is critical for applications requiring high-stakes decision-making or code generation.

How to implement this in your domain

  1. 1Integrate ACPO into your LLM fine-tuning pipelines for tasks requiring complex reasoning or code generation.
  2. 2Experiment with ACPO on internal benchmarks to assess its impact on model performance and robustness.
  3. 3Develop tools to visualize and analyze the token-level credit assignments made by ACPO to gain insights into model learning.
  4. 4Consider applying the concept of fine-grained, adaptive credit assignment to other sequence generation tasks beyond LLMs.

Who benefits

Software DevelopmentEducationAI ResearchCustomer ServiceFinance

Key takeaways

  • Token-level credit assignment is crucial but challenging for RL in LLMs.
  • ACPO uses a mode-local surrogate entropy for adaptive policy updates.
  • It emphasizes uncertain decisions in successes and overconfident tokens in failures.
  • ACPO consistently improves LLM performance on reasoning and coding tasks.

Original post by Zijun Xie, Yuyang You, Yongzhi Li, Enlei Gong, Zeyu Chen, Quan Chen, Yanhua Cheng, Peng Jiang, Yadong Mu

"arXiv:2607.03126v1 Announce Type: new Abstract: Reinforcement Learning (RL) has substantially improved the reasoning ability of large language models (LLMs), but sparse outcome rewards still make token-level credit assignment difficult. Existing scalable RL methods typically assi…"

View on X

Originally posted by Zijun Xie, Yuyang You, Yongzhi Li, Enlei Gong, Zeyu Chen, Quan Chen, Yanhua Cheng, Peng Jiang, Yadong Mu on X · view source

Want to go deeper?

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

Explore courses