Conformal Prediction-Enabled Sequence-level Policy Optimization for Reasoning Language Models
Organizations: Peking University · Beihang University · Nanjing University
Abstract
Reinforcement learning improves the reasoning capability of large language models through verifiable rewards for mathematical and coding tasks. In token-level methods such as GRPO, clipping acts on individual token ratios while the reward and advantage apply to a complete response. Recent sequence-level methods align these units but still use fixed clipping bands, whose tail probabilities change as the score distribution evolves. We propose Conformal Prediction-Enabled Sequence-level Policy Optimization (CSPO), which adaptively calibrates sequence-level clipping bounds using recent training statistics. The resulting band targets a specified marginal tail budget and adapts to policy drift; calibration conditional on response length requires additional assumptions. We bound the probability that a fresh score falls outside the band by the target miscoverage level plus explicit CDF discrepancy and drift terms. Under additional moment conditions, this bound also controls the objective and gradient error relative to the unclipped GSPO power surrogate. Experiments with the Qwen3-1.7B series show improvements over GSPO on difficult mathematical reasoning benchmarks and ALFWorld, with clipping endpoints that adapt during training.
Figures & tables
| Method | ALFWorld | ||||||
|---|---|---|---|---|---|---|---|
| Pick | Look | Clean | Heat | Cool | Pick2 | All | |
| Qwen3-1.7B | 0.00 | 40.00 | 3.23 | 0.00 | 0.00 | 0.00 | 3.91 |
| GSPO | 58.06 | 66.67 | 45.00 | 61.54 | 22.22 | 20.83 | 31.25 |
| GRPO | 67.86 | 66.67 | 56.52 | 61.54 | 55.17 | 45.45 | 52.34 |
| DAPO | 76.92 | 70.00 | 72.00 | 75.00 | 56.52 | 33.33 | 53.12 |
| CSPO | 93.33 | 70.00 | 68.00 | 75.00 | 60.00 | 44.00 | 64.06 |
| Method | AIME24 | AIME25 | AMC | MATH500 | Minerva | Oly. | Average |
|---|---|---|---|---|---|---|---|
| Qwen3-1.7B-Base | 5.42 | 3.54 | 25.38 | 54.43 | 13.72 | 22.47 | 20.83 |
| GRPO | 11.40 | 5.65 | 39.43 | 70.49 | 21.90 | 33.27 | 30.36 |
| DAPO | 11.05 | 5.72 | 41.56 | 73.04 | 22.82 | 34.32 | 31.42 |
| GSPO | 15.08 | 8.33 | 42.86 | 73.36 | 22.68 | 37.83 | 33.36 |
| CSPO | 16.87 | 8.57 | 43.36 | 74.22 | 23.19 | 38.44 | 34.11 |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Update element or score | Band/control construction | Primary controlled object |
|---|---|---|---|
| PPO / GRPO | Token ratio | Fixed numerical endpoints | Pessimistic token surrogate |
| GSPO | Response | Fixed response-level endpoints | Sequence-coherent power surrogate |
| FSPO | Sequence log-ratio | Fixed -scaled geometry | Length reweighting and gradient direction |
| CTPO ( Zhang et al., 2026 ) | Prefix/cumulative ratio | Position-dependent scaling | Prefix correction and position scaling |
| BandPO ( Li et al., 2026 ) | Probability ratio | Divergence-derived probability-aware bounds | Projected trust-region geometry |
| BAPO ( Xi et al., 2025 ) | Ratio contributions | Advantage/entropy-driven asymmetry | Contribution balance and entropy |
| Method | AIME24 | AIME25 | AMC | MATH500 | Minerva | Oly. | Average |
|---|---|---|---|---|---|---|---|
| CSPO- -0.1 | 16.87 | 8.57 | 43.36 | 74.22 | 23.19 | 38.44 | 34.11 |
| CSPO- -0.2 | 14.85 | 8.58 | 42.86 | 74.49 | 23.02 | 37.23 | 33.50 |
| CSPO- -0.4 | 13.52 | 7.40 | 41.50 | 72.98 | 22.47 | 36.34 | 32.37 |
RL for Language Models
Reinforcement learning for language models uses feedback-driven policy updates to improve reasoning, alignment, and interactive behavior.
See all 410 papers