用强化学习让小模型自主学机器学习,效果接近大模型。
ML-Agent: Reinforcing LLM Agents for Autonomous Machine Learning Engineering
- 用在线强化学习让语言模型通过试错自主学机器学习。
- 70亿参数模型在9个任务上表现媲美大模型,成本更低。
- 适合想低成本实现自动机器学习的开发者和研究者。
大型语言模型(LLM)驱动的智能体推动了自主机器学习工程的发展,但主流的提示工程范式存在局限:小型模型无法从执行轨迹中学习以实现泛化,而大型专有模型则带来高昂的计算开销,限制可及性与扩展性。为此,我们首次探索基于学习的智能体机器学习范式,即通过在线强化学习(RL)让LLM智能体在机器学习任务中进行交互式实验来学习。为此提出一种新型智能体机器学习训练框架,包含三个核心组件:(1) 探索增强微调,使智能体生成多样化动作以提升强化学习探索能力;(2) 步进式强化学习,支持单步动作训练,加速经验收集并提升训练效率;(3) 面向智能体机器学习的奖励模块,将多样的机器学习反馈信号统一为一致的强化学习奖励。基于该框架,我们训练了由70亿参数的Qwen-2.5 LLM驱动的ML-Agent,在仅9个机器学习任务上训练后,其性能可与使用更大专有模型(如GPT-5)的智能体相当,但计算成本显著降低,展现出优异性能与跨任务泛化能力。
原文摘要 · Abstract (English)
The emergence of large language model (LLM)-based agents has significantly advanced the development of autonomous machine learning (ML) engineering. However, the dominant prompt-based paradigm exhibits limitations: smaller models lack the capacity to learn from execution trajectories for generalization, while large proprietary models incur high computational overhead, restricting accessibility and scalability. Focusing on this, for the first time, we explore the paradigm of learning-based agentic ML, where an LLM agent learns through interactive experimentation on ML tasks using online reinforcement learning (RL). To realize this, we propose a novel agentic ML training framework with three key components: (1) exploration-enriched fine-tuning, which enables LLM agents to generate diverse actions for enhanced RL exploration; (2) step-wise RL, which enables training on a single action step, accelerating experience collection and improving training efficiency; (3) an agentic ML-specific reward module, which unifies varied ML feedback signals into consistent rewards for RL optimization. Leveraging this framework, we train ML-Agent, driven by a 7B-sized Qwen-2.5 LLM for autonomous ML. Despite training on only 9 ML tasks, our 7B-sized ML-Agent achieves comparable performance to agents using much larger proprietary LLMs (e.g., GPT-5) but at significantly lower computational cost, demonstrating strong performance and cross-task generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。