Saved in:
Bibliographic Details
Main Authors: Tan, Zelin, Yu, Zhouliang, Lin, Bohan, Geng, Zijie, Geng, Hejia, Zhang, Yudong, Zhang, Mulei, Chen, Yang, Hu, Shuyue, Yin, Zhenfei, Zhang, Chen, Bai, Lei
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2603.26535
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908934233653248
author Tan, Zelin
Yu, Zhouliang
Lin, Bohan
Geng, Zijie
Geng, Hejia
Zhang, Yudong
Zhang, Mulei
Chen, Yang
Hu, Shuyue
Yin, Zhenfei
Zhang, Chen
Bai, Lei
author_facet Tan, Zelin
Yu, Zhouliang
Lin, Bohan
Geng, Zijie
Geng, Hejia
Zhang, Yudong
Zhang, Mulei
Chen, Yang
Hu, Shuyue
Yin, Zhenfei
Zhang, Chen
Bai, Lei
contents We propose Process-Aware Policy Optimization (PAPO), a method that integrates process-level evaluation into Group Relative Policy Optimization (GRPO) through decoupled advantage normalization, to address two limitations of existing reward designs. Outcome reward models (ORM) evaluate only final-answer correctness, treating all correct responses identically regardless of reasoning quality, and gradually lose the advantage signal as groups become uniformly correct. Process reward models (PRM) offer richer supervision, but directly using PRM scores causes reward hacking, where models exploit verbosity to inflate scores while accuracy collapses. PAPO resolves both by composing the advantage from an outcome component Aout, derived from ORM and normalized over all responses, and a process component Aproc, derived from a rubric-based PRM and normalized exclusively among correct responses. This decoupled design ensures that Aout anchors training on correctness while Aproc differentiates reasoning quality without distorting the outcome signal. Experiments across multiple model scales and six benchmarks demonstrate that PAPO consistently outperforms ORM, reaching 51.3% vs.\ 46.3% on OlympiadBench while continuing to improve as ORM plateaus and declines.
format Preprint
id arxiv_https___arxiv_org_abs_2603_26535
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle PAPO: Stabilizing Rubric Integration Training via Decoupled Advantage Normalization
Tan, Zelin
Yu, Zhouliang
Lin, Bohan
Geng, Zijie
Geng, Hejia
Zhang, Yudong
Zhang, Mulei
Chen, Yang
Hu, Shuyue
Yin, Zhenfei
Zhang, Chen
Bai, Lei
Artificial Intelligence
We propose Process-Aware Policy Optimization (PAPO), a method that integrates process-level evaluation into Group Relative Policy Optimization (GRPO) through decoupled advantage normalization, to address two limitations of existing reward designs. Outcome reward models (ORM) evaluate only final-answer correctness, treating all correct responses identically regardless of reasoning quality, and gradually lose the advantage signal as groups become uniformly correct. Process reward models (PRM) offer richer supervision, but directly using PRM scores causes reward hacking, where models exploit verbosity to inflate scores while accuracy collapses. PAPO resolves both by composing the advantage from an outcome component Aout, derived from ORM and normalized over all responses, and a process component Aproc, derived from a rubric-based PRM and normalized exclusively among correct responses. This decoupled design ensures that Aout anchors training on correctness while Aproc differentiates reasoning quality without distorting the outcome signal. Experiments across multiple model scales and six benchmarks demonstrate that PAPO consistently outperforms ORM, reaching 51.3% vs.\ 46.3% on OlympiadBench while continuing to improve as ORM plateaus and declines.
title PAPO: Stabilizing Rubric Integration Training via Decoupled Advantage Normalization
topic Artificial Intelligence
url https://arxiv.org/abs/2603.26535