少即是多!10亿参数小巨人击败ChatGPT

简介: 【7月更文挑战第9天】Salesforce AI Research团队的APIGen提出了一种自动化方法,生成可验证的函数调用数据集,用于提升LLMs的微调。使用APIGen,即使10亿参数的模型也能在功能调用基准上超越GPT-4等大模型。发布的60K数据集旨在促进该领域的研究。尽管目前局限于Python和REST API,APIGen展示了小模型如何通过高质量数据挑战大模型,为语言模型的效率提升开辟新途径。[论文链接](https://arxiv.org/pdf/2406.18518)

最近,一篇名为"APIGen: Automated PIpeline for Generating Verifiable and Diverse Function-Calling Datasets"的论文引起了广泛关注。该论文提出了一种名为APIGen的自动化数据生成管道,旨在为函数调用应用生成可验证的高质量数据集。这个研究项目由Salesforce AI Research的团队完成,他们使用APIGen收集了3,673个可执行的API,涵盖了21个不同类别,以生成各种函数调用数据集。

该研究的目的是解决当前函数调用代理模型在实际应用中面临的挑战。这些模型,如GPT-4、Gemini和Mistral,已经发展到不仅能理解和生成类似人类的文本,还能根据自然语言指令执行功能性API调用。然而,这些模型的部署往往受到训练数据质量的限制。当前的数据集通常是静态的,缺乏全面的验证,导致模型在实际应用中的微调可能存在潜在的不准确性和低效率。

为了解决这些挑战,研究团队提出了APIGen框架。该框架旨在通过提供高质量、多样化的数据集来促进函数调用大型语言模型(LLMs)的微调,这些数据集更好地反映了真实世界API使用的多样性和复杂性。每个生成的数据点都经过了严格的多阶段验证过程:格式检查、实际函数执行和语义验证,以确保其可靠性和正确性。

研究团队使用APIGen生成的数据集对函数调用模型进行了微调。结果显示,即使只有70亿参数,这些模型也能在Berkeley函数调用基准测试中实现最先进的性能,超过了多个GPT-4模型。此外,他们的10亿参数模型也表现出色,超过了GPT-3.5-Turbo和Claude-3 Haiku。

研究团队还发布了一个包含60,000个高质量条目的数据集,以促进函数调用代理领域的发展。这个大规模的合成数据集旨在促进进一步的研究和开发,为研究人员和开发人员提供训练和测试其模型的基础。

然而,APIGen框架和生成的数据集也存在一些限制。目前,该框架和数据集只考虑了REST API和Python函数。此外,虽然APIGen是一个通用框架,但目前只实现了单轮函数调用的生成过程。未来的工作将重点扩展APIGen以支持更多场景、编程语言和API。

尽管存在这些限制,但APIGen框架和生成的数据集代表了在开发高效、有效的函数调用代理方面迈出了重要一步。通过关注数据质量和多样性,研究团队已经能够显著提高模型的函数调用能力,甚至使较小的模型能够与更大的模型相竞争。这为开发更高效、更强大的语言模型打开了新的可能性,特别是在代理工具使用领域。

论文地址:https://arxiv.org/pdf/2406.18518

目录
相关文章
|
5月前
|
人工智能 搜索推荐 数据可视化
国产黑马一年肝出万亿参数MoE!霸榜多模态
【4月更文挑战第2天】阶跃星辰推出万亿参数的MoE多模态大模型,引领AI新突破。采用混合专家架构,适应不同任务,提升效率与性能。MoE已应用于跃问助手和冒泡鸭AI平台,提供个性化服务与丰富互动体验。然而,巨大模型的训练管理、过拟合、知识表示及伦理问题仍是AGI发展道路上的挑战。
59 4
国产黑马一年肝出万亿参数MoE!霸榜多模态
|
5月前
|
人工智能 测试技术 量子技术
Claude 3智商测试101分超越人类/碾压GPT-4!
【2月更文挑战第16天】Claude 3智商测试101分超越人类/碾压GPT-4!
61 1
Claude 3智商测试101分超越人类/碾压GPT-4!
|
5月前
|
机器学习/深度学习 人工智能 数据可视化
太强!AI没有落下的腾讯出YOLO-World爆款 | 开集目标检测速度提升20倍,效果不减
太强!AI没有落下的腾讯出YOLO-World爆款 | 开集目标检测速度提升20倍,效果不减
603 0
|
机器人
第一个超越ChatGPT的开源模型来了?网友并不买账
第一个超越ChatGPT的开源模型来了?网友并不买账
322 1
|
机器学习/深度学习 数据采集 人工智能
150亿参数、一次通过率超60%,华为代码能力超GPT3.5的大模型来了
150亿参数、一次通过率超60%,华为代码能力超GPT3.5的大模型来了
180 0
|
机器学习/深度学习 人工智能 自然语言处理
【ChatGPT】阿里版 ChatGPT 突然官宣意味着什么?
【ChatGPT】阿里版 ChatGPT 突然官宣意味着什么?
717 0
|
机器学习/深度学习 存储 人工智能
英伟达把GPT-4塞进我的世界,打游戏快15倍:AI大佬沉默了
英伟达把GPT-4塞进我的世界,打游戏快15倍:AI大佬沉默了
152 0
|
机器人 PyTorch 算法框架/工具
300美元复刻ChatGPT九成功力,GPT-4亲自监考,130亿参数开源模型「小羊驼」来了
300美元复刻ChatGPT九成功力,GPT-4亲自监考,130亿参数开源模型「小羊驼」来了
331 0
|
机器学习/深度学习 人工智能 API
斯坦福70亿参数开源模型媲美GPT-3.5,100美元即可复现
斯坦福70亿参数开源模型媲美GPT-3.5,100美元即可复现
130 0
|
机器学习/深度学习 人工智能 自然语言处理
150亿参数,谷歌开源了史上最大视觉模型V-MoE的全部代码
150亿参数,谷歌开源了史上最大视觉模型V-MoE的全部代码
535 0