用低成本实验训练智能体,自动优化高成本大模型配置。
AutoLLMResearch: Training Research Agents for Automating LLM Experiment Configuration - Learning from Cheap, Optimizing Expensive

- 模拟人类研究者从低精度实验中提炼规律,跨精度推断最优配置。
- 在百万级GPU小时数据上验证,显著提升配置效率与泛化能力。
- 适合需要高效调参的大模型研究者,尤其适用于资源受限场景。
高效配置可扩展大语言模型(LLM)实验,涵盖架构设计、超参数调优等,对推进LLM研究至关重要,因不当配置会浪费大量计算资源并阻碍模型潜力发挥。现有自动化方法仅适用于低成本场景,而可扩展的LLM实验成本过高,无法承受反复试错。目前尚无工作解决高成本实验配置自动化问题,导致该过程依赖人工经验且效率低下。为此,我们提出AutoLLMResearch,一个类人研究者的学习框架:通过低保真度实验学习通用原则,并外推至高成本设置以快速定位优质配置。核心挑战在于如何让智能体在多保真度实验环境中学习配置空间结构。为此,我们构建了两个关键组件:1)LLMConfig-Gym,一个包含四项关键任务的多保真度环境,基于超过一百万GPU小时的可验证实验结果;2)结构化训练流程,将配置研究建模为长时程马尔可夫决策过程,激励跨保真度推理。在多个独立测试集上对比多种强基线的评估表明,本框架在有效性、泛化性与可解释性方面均表现优异,具备作为真实世界大规模LLM实验自动化的实用解决方案潜力。
原文摘要 · Abstract (English)
Effectively configuring scalable large language model (LLM) experiments, spanning architecture design, hyperparameter tuning, and beyond, is crucial for advancing LLM research, as poor configuration choices can waste substantial computational resources and prevent models from realizing their full potential. Prior automated methods are designed for low-cost settings where repeated trial and error is feasible, but scalable LLM experiments are too expensive for such extensive iteration. To our knowledge, no work has addressed the automation of high-cost LLM experiment configurations, leaving this problem labor-intensive and dependent on expert intuition. Motivated by this gap, we propose AutoLLMResearch, an agentic framework that mimics how human researchers learn generalizable principles from low-fidelity experiments and extrapolate to efficiently identify promising configurations in expensive LLM settings. The core challenge is how to enable an agent to learn, through interaction with a multi-fidelity experimental environment that captures the structure of the LLM configuration landscape. To achieve this, we propose a systematic framework with two key components: 1) LLMConfig-Gym, a multi-fidelity environment encompassing four critical LLM experiment tasks, supported by over one million GPU hours of verifiable experiment outcomes; 2) A structured training pipeline that formulates configuration research as a long-horizon Markov Decision Process and accordingly incentivizes cross-fidelity extrapolation reasoning. Extensive evaluation against diverse strong baselines on held-out experiments demonstrates the effectiveness, generalization, and interpretability of our framework, supporting its potential as a practical and general solution for scalable real-world LLM experiment automation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。