arXiv:2412.21154cs.AIcs.CL2024-12被引 35

用开放模型让语言代理在科学任务上达到专家水平,成本降低100倍。

Aviary: training language agents on challenging scientific tasks

  • 将语言代理建模为解决语言决策过程的策略,支持多步推理与工具使用。
  • 在分子克隆、文献问答、蛋白稳定工程三类科学任务中超越人类专家。
  • 仅需开源模型+动态计算扩展,推理成本仅为前沿模型的1/100。

解决复杂现实任务需要反复的动作与观察循环,尤其在科学领域,任务常涉及多轮分析、工具使用和实验。语言代理因其能通过自然语言或代码与工具交互,有望自动化科学中的智力工作。然而其灵活性也带来实现上的概念与实践挑战,因代理可能包含非标准组件如内部推理、规划、工具调用,以及温度采样语言模型的固有随机性。本文提出 Aviary,一个可扩展的语言代理训练环境。我们将代理形式化为解决语言基部分可观测马尔可夫决策过程的策略,称为语言决策过程。随后构建了五个环境,其中三个为高挑战性的科学任务:(1)分子克隆中操作DNA构建,(2)通过访问科学文献回答研究问题,(3)工程化蛋白稳定性。这些环境聚焦于多步推理,且与现代生物学研究高度相关。通过在线训练与按需扩展推理算力,我们证明,基于开源非前沿大模型的语言代理可在多个任务上达到甚至超越前沿大模型代理与人类专家的表现,推理成本最高降低100倍。

原文摘要 · Abstract (English)

Solving complex real-world tasks requires cycles of actions and observations. This is particularly true in science, where tasks require many cycles of analysis, tool use, and experimentation. Language agents are promising for automating intellectual tasks in science because they can interact with tools via natural language or code. Yet their flexibility creates conceptual and practical challenges for software implementations, since agents may comprise non-standard components such as internal reasoning, planning, tool usage, as well as the inherent stochasticity of temperature-sampled language models. Here, we introduce Aviary, an extensible gymnasium for language agents. We formalize agents as policies solving language-grounded partially observable Markov decision processes, which we term language decision processes. We then implement five environments, including three challenging scientific environments: (1) manipulating DNA constructs for molecular cloning, (2) answering research questions by accessing scientific literature, and (3) engineering protein stability. These environments were selected for their focus on multi-step reasoning and their relevance to contemporary biology research. Finally, with online training and scaling inference-time compute, we show that language agents backed by open-source, non-frontier LLMs can match and exceed both frontier LLM agents and human experts on multiple tasks at up to 100x lower inference cost.

语言代理科学智能多步推理开源模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。