Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

ToolPRM: Fine-Grained Inference Scaling of Structured Outputs for Function Calling

About

Large language models (LLMs) excel at function calling, but inference scaling has been explored mainly for unstructured generation. We propose an inference-scaling framework for structured outputs that combines fine-grained beam search with \textbf{ToolPRM}, a process reward model scoring each intra-call decision (function name and argument filling). We build the first fine-grained intra-call supervision dataset via function masking, rollout collection, and step-level annotation. ToolPRM outperforms outcome and coarse-grained reward models in predictive accuracy and yields consistent test-time gains on multiple function-calling benchmarks. We further show that structured generation follows ``\textbf{explore more but retain less}'', since early JSON errors are unrecoverable.

Jianghao Lin, Yuanyuan Shi, Xin Peng, Renjie Ding, Hairui Wang, Yuxuan Peng, Bizhe Bai, Weixi Song, Fengshuo Bai, Huacan Chai, Weinan Zhang, Fei Huang, Ying Wen• 2025

Related benchmarks

TaskDatasetResultRank
Function CallingTool-Alpaca
F1 Score73.36
40
Function CallingBFCL
Success Rate (Simple)80.5
29
Tool CallingAPI-Bank L-1 v1 (test)--
12
Tool UseAPI-Bank L2 cleaned (test)
F1 (API Matching)87.32
5
Showing 4 of 4 rows

Other info

Follow for update