xLAM: A Family of Large Action Models to Empower AI Agent Systems

TL;DR

xLAM models outperform GPT-4 on the Berkeley Function-Calling Leaderboard, enhancing AI agent task performance.

cs.CL 🔴 Advanced 2024-09-05 7 views
Jianguo Zhang Tian Lan Ming Zhu Zuxin Liu Thai Hoang Shirley Kokane Weiran Yao Juntao Tan Akshara Prabhakar Haolin Chen Zhiwei Liu Yihao Feng Tulika Awalgaonkar Rithesh Murthy Eric Hu Zeyuan Chen Ran Xu Juan Carlos Niebles Shelby Heinecke Huan Wang Silvio Savarese Caiming Xiong
large models AI agents function calling open-source data synthesis

Key Findings

Methodology

xLAM series employs dense and mixture-of-expert architectures, ranging from 1B to 8x22B parameters. It enhances AI agents' generalizability and performance by unifying, augmenting, and synthesizing diverse datasets.

Key Results

  • xLAM ranks first on the Berkeley Function-Calling Leaderboard, outperforming GPT-4 and Claude-3, especially in tool use.
  • Experimental results show xLAM excels in multiple agent ability benchmarks, particularly in multi-task environments.
  • Smaller models achieve performance comparable to larger counterparts, demonstrating the potential of data synthesis.

Significance

The release of xLAM aims to advance open-source LLMs' performance in autonomous AI agents, accelerating progress and democratizing access to high-performance models, addressing the scarcity of high-quality agent datasets.

Technical Contribution

xLAM addresses the heterogeneity of existing datasets through data unification and augmentation, improving model performance and generalization capabilities via standardized function-calling formats.

Novelty

xLAM is the first to integrate multiple data processing techniques into a flexible pipeline, significantly enhancing model adaptability and performance in multi-task environments.

Limitations

  • While xLAM excels in multi-task environments, its fine-tuning capability in specific domains remains limited.
  • The quality of data synthesis may affect model performance in real-world scenarios.

Future Work

Future work will focus on enhancing fine-tuning capabilities in specific domains and exploring more data synthesis techniques to improve model robustness.

AI Executive Summary

The field of autonomous agents has seen significant advancements in recent years, particularly with the application of large language models (LLMs). However, the open-source community faces challenges in developing specialized models for agent tasks due to the scarcity of high-quality agent datasets and the absence of standard protocols. To address these issues, we introduce and publicly release xLAM, a series of large action models designed for AI agent tasks. The xLAM series includes five models with both dense and mixture-of-expert architectures, ranging from 1B to 8x22B parameters. By unifying, augmenting, and synthesizing diverse datasets, xLAM significantly enhances AI agents' generalizability and performance across varied environments. Our experimental results demonstrate that xLAM consistently delivers exceptional performance across multiple agent ability benchmarks, notably securing the first position on the Berkeley Function-Calling Leaderboard, outperforming GPT-4 and Claude-3. By releasing the xLAM series, we aim to advance the performance of open-source LLMs for autonomous AI agents, potentially accelerating progress and democratizing access to high-performance models.

Deep Analysis

Background

The field of autonomous agents has witnessed significant advancements, especially with the application of large language models (LLMs). Researchers have made substantial progress in developing sophisticated frameworks and specialized environments to enhance agent capabilities, such as tool use and web browsing. Concurrently, comprehensive benchmarks like AgentBench, ToolBench, and AgentBoard have been established to rigorously assess agent performance in reasoning, planning, and multi-turn interactions.

Core Problem

Despite the competitive performance of proprietary LLMs in various agent tasks, the open-source community faces limited choices for specialized models. This scarcity stems from the lack of comprehensive, high-quality datasets and the heterogeneity of existing data formats.

Innovation

xLAM addresses the heterogeneity of existing datasets through data unification and augmentation, improving model performance and generalization capabilities. It is the first to integrate multiple data processing techniques into a flexible pipeline, significantly enhancing model adaptability and performance in multi-task environments.

Methodology

  • �� Data Processing: Enhances dataset diversity and mitigates overfitting through data unification and augmentation.

  • �� Data Synthesis: Demonstrates the impact of scalable, high-quality data synthesis on agent model performance.

  • �� Model Training: Utilizes supervised fine-tuning and direct preference optimization to enhance model robustness.

Experiments

We evaluate the xLAM series on Webshop, ToolQuery, ToolBench, and the Berkeley Function-Calling Benchmark, demonstrating exceptional performance across various agent tasks. Notably, the xLAM-8x22b model secures the top position on the Berkeley Function-Calling Leaderboard with a wide margin.

Results

xLAM ranks first on the Berkeley Function-Calling Leaderboard, outperforming GPT-4 and Claude-3, especially in tool use. Experimental results show xLAM excels in multiple agent ability benchmarks, particularly in multi-task environments.

Applications

The xLAM series models are suitable for AI agent tasks in multi-task environments, particularly in scenarios requiring complex function calling and tool use.

Limitations & Outlook

While xLAM excels in multi-task environments, its fine-tuning capability in specific domains remains limited. The quality of data synthesis may affect model performance in real-world scenarios.

Plain Language Accessible to non-experts

Imagine a large factory where xLAM acts as a super-intelligent robot capable of working in different workshops. This robot not only understands the workflow of each workshop but can also switch tools as needed to complete complex tasks. It's like cooking in a kitchen, where you need to know how to chop vegetables and when to use a pan or an oven. By learning a vast array of 'recipes,' xLAM can operate seamlessly in various 'kitchens.'

ELI14 Explained like you're 14

Hey there! Imagine you're playing a super complex game with lots of tasks and tools. xLAM is like a super smart game character that helps you solve all kinds of puzzles. For example, if you need to find a mysterious key, xLAM will tell you where to look and even help you unlock hidden doors. It's like having a game guide that helps you level up and conquer challenges!

Glossary

Large Language Model (LLM)

A large neural network model capable of processing and generating natural language.

Used in xLAM to enhance AI agent capabilities.

Function Calling

The process of generating and executing specific functions by the model.

xLAM excels in function calling on the Berkeley leaderboard.

Data Synthesis

A method to enhance dataset diversity by generating new data samples.

Used to improve xLAM model performance.

Supervised Fine-Tuning

Further training of a model using labeled data to improve its performance.

One of the training methods for xLAM models.

Direct Preference Optimization (DPO)

A method to improve model performance by optimizing its preferences.

Used for alignment training of xLAM models.

Open Questions Unanswered questions from this research

  • 1 How to enhance xLAM's fine-tuning capability in specific domains remains to be explored.
  • 2 The impact of data synthesis quality on model performance in real-world scenarios.

Applications

Immediate Applications

AI Agents in Multi-Task Environments

xLAM can be used for AI agent tasks requiring complex function calling and tool use.

Long-term Vision

Democratizing High-Performance Models

The release of xLAM is expected to accelerate the democratization of high-performance open-source models.

Abstract

Autonomous agents powered by large language models (LLMs) have attracted significant research interest. However, the open-source community faces many challenges in developing specialized models for agent tasks, driven by the scarcity of high-quality agent datasets and the absence of standard protocols in this area. We introduce and publicly release xLAM, a series of large action models designed for AI agent tasks. The xLAM series includes five models with both dense and mixture-of-expert architectures, ranging from 1B to 8x22B parameters, trained using a scalable, flexible pipeline that unifies, augments, and synthesizes diverse datasets to enhance AI agents' generalizability and performance across varied environments. Our experimental results demonstrate that xLAM consistently delivers exceptional performance across multiple agent ability benchmarks, notably securing the 1st position on the Berkeley Function-Calling Leaderboard, outperforming GPT-4, Claude-3, and many other models in terms of tool use. By releasing the xLAM series, we aim to advance the performance of open-source LLMs for autonomous AI agents, potentially accelerating progress and democratizing access to high-performance models for agent tasks. Models are available at https://huggingface.co/collections/Salesforce/xlam-models-65f00e2a0a63bbcd1c2dade4

cs.CL cs.AI cs.LG