Scaf-GRPO: Scaffolded Group Relative Policy Optimization for Enhancing LLM Reasoning
Xichen Zhang, Sitong Wu, Yinghao Zhu, Haoru Tan, Shaozuo Yu, Ziyi He, Jiaya Jia
TL;DR
This work tackles the learning cliff in reinforcement learning from verifier rewards (RLVR) for LLM reasoning by introducing Scaf-GRPO, a scaffolded, on-policy training framework. It diagnoses stagnation through a guidance exemption phase and then employs a hierarchical hint hierarchy (knowledge, planning, solution) to inject minimal, progressively concrete prompts that enable the model to construct valid solutions while maintaining the GRPO objective. Empirical results across diverse math benchmarks and models show substantial gains over vanilla GRPO and prefix-based guidance (e.g., up to a 44.3% relative improvement on AIME24 for Qwen2.5-Math-7B) and demonstrated generalization to non-Qwen architectures and LongCoT settings, as well as strong OOD performance on GPQA-Diamond. The approach underscores the value of structured, in-prompt scaffolding for unlocking autonomous reasoning and provides a robust, model-agnostic pathway toward extending LLM capabilities while preserving exploration. Future work includes automating hint generation and adapting scaffolding to individual model proficiency levels, broadening applicability beyond strictly verifiable domains.
Abstract
Reinforcement learning from verifiable rewards has emerged as a powerful technique for enhancing the complex reasoning abilities of Large Language Models (LLMs). However, these methods are fundamentally constrained by the ''learning cliff'' phenomenon: when faced with problems far beyond their current capabilities, models consistently fail, yielding a persistent zero-reward signal. In policy optimization algorithms like GRPO, this collapses the advantage calculation to zero, rendering these difficult problems invisible to the learning gradient and stalling progress. To overcome this, we introduce Scaf-GRPO (Scaffolded Group Relative Policy Optimization), a progressive training framework that strategically provides minimal guidance only when a model's independent learning has plateaued. The framework first diagnoses learning stagnation and then intervenes by injecting tiered in-prompt hints, ranging from abstract concepts to concrete steps, enabling the model to construct a valid solution by itself. Extensive experiments on challenging mathematics benchmarks demonstrate Scaf-GRPO's effectiveness, boosting the pass@1 score of the Qwen2.5-Math-7B model on the AIME24 benchmark by a relative 44.3% over a vanilla GRPO baseline. This result demonstrates our framework provides a robust and effective methodology for unlocking a model's ability to solve problems previously beyond its reach, a critical step towards extending the frontier of autonomous reasoning in LLM.
