Search papers, labs, and topics across Lattice.
The paper introduces a scalable method for generating Process Reward Model (PRM) training data by leveraging planning problems expressed in PDDL, addressing limitations of existing PRM datasets which are costly, error-prone, and math-centric. They generate a dataset of ~1M reasoning steps across diverse PDDL domains and train PRMs on it. Results show that augmenting existing PRM datasets with this PDDL-derived data significantly improves LLM reasoning performance in both mathematical and non-mathematical tasks.
Forget expensive, error-prone math problems: PDDL planning offers a surprisingly effective and scalable route to training better Process Reward Models for LLM reasoning.
Process Reward Models (PRMs) have emerged as a powerful tool for providing step-level feedback when evaluating the reasoning of Large Language Models (LLMs), which frequently produce chains of thought (CoTs) containing errors even when the final answer is correct. However, existing PRM datasets remain expensive to construct, prone to annotation errors, and predominantly limited to the mathematical domain. This work introduces a novel and scalable approach to PRM dataset generation based on planning logical problems expressed in the Planning Domain Definition Language (PDDL). Using this method, we generate a corpus of approximately one million reasoning steps across various PDDL domains and use it to train PRMs. Experimental results show that augmenting widely-used PRM training datasets with PDDL-derived data yields substantial improvements in both mathematical and non-mathematical reasoning, as demonstrated across multiple benchmarks. These findings indicate that planning problems constitute a scalable and effective resource for generating robust, precise, and fine-grained training data for PRMs, going beyond the classical mathematical sources that dominate this field.
