π€ AI Summary
Existing chain-of-thought (CoT) code generation methods suffer from three key limitations: (1) reliance on a single, rigid reasoning path; (2) absence of real-time quality assessment for intermediate reasoning steps; and (3) susceptibility to redundant, overly complex βover-reasoning.β To address these, we formulate CoT as a sequential decision-making problem and propose SEERβa unified framework comprising: (1) a self-exploration mechanism enabling label-free, diverse reasoning path search; (2) a lightweight value model for online quality evaluation of intermediate steps; and (3) an adaptive controller that dynamically switches between direct generation and stepwise reasoning modes. SEER jointly optimizes the policy and value models via reinforcement learning. Empirical evaluation across multiple code generation benchmarks demonstrates substantial improvements in correctness and generalization, while effectively mitigating over-reasoning and enhancing both reasoning path diversity and output reliability.
π Abstract
Code generation, the task of creating executable programs from natural language requirements, has recently seen tremendous advances through Chain-of-Thought (CoT) reasoning, which enables Large Language Models (LLMs) to develop high-level reasoning plans before writing code. Recent research has proposed various methods to enhance models' CoT reasoning for code generation such as prompt engineering and supervised fine-tuning. However, existing approaches still face three critical limitations: (1) limited exploration of diverse reasoning paths, which constrains generalization across various programming scenarios, (2) lack of quality assessment for intermediate reasoning steps, which hampers the reliability of the generated plans and code, and (3) the potential negative impact of "overthinking", potentially leading to unnecessarily complex and incorrect solutions. To address these limitations, we frame CoT code generation as a decision making problem and present SEER, a SElf-Exploring deep Reasoning framework that enables accurate and adaptive reasoning for code generation. SEER introduces three key components: (1) Diverse reasoning path exploration, which aims at exploring diverse reasoning paths and annotating intermediate steps without relying on manual experts or closed-source proprietary models; (2) Reasoning quality-aware model training, which trains a policy model for generating candidate reasoning steps and a value model for assessing their quality; and (3) Adaptive CoT reasoning, which dynamically switches between direct generation and step-by-step reasoning for different problems.