Surprising Success, Repeated Failure: Entropy-Guided Credit Assignment for Exploration in LLM Reasoning

Woongyeong Yeo, Minki Kang, Chanuk Lee, Sangwoo Park, Jinheon Baek, Sung Ju Hwang

KAIST · DeepAuto.ai

Published September 27, 2026 · on the feed September 29, 2026 · ▲ 41 · arXiv 2609.33781

Reasoning & mathLLM training & RL Code arXiv ↗Hugging Face ↗

The abstract

Reinforcement learning with verifiable rewards (RLVR) enhances reasoning in large language models (LLMs) through outcome-level feedback, yet recent approaches to finer-grained credit assignment often require auxiliary models, additional sampling, or privileged information. Although policy entropy provides a readily available signal, prioritizing uncertain positions under both reinforcement and penalization concentrates penalties where failed responses still retain alternatives for recovery, which can suppress opportunities for exploration. To address this, we introduce Entropic Advantage Policy Optimization (EAPO), an entropy-guided credit assignment method that treats success and failure asymmetrically. Specifically, motivated by the observation that success under uncertainty is less repeatable while confident failures tend to recur, EAPO couples normalized policy entropy with the sign of the response advantage to reinforce surprising success and correct repeated failure. It assigns stronger reinforcement to high-entropy decisions in successful responses and stronger penalties to low-entropy decisions in failed responses, while attenuating penalties at uncertain positions to preserve opportunities for recovery. By redistributing the response advantage across tokens, EAPO derives token-level credit directly from existing rollout signals without additional supervision. We validate EAPO on a range of reasoning tasks across both base and reasoning backbones, demonstrating that it achieves the best overall performance. We further show that EAPO promotes more effective exploration, broadening problem coverage and generating more diverse candidate answers.

The Brief

2:40

A short podcast summary of the paper. Written and voiced by AI.

Transcript

Surprising Success, Repeated Failure. Entropy-Guided Credit Assignment for Exploration in LLM Reasoning. By Woongyeong Yeo and five others, from KAIST and DeepAuto A I. Published on the twenty-seventh of September. Its code repository is public and real: about twenty source files.

Reinforcement learning with verifiable rewards improves reasoning in large language models using feedback about the outcome alone. A whole response is right or wrong, and every token in it shares that verdict. Recent attempts to assign credit more finely tend to need extra machinery: auxiliary models, additional sampling, or privileged information. Policy entropy, a measure of how uncertain the model was at each position, is available for free. But using it naively backfires. If uncertain positions are prioritized both when rewarding and when penalizing, the penalties pile up exactly where a failed response still had alternatives open, places from which it might have recovered. That can choke off exploration.

Entropic Advantage Policy Optimization, EAPO, treats success and failure differently. It rests on an observation about repeatability. When the model succeeds while uncertain, that success is less likely to recur on its own. When the model fails while confident, that failure tends to come back. So EAPO aims to reinforce the surprising successes and to correct the repeated failures.

It does this by coupling normalized policy entropy with the sign of the response's advantage. In a successful response, the decisions made under high entropy get stronger reinforcement. In a failed response, the decisions made under low entropy, the confident ones, get stronger penalties. And at uncertain positions within a failure, penalties are softened, preserving room to recover. All of this works by redistributing the response's advantage across its tokens, so token-level credit comes straight from signals the rollouts already contain, with no extra supervision.

The authors validate EAPO on a range of reasoning tasks, using both base backbones and reasoning backbones, and report that it achieves the best overall performance. They further show that it explores more effectively: it broadens the set of problems covered and produces more diverse candidate answers. That second result matters for the argument, because protecting exploration was the reason for treating failures asymmetrically in the first place. The abstract gives no individual scores.

A tutor might grade the same way. A lucky guess that happened to work deserves extra encouragement, because the student may not find it again. A confident mistake deserves the firmest correction, because it will otherwise be repeated. And a hesitant wrong step deserves a lighter touch, since that is where the student was still searching. EAPO writes that intuition into the advantage, token by token.

Paper Quest · 7 chapters · about 41 min

Play this paper

Learn it by doing: make the calls the authors made, guess the result before you see it, and defend it in a viva graded on your own words. Chapter 1 is right here.

Chapter 1 🧩 · The problem

A right-or-wrong verdict is a blunt tool. The wall here is giving each token the credit it earned, for free.

You train a reasoning model with reinforcement learning. A checker marks each whole response right or wrong, and every token in it shares that verdict, though some decisions mattered far more than others.

The road ahead

  1. 1 · The problem ~7 min A checker only says whether a whole response was right, and a free per-token uncertainty signal can push penalties onto exactly the wrong places.
  2. 2 · What came before ~3 min
  3. 3 · The key idea ~9 min Reward uncertain tokens most in successes, penalize confident tokens most in failures, by reshaping the response's existing advantage with entropy.
  4. 4 · Choose your own research ~3 min
  5. 5 · Build the method ~8 min Score responses with a checker, turn entropy into a signed weight using the advantage's sign, then spread each response's advantage over its tokens.
  6. 6 · The experiment ~6 min You’ll guess: Beyond accuracy, what did EAPO's entropy-guided credit assignment do to exploration during reinforcement learning for reasoning?
  7. 7 · The viva ~5 min Three questions from the examiners, answered in your own words. Win it for the paper’s trophy.
Join free to play the whole quest ›

Free, with Google or an email address. Your stars and progress are kept.

KipQuest: the day’s AI research, as podcasts and quests. All papers Android app Privacy Contact Episodes are written and voiced by AI from each paper. Summaries can contain mistakes: the paper is the source.