Guardado en:
Detalles Bibliográficos
Autores principales: Ai, Zhengyang, Shan, Zikang, Ai, Xiaodong, Tang, Jingxian, Hu, Hangkai, Lu, Pinyan
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:https://arxiv.org/abs/2604.06636
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910110445469696
author Ai, Zhengyang
Shan, Zikang
Ai, Xiaodong
Tang, Jingxian
Hu, Hangkai
Lu, Pinyan
author_facet Ai, Zhengyang
Shan, Zikang
Ai, Xiaodong
Tang, Jingxian
Hu, Hangkai
Lu, Pinyan
contents Process supervision has emerged as a promising approach for enhancing LLM reasoning, yet existing methods fail to distinguish meaningful progress from mere verbosity, leading to limited reasoning capabilities and unresolved token inefficiency. To address this, we propose Stage-aware Hierarchical Advantage via Potential Estimation (SHAPE), a framework that formalizes reasoning as a trajectory through a state space of empirical solvability. SHAPE introduces a hierarchical credit assignment mechanism: at the segment level, it employs a stage-aware advantage function to prioritize efficient breakthroughs in low-potential states; at the token level, it utilizes entropy-driven redistribution to sharpen execution signals. Extensive experiments in math reasoning across three base models and five benchmarks demonstrate that SHAPE achieves an average accuracy gain of 3% with 30% reduced token consumption.
format Preprint
id arxiv_https___arxiv_org_abs_2604_06636
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle SHAPE: Stage-aware Hierarchical Advantage via Potential Estimation for LLM Reasoning
Ai, Zhengyang
Shan, Zikang
Ai, Xiaodong
Tang, Jingxian
Hu, Hangkai
Lu, Pinyan
Machine Learning
Artificial Intelligence
Computation and Language
Process supervision has emerged as a promising approach for enhancing LLM reasoning, yet existing methods fail to distinguish meaningful progress from mere verbosity, leading to limited reasoning capabilities and unresolved token inefficiency. To address this, we propose Stage-aware Hierarchical Advantage via Potential Estimation (SHAPE), a framework that formalizes reasoning as a trajectory through a state space of empirical solvability. SHAPE introduces a hierarchical credit assignment mechanism: at the segment level, it employs a stage-aware advantage function to prioritize efficient breakthroughs in low-potential states; at the token level, it utilizes entropy-driven redistribution to sharpen execution signals. Extensive experiments in math reasoning across three base models and five benchmarks demonstrate that SHAPE achieves an average accuracy gain of 3% with 30% reduced token consumption.
title SHAPE: Stage-aware Hierarchical Advantage via Potential Estimation for LLM Reasoning
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2604.06636