πͺ Agent Lumos: Unified and Modular Training for Open-Source Language Agents
π[Website] π[Paper] π€[Data] π€[Model] π€[Demo]
We introduce πͺLumos, Language Agents with Unified Formats, Modular Design, and Open-Source LLMs. Lumos unifies a suite of complex interactive tasks and achieves competitive performance with GPT-4/3.5-based and larger open-source agents.
Lumos has following features:
- 𧩠Modular Architecture:
- 𧩠Lumos consists of planning, grounding, and execution modules built based on LLAMA-2-7B/13B and off-the-shelf APIs.
- π€ Lumos utilizes a unified data format that encompasses multiple task types, thereby enabling the developed agent framework to conveniently support a range of interactive tasks.
- π Diverse Training Data:
- π Lumos is trained with ~56K diverse high-quality subgoal/action annotations from ground-truth reasoning steps in existing benchmarks with GPT-4.
- βοΈ Lumos data can be instrumental for future research in developing open-source agents for complex interactive tasks.
- π Competitive Performance:
- π Lumos is comparable or even beats GPT-series agents on web/complex QA tasks Mind2Web and HotpotQA, and larger open agents on math and multimodal tasks.
- π Lumos exceeds contemporaneous agents that have been fine-tuned with in-domain HotpotQA, Mind2Web and ScienceQA annotations, such as FiReAct, AgentLM, and AutoAct.
- π Lumos performs better than open agent baseline formulations including chain-of-thoughts and integrated training.
- π Lumos surpasses larger open LLM agents and domain-specific agents on unseen tasks, WebShop and InterCode_SQL.
Model Overview
lumos_maths_plan_onetime-13B
is a planning module checkpoint finetuned on maths task in Lumos-Onetime (Lumos-O) formulation.
The training annotation is shown below:
Training Data | Number |
---|---|
lumos_maths_plan_onetime |
19778 |
Citation
If you find this work is relevant with your research, please feel free to cite our work!
@article{yin2023lumos,
title={Agent Lumos: Unified and Modular Training for Open-Source Language Agents},
author={Yin, Da and Brahman, Faeze and Ravichander, Abhilasha and Chandu, Khyathi and Chang, Kai-Wei and Choi, Yejin and Lin, Bill Yuchen},
journal={arXiv preprint arXiv:2311.05657},
year={2023}
}
- Downloads last month
- 12
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social
visibility and check back later, or deploy to Inference Endpoints (dedicated)
instead.