Priority-Preserving Advantage Aggregation for Multi-Reward Policy Optimization
LLM post-training often optimizes multiple objectives with unequal priorities, such as correctness and reasoning efficiency. However, the widely used additive reward aggregation in RL can violate these priorities: when primary and auxiliary advantages conflict, a sufficiently large auxiliary signal can overturn the preference induced by the primary objective. We term this problem \emph{advantage priority inversion}. Although reward-level gating can protect priority more directly, it still removes potentially useful auxiliary distinctions among rollout samples that fail the primary condition. To address this trade-off, we propose Primary Advantage Policy Optimization (\textbf{PAPO}), a priority-aware advantage aggregation method that asymmetrically combines a primary and an auxiliary advantage. PAPO retains auxiliary supervision when the objectives agree and smoothly attenuates conflicting auxiliary signals, preserving the primary sign during fusion for \emph{primary-consistent ordering}. This rule applies to sequence-level advantages in GRPO and token-level advantages in actor--critic PPO. Experiments on multiple domains and models show that PAPO consistently achieves the best primary objective while maintaining best or competitive auxiliary-objective performance across both GRPO- and PPO-based optimization. \href{https://xiangchi-yuan.github.io/papo-project/}{Code is available}.
Authors
- Soroush Vosoughi (ORCID: https://orcid.org/0000-0002-2564-8909)
- Zhenbang Du (ORCID: https://orcid.org/0000-0002-1386-8381)
- Xingjian Diao (ORCID: https://orcid.org/0000-0001-9605-4991)
- Kejing Xia
- Dachuan Shi
- Shenglong Yao
- Chunhui Zhang
- Zheyuan Liu
- Wenke Lee
- Xiangchi Yuan
Publication Details
- Journal
- arXiv (Cornell University)
- Published
- 2026-10-26
- DOI
- https://doi.org/10.48550/arxiv.2610.34862
- Primary Topic
- Reinforcement Learning in Robotics
- Type
- preprint