返回评测
Agent Agent-数据科学 未饱和 模态 · 表格 任务 · 执行验证

ML-Bench

评测 LLM Agent 完成真实 ML 工程任务的能力,涵盖数据处理到模型部署

Shanghai AI Lab ·

考察能力

ML工程 代码执行 环境配置 实验管理

构建信息

数据规模
9,641 tasks from 18 repos
评测方式
执行验证
核心指标
Pass Rate

模型效果排行

🥇 GPT-4o
~43%
🥈 Claude 3.5 Sonnet
~40%

为什么重要

ML-Bench 覆盖了 ML 工程全链路:数据预处理、模型训练、超参调优、结果评估。它基于真实 GitHub ML 仓库,反映了 AI 研究人员的日常工作。

构建方式

  1. 从 18 个真实 ML GitHub 仓库收集任务
  2. 包含 README 指令到实际执行的映射
  3. Docker 容器化执行环境

任务示例

任务:“Run the training script for the sentiment classifier with learning rate 1e-4 and batch size 32 on the SST-2 dataset.” 验证:脚本正确执行并产出期望的训练日志

关键发现

  • 环境配置是最大障碍(依赖冲突等)
  • 模型在简单任务上表现不错,复杂任务快速退化
  • 文档理解能力是关键

关联基准

不足与展望

  • 仓库可能过时
  • 执行环境差异影响复现
  • 但 ML 工程评测方向有价值
ML工程 Agent 代码执行