论文

TREX:通过智能体驱动的树状探索实现LLM微调自动化

TREX: Automating LLM Fine-tuning via Agent-Driven Tree-based Exploration

智能体系统Agent 协作Agent Harness

摘要

尽管大语言模型(LLM)已赋能AI研究智能体执行孤立的科学任务,但自动化LLM训练这类复杂的真实世界工作流仍是重大挑战。本文提出TREX,一个将LLM训练全生命周期自动化的多智能体系统。通过编排Researcher与Executor两个核心模块之间的协作,该系统无缝完成需求分析、开放域文献与数据调研、训练策略制定、数据配方准备以及模型训练与评估。多轮实验过程被建模为搜索树,使系统能够高效规划探索路径、复用历史结果,并从迭代试验中提炼高层洞见。为评估自动化LLM训练能力,我们构建了FT-Bench,一个由源自真实场景的10个任务组成的基准,涵盖从优化基础模型能力到提升领域特定任务表现。实验结果表明,TREX智能体能在目标任务上持续优化模型性能。

TREX:通过智能体驱动的树状探索实现LLM微调自动化:论文配图
图 1:TREX 在分子生成方面的研究进展图示。左上:系统将任务定义作为输入,包括任务描述、基准和其他信息。右上:随着 TREX 自动化研究的进展,模型性能不断提高。左下:关键步骤的详细信息。右下:TREX(基于 Qwen3-1.7B)与基于 Llama 系列基线的人类专家微调模型的比较。