Text Generation
Transformers
PyTorch
internlm
agent
tool
custom_code
lovesnowbest commited on
Commit
0a8cd42
·
verified ·
1 Parent(s): 7845963

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +77 -0
README.md CHANGED
@@ -1,3 +1,80 @@
1
  ---
2
  license: apache-2.0
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: apache-2.0
3
+ datasets:
4
+ - lovesnowbest/Agent-FLAN
5
+ pipeline_tag: text-generation
6
+ tags:
7
+ - agent
8
+ - tool
9
  ---
10
+ ---
11
+ license: apache-2.0
12
+ ---
13
+ # Agent-FLAN: Designing Data and Methods of Effective Agent Tuning for Large Language Models
14
+
15
+ ## ✨ Introduction
16
+
17
+ [[🤗 HuggingFace](https://huggingface.co/lovesnowbest/Agent-FLAN-7b)]
18
+ [[📃 Paper](https://arxiv.org/abs/)]
19
+ [[🌐 Project Page](https://interlm.github.io/agent-flan/)]
20
+
21
+ > Open-sourced Large Language Models (LLMs) have achieved great success in various NLP tasks, however, they are still far inferior to API-based models when acting as agents. How to integrate agent ability into general LLMs becomes a crucial and urgent problem. This paper first delivers three key observations: (1) the current agent training corpus is entangled with both formats following and agent reasoning, which significantly shifts from the distribution of its pre-training data; (2) LLMs exhibit different learning speeds on the capabilities required by agent tasks; and (3) current approaches have side-effects when improving agent abilities by introducing hallucinations. Based on the above findings, we propose Agent-FLAN to effectively Fine-tune LANguage models for Agents. Through careful decomposition and redesign of the training corpus, Agent-FLAN enables Llama2-7B to outperform prior best works by 3.5% across various agent evaluation datasets. With comprehensively constructed negative samples, Agent-FLAN greatly alleviates the hallucination issues based on our established evaluation benchmark. Besides, it consistently improves the agent capability of LLMs when scaling model sizes while slightly enhancing the general capability of LLMs.
22
+
23
+ ## ♟️ Agent-FLAN
24
+
25
+ Agent-FLAN series are finetuned on AgentInstruct and Toolbench by applying the data generation pipeline proposed in Agent-FLAN paper, which holds strong abilities on various agent tasks and tool utilization~
26
+
27
+ <div>
28
+ <center>
29
+ <img src="docs/figure/teaser.png">
30
+
31
+ Comparison of recent agent tuning approaches on Held-In, Held-Out tasks. Performances are normalized with GPT-4 results for better visualization. * denotes our re-implementation for a fair comparison.
32
+ </div>
33
+
34
+ ### 🤗 HuggingFace Model & Dataset
35
+
36
+ Agent-FLAN is produced by mixed training on AgentInstruct, ToolBench, and ShareGPT datasets from the Llama2-chat series.
37
+
38
+ The models follow the conversation format of Llama-2-chat, with the template protocol as:
39
+ ```python
40
+ dict(role='user', begin='<|Human|>െ', end='\n '),
41
+ dict(role='system', begin='<|Human|>െ', end='\n '),
42
+ dict(role='assistant', begin='<|Assistant|>െ', end='ി\n '),
43
+ ```
44
+
45
+ The 7B model is available on Huggingface model hub.
46
+
47
+ | Model | Huggingface Repo |
48
+ | :---------: | :------------------------------------------------------------: |
49
+ | Agent-FLAN-7B | [🤗Huggingface Link](https://huggingface.co/lovesnowbest/Agent-FLAN-7b) |
50
+
51
+ The Agent-FLAN dataset is also available on Huggingface dataset hub.
52
+
53
+ | Dataset | Huggingface Repo |
54
+ | :---------: | :------------------------------------------------------------: |
55
+ | Agent-FLAN Dataset | [🤗Huggingface Link](https://huggingface.co/datasets/lovesnowbest/Agent-FLAN) |
56
+
57
+ ## 💫 Detailed Results
58
+
59
+ <div>
60
+ <center>
61
+ <img src="docs/figure/performance.png" width="100%">
62
+ </div>
63
+
64
+ Main results of Agent-FLAN. Agent-FLAN significantly outperforms previous agent-tuning approaches by a large margin on both held-in and held-out tasks. * denotes our re-implementation with the same amount of training data for a fair comparison. Since FireAct does not train on AgentInstruct dataset, we omit its performance on the HELD-IN set. Bold: the best in API-based and open-sourced models.
65
+
66
+
67
+ ## ❤️ Acknowledgements
68
+
69
+ Agent-FLAN is built with [Lagent](https://github.com/InternLM/lagent) and [T-Eval](https://github.com/open-compass/t-eval). Thanks for their awesome work!
70
+
71
+ ## 🖊️ Citation
72
+
73
+ If you find this project useful in your research, please consider cite:
74
+ ```
75
+ TBD
76
+ ```
77
+
78
+ ## 💳 License
79
+
80
+ This project is released under the Apache 2.0 [license](./LICENSE).