Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This paper introduces Function-Structured Graph Reinforcement Learning (FSG-RL) to address the challenges of reliable decomposition, computation, and aggregation in algorithmic mathematical reasoning by integrating subproblem graphs with executable Python code and multi-verifier feedback. The approach first employs supervised fine-tuning to train a policy for code generation from function graphs, followed by Group Relative Policy Optimization (GRPO) that utilizes answer-gated rewards and span-level credit assignment for policy optimization. Evaluated on a benchmark of diverse mathematical datasets, FSG-RL significantly improves final-answer accuracy and full solution success rates over SFT, demonstrating the efficacy of verifier-guided reinforcement learning for complex mathematical tasks.
Verifier-guided RL unlocks a 34% accuracy boost in mathematical reasoning by directly penalizing incorrect intermediate steps, not just final answers.
Algorithmic mathematical reasoning requires reliable decomposition, computation, and aggregation. Final-answer rewards provide limited guidance on intermediate errors, while successful execution does not guarantee mathematical correctness. This work proposes Function-Structured Graph Reinforcement Learning (FSG-RL), connecting subproblem graphs and Python implementations with multi-verifier feedback. The policy first learns to generate code from function graphs through supervised fine-tuning (SFT). Group Relative Policy Optimization (GRPO) then optimizes the policy using answer-gated rewards and span-level credit assignment. The framework also supports teacher supervision and structured memory. A benchmark curated from Grade School Math 8K (GSM8K), MathQA, MATH, and Omni-MATH pairs public function graphs with private verification specifications. Under a unified evaluation protocol, GRPO improves final-answer accuracy from 43.25% to 67.50% and full solution success from 32.25% to 52.25% over SFT. Continued reinforcement learning (RL) with teacher supervision yields additional gains. The gains extend beyond producing correctly formatted code, supporting verifier-guided reinforcement learning for mathematical reasoning. Code is available at https://github.com/ZihanLiummyycc/FSG-RL.