{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/arxiv-2601-22648","title":"UCPO: Uncertainty-Aware Policy Optimization","arxiv_id":"2601.22648","date":"2026-01-30","proceeding":"ICML","authors":["Xianzhou Zeng","Jing Huang","Chunmei Xie","Gongrui Nan","Siye Chen","Mengyu Lu","Weiqi Xiong","Qixuan Zhou","Junhao Zhang","Qiang Zhu","Yadong Li","Xingzhong Xu"],"abstract":"The key to building trustworthy large language models (LLMs) lies in endowing them with inherent uncertainty expression capabilities, thereby mitigating overconfident errors in high-stakes applications. However, existing RL paradigms such as GRPO often suffer from Advantage Bias due to binary decision spaces and static uncertainty rewards, inducing either excessive conservatism or overconfidence. To tackle this challenge, this paper unveils the root causes of reward hacking and overconfidence in current RL paradigms incorporating uncertainty-based rewards, based on which we propose the UnCertainty-Aware Policy Optimization (UCPO) framework. UCPO employs Ternary Advantage Decoupling to separate and independently normalize deterministic and uncertain rollouts, thereby eliminating advantage bias. Furthermore, a Dynamic Uncertainty Reward Adjustment mechanism adapts uncertainty weights in real-time according to model evolution and instance difficulty. Experimental results in mathematical reasoning and general tasks demonstrate that UCPO effectively resolves the reward imbalance, significantly improving the reliability of the model beyond their knowledge boundaries.","url_abs":"https://arxiv.org/abs/2601.22648","url_pdf":"https://arxiv.org/pdf/2601.22648","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2601.22648","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2601.22648"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/xzhouzeng/ucpo","reach":null}],"summary":{"ran":1,"unverified":9},"by_repo_kind":{"found_in_text":{"samples":10,"ran":1,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"8896f4e2bafe8abc","entry":"generate_config_from_args","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"train/verl/model_merger/base_model_merger.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/train/verl/model_merger/base_model_merger.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"8896f4e2bafe8abc"}},{"code_sha256_prefix":"36bc11ec34c0b777","entry":"cal_metrics_whole_mc","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"eval/metrics.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/eval/metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"36bc11ec34c0b777"}},{"code_sha256_prefix":"57cabc812db1e524","entry":"compute_score","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"train/recipe/ucpo/reward_fn/uc_reward_mc.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/train/recipe/ucpo/reward_fn/uc_reward_mc.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"57cabc812db1e524"}},{"code_sha256_prefix":"bd2a0c243efb8933","entry":"extract_answer_mc","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"eval/metrics.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/eval/metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"bd2a0c243efb8933"}},{"code_sha256_prefix":"a34eb4a9ee65c5fb","entry":"load_json","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"eval/eval_general.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/eval/eval_general.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"a34eb4a9ee65c5fb"}},{"code_sha256_prefix":"0388e4fa2f36659e","entry":"mathd_normalize_answer","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"eval/utils.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/eval/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"0388e4fa2f36659e"}},{"code_sha256_prefix":"5662e5ee755b541c","entry":"normalize_final_answer","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"eval/utils.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/eval/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"5662e5ee755b541c"}},{"code_sha256_prefix":"3dcb40e5ce9bac48","entry":"repeatness","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"eval/utils.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/eval/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"3dcb40e5ce9bac48"}},{"code_sha256_prefix":"bf0242950c3037b8","entry":"think_boxed_reward_fn_mc","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"eval/metrics.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/eval/metrics.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"bf0242950c3037b8"}},{"code_sha256_prefix":"9fc79b9da77368ec","entry":"think_boxed_reward_fn_mc","repo":"xzhouzeng/ucpo","repo_kind":"found_in_text","path":"train/recipe/ucpo/reward_fn/uc_reward_mc.py","file_url":"https://github.com/xzhouzeng/ucpo/blob/HEAD/train/recipe/ucpo/reward_fn/uc_reward_mc.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"9fc79b9da77368ec"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}