RolandXMR commited on
Commit
cd4c971
·
verified ·
1 Parent(s): 48d4d5b

Upload EnvFactory-1.7B.md

Browse files
Files changed (1) hide show
  1. EnvFactory-1.7B.md +150 -0
EnvFactory-1.7B.md ADDED
@@ -0,0 +1,150 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ tags:
4
+ - tool-use
5
+ - agentic-rl
6
+ - environment-synthesis
7
+ - EnvFactory
8
+ license: apache-2.0
9
+ datasets:
10
+ - LARK-Lab/EnvFactory-RL
11
+ - LARK-Lab/EnvFactory-SFT-FILTERED
12
+ language:
13
+ - en
14
+ base_model:
15
+ - Qwen/Qwen3-1.7B
16
+ ---
17
+
18
+ <h2 align="center">
19
+ EnvFactory: Scaling Tool-Use Agents via Executable Environments Synthesis and Robust RL
20
+ </h2>
21
+
22
+ <p align="center">
23
+ <a href="https://arxiv.org/abs/2605.18703">
24
+ <img
25
+ src="https://img.shields.io/badge/Paper-Arxiv-red?logo=arxiv&logoColor=red"
26
+ alt="EnvFactory Paper on arXiv"
27
+ />
28
+ </a>
29
+ <a href="https://github.com/LARK-AI-Lab/EnvFactory">
30
+ <img
31
+ src="https://img.shields.io/badge/GitHub-Code-181717?logo=github&logoColor=white"
32
+ alt="GitHub Code"
33
+ />
34
+ </a>
35
+ <a href="https://lark-ai-lab.github.io/envfactory.github.io/">
36
+ <img
37
+ src="https://img.shields.io/badge/GitHub-Page-4078c0?logo=github&logoColor=white"
38
+ alt="GitHub Page"
39
+ />
40
+ </a>
41
+ <a href="https://huggingface.co/collections/LARK-Lab/envfactory">
42
+ <img
43
+ src="https://img.shields.io/badge/Datasets-Hugging%20Face%20Data-orange?logo=huggingface&logoColor=yellow"
44
+ alt="Datasets on Hugging Face"
45
+ />
46
+ </a>
47
+ <a href="https://huggingface.co/collections/LARK-Lab/envfactory">
48
+ <img
49
+ src="https://img.shields.io/badge/EnvFactory-Hugging%20Face%20Model-FFCC00?logo=huggingface&logoColor=yellow"
50
+ alt="EnvFactory on Hugging Face"
51
+ />
52
+ </a>
53
+ </p>
54
+
55
+ ## Overview
56
+
57
+ We propose **EnvFactory**, a fully automated framework that addresses the challenges of equipping LLMs with tool-use capabilities via Agentic Reinforcement Learning (Agentic RL). EnvFactory autonomously explores and verifies stateful, executable tool environments from authentic resources, and synthesizes natural multi-turn trajectories through topology-aware sampling and calibrated refinement, producing grounded queries with implicit intents.
58
+
59
+ This model is the official **EnvFactory-1.7B** trained from Qwen/Qwen3-1.7B using SFT and RL on synthesized tool-use trajectories.
60
+
61
+ ## Key Features
62
+
63
+ - **Executable Environment Synthesis**: Automatically discovers, validates, and deploys MCP-based tool environments from real-world APIs
64
+ - **Topology-Aware Trajectory Sampling**: Generates natural multi-turn tool-use trajectories that capture implicit human reasoning
65
+ - **Robust RL Training**: Uses verified environments and calibrated refinement for stable reinforcement learning
66
+ - **Scalable Architecture**: Achieves superior performance with significantly fewer environments (85 environments across 7 domains)
67
+
68
+ ## Training Details
69
+
70
+ ### Training Data
71
+
72
+ - **SFT Data**: [LARK-Lab/EnvFactory-SFT-FILTERED](https://huggingface.co/datasets/LARK-Lab/EnvFactory-SFT-FILTERED) - 53.4k filtered trajectories
73
+ - **RL Data**: [LARK-Lab/EnvFactory-RL](https://huggingface.co/datasets/LARK-Lab/EnvFactory-RL) - 3.09k trajectories
74
+
75
+ ### Training Procedure
76
+
77
+ - **SFT Stage**: Full fine-tuning using LlamaFactory with DeepSpeed ZeRO-3
78
+ - **RL Stage**: Reinforcement learning using forked VeRL framework
79
+ - **Base Model**: Qwen/Qwen3-1.7B
80
+ - **Training Epochs**: 1 epoch for SFT
81
+ - **Learning Rate**: 1.0e-6 with cosine scheduler
82
+ - **Batch Size**: 1 per device with gradient accumulation of 32
83
+
84
+ ## Performance
85
+
86
+ EnvFactory models achieve significant improvements on tool-use benchmarks:
87
+
88
+ | Benchmark | Improvement |
89
+ |-----------|-------------|
90
+ | BFCLv3 | +15% |
91
+ | MCP-Atlas | +8.6% |
92
+ | τ²-Bench | +6% |
93
+ | VitaBench | +6% |
94
+
95
+ ## Usage
96
+
97
+ ### Tool-Use Agent
98
+
99
+ ```python
100
+ from transformers import AutoTokenizer, AutoModelForCausalLM
101
+ import torch
102
+
103
+ model_path = "LARK-Lab/EnvFactory-1.7B"
104
+ tokenizer = AutoTokenizer.from_pretrained(model_path)
105
+ model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.bfloat16, device_map="auto")
106
+
107
+ # Example tool-use conversation
108
+ messages = [
109
+ {"role": "system", "content": "You are a helpful assistant with access to various tools."},
110
+ {"role": "user", "content": "Search for recent papers about tool-use agents on arxiv."}
111
+ ]
112
+
113
+ input_ids = tokenizer.apply_chat_template(messages, tokenize=True, return_tensors="pt").to(model.device)
114
+ outputs = model.generate(input_ids, max_new_tokens=1024, temperature=0.7, top_p=0.9)
115
+ response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
116
+ print(response)
117
+ ```
118
+
119
+ ### With MCP Tools
120
+
121
+ ```python
122
+ # Load MCP tool configuration
123
+ import json
124
+
125
+ with open("configs/mcp_server.json", "r") as f:
126
+ mcp_config = json.load(f)
127
+
128
+ # Use with your preferred MCP client
129
+ # See https://github.com/LARK-AI-Lab/EnvFactory for integration details
130
+ ```
131
+
132
+ ## Citation
133
+
134
+ If you find our work helpful, please consider citing:
135
+
136
+ ```bibtex
137
+ @misc{xu2026envfactoryscalingtooluseagents,
138
+ title={EnvFactory: Scaling Tool-Use Agents via Executable Environments Synthesis and Robust RL},
139
+ author={Minrui Xu and Zilin Wang and Mengyi DENG and Zhiwei Li and Zhicheng Yang and Xiao Zhu and Yinhong Liu and Boyu Zhu and Baiyu Huang and Chao Chen and Heyuan Deng and Fei Mi and Lifeng Shang and Xingshan Zeng and Zhijiang Guo},
140
+ year={2026},
141
+ eprint={2605.18703},
142
+ archivePrefix={arXiv},
143
+ primaryClass={cs.CL},
144
+ url={https://arxiv.org/abs/2605.18703},
145
+ }
146
+ ```
147
+
148
+ ## License
149
+
150
+ This model is released under the Apache 2.0 License.