atlas

Research map / Large language models / Agents, code and math

APIGen: Automated PIpeline for Generating Verifiable and Diverse Function-Calling Datasets

NeurIPS 2024 · Large language models › Agents, code and math · approach: Software engineering agents · 8 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Introduces APIGen, a multi-stage verified (format, execution, semantic) pipeline for generating diverse function-calling data, releasing a 60K-entry dataset and xLAM models that rank highly on BFCL.

Problem

Function-calling agents need diverse, reliable, high-quality training data, but existing agent datasets are static, rarely verified, and noisy, and rarely cover parallel function calls.

Approach

Generate query-answer pairs with LLMs from sampled executable APIs and seed examples, then verify each sample via format checking, real API execution, and an LLM semantic checker, with diversity from four query styles (simple, multiple, parallel, parallel multiple) and randomized sampling.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune DeepSeek-Coder-7B-instruct-v1.5 on APIGen-generated dataDeepSeek-Coder-7B-instruct-v1.5 · BFCLoverall accuracyHelped
Fine-tune DeepSeek-Coder-1.3B-instruct on APIGen-generated dataDeepSeek-Coder-1.3B-instruct · BFCLoverall accuracyHelped
Add data filtered by the semantic checker to the training set (ablation)xLAM models trained on APIGen data · BFCLoverall accuracyHurt
Add data filtered by the execution checker to the training set (ablation)xLAM models trained on APIGen data · BFCLoverall accuracyHurt

Surprising findings

Limitations

Experimental setup

Datasets: ToolBench Berkeley Function-Calling Benchmark seed styles

Benchmarks: Berkeley Function-Calling Leaderboard (BFCL)

Metrics: overall accuracy AST accuracy executable function accuracy relevance detection accuracy

Baselines: GPT-4o GPT-4-Turbo Gemini-1.5-Pro Claude-3-Haiku GPT-3.5-Turbo Gorilla OpenFunctions-v2 Command-R-Plus Llama-3-70B-Instruct

Builds on: ToolBench AgentOhana xLAM

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy