arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2412.15660cs.AIcs.CLcs.SE

可适配且精准:企业场景 LLM Function-Calling 能力训练 Pipeline

Adaptable and Precise: Enterprise-Scenario LLM Function-Calling Capability Training Pipeline

Digital China AI Research(神州数码AI研究院)

Guancheng Zeng, Wentao Ding, Beining Xu, Chi Zhang, Wenqiang Han, Gang Li, Jingjing Mo, Pengxu Qiu, Xinran Tao, Wang Tao, Haowen Hu

AI总结:

针对企业场景中通用 LLM function-calling 能力不足的问题,提出包含数据合成增强、模型微调与评估的训练 pipeline,基于 Qwen2.5-Coder-7B-Instruct 和 LoRA 微调的模型在测试集准确率上超越 GPT-4 和 GPT-4o。

摘要:

企业拥有散布于各种功能中的海量 API 资产,构成了现有业务流程的骨干。通过将这些 API 作为功能工具,企业可以设计由本地化 function-calling 模型作为核心引擎驱动的、多样化的场景特定智能体应用。然而,通用模型在计算效率、输出准确性和稳定性方面往往无法满足企业需求,因此需要进行场景特定的适配。在本文中,我们提出了一种针对真实业务场景量身定制的 function-calling 能力训练 pipeline。该 pipeline 包括场景特定 function-calling 数据的合成与增强、模型微调以及性能评估与分析。使用该 pipeline,我们在数字 HR agent 场景中生成了 1,260 个完全由 AI 生成的样本和 1,035 个增强的人工标注样本。采用 Qwen2.5-Coder-7B-Instruct 模型作为基础模型,并在四块具有 24GB VRAM 的 GPU 上使用 LoRA 方法进行微调。我们微调后的模型在评估和实际应用中表现出色,在测试集上的准确率超越了 GPT-4 和 GPT-4o。这些结果验证了所提出的 pipeline 在训练场景特定 function-calling 模型方面的可靠性。

英文摘要:

Enterprises possess a vast array of API assets scattered across various functions, forming the backbone of existing business processes. By leveraging these APIs as functional tools, enterprises can design diverse, scenario-specific agent applications, driven by on-premise function-calling models as the core engine. However, generic models often fail to meet enterprise requirements in terms of computational efficiency, output accuracy, and stability, necessitating scenario-specific adaptation. In this paper, we propose a training pipeline for function-calling capabilities tailored to real-world business scenarios. This pipeline includes the synthesis and augmentation of scenario-specific function-calling data, model fine-tuning, and performance evaluation and analysis. Using this pipeline, we generated 1,260 fully AI-generated samples and 1,035 augmented manually-labeled samples in digital HR agent scenario. The Qwen2.5-Coder-7B-Instruct model was employed as the base model and fine-tuned using the LoRA method on four GPUs with 24GB VRAM. Our fine-tuned model demonstrated outstanding performance in evaluations and practical applications, surpassing GPT-4 and GPT-4o in accuracy on the test set. These results validate the reliability of the proposed pipeline for training scenario-specific function-calling models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑