Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

The Easy, the Hard, and the Learnable: Confidence and Difficulty-Adaptive Policy Optimization for LLM Reasoning

About

RL with verifiable rewards can substantially improve LLM reasoning, yet standard GRPO-style training often treats easy, hard, and learnable questions alike through uniform sampling and weighting, leading to inefficient compute allocation. We study GRPO by tracking token log-probabilities, group-normalized advantages, and the induced token-level update weights. This reveals three recurring dynamics as training proceeds: (1) confidence inflation, (2) advantage contraction, and (3) hierarchical convergence. These findings suggest that the utility of each update depends strongly on both question difficulty and the model's current competence. Motivated by this, we propose Confidence and Difficulty-adaptive Policy Optimization (CoDaPO), which assigns each question a bounded value from rollout confidence and empirical difficulty. CoDaPO then uses this value to reweight policy updates and resample high-value learnable questions within mini-batches, thereby increasing discovery within the learnable band under a fixed compute budget. Across twelve benchmarks, CoDaPO consistently improves accuracy over existing RL methods. Our code is publicly available at https://github.com/tmlr-group/CoDaPO.

Zhanke Zhou, Xiangyu Lu, Chentao Cao, Brando Miranda, Tongliang Liu, Bo Han, Sanmi Koyejo• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH 500
Accuracy74.39
589
Mathematical ReasoningAIME 2024
Accuracy30.49
394
Mathematical ReasoningAIME 2025
Accuracy17.2
378
Language UnderstandingMMLU
MMLU Accuracy44.81
307
Code GenerationHumanEval
Accuracy50.61
212
Mathematical ReasoningOlympiadBench
Accuracy41.63
57
Science ReasoningGPQA
Accuracy24.45
18
Code GenerationHumanEval
HumanEval Accuracy (greedy pass@1)54.88
14
Code GenerationLiveCodeBench
Pass Rate54
3
Showing 9 of 9 rows

Other info

Follow for update