Abstract:Despite Large Language Models (LLMs) like GPT-4 achieving impressive results in function-level code generation, they struggle with repository-scale code understanding (e.g., coming up with the right arguments for calling routines), requiring a deeper comprehension of complex file interactions. Also, recently, people have developed LLM agents that attempt to interact with repository code (e.g., compiling and evaluating its execution), prompting the need to evaluate their performance. These gaps have motivated our development of ML-Bench, a benchmark rooted in real-world programming applications that leverage existing code repositories to perform tasks. Addressing the need for LLMs to interpret long code contexts and translate instructions into precise, executable scripts, ML-Bench encompasses annotated 9,641 examples across 18 GitHub repositories, challenging LLMs to accommodate user-specified arguments and documentation intricacies effectively. To evaluate both LLMs and AI agents, two setups are employed: ML-LLM-Bench for assessing LLMs' text-to-code conversion within a predefined deployment environment, and ML-Agent-Bench for testing autonomous agents in an end-to-end task execution within a Linux sandbox environment. Our findings indicate that while GPT-4o leads with a Pass@5 rate surpassing 50%, there remains significant scope for improvement, highlighted by issues such as hallucinated outputs and difficulties with bash script generation. Notably, in the more demanding ML-Agent-Bench, GPT-4o achieves a 76.47% success rate, reflecting the efficacy of iterative action and feedback in complex task resolution. Our code, dataset, and models are available at <a class="link-external link-https" href="https://github.com/gersteinlab/ML-bench" rel="external noopener nofollow">this https URL</a>.

MLE-bench: Evaluating Machine Learning Agents on Machine Learning Engineering

MLE-bench: Evaluating Machine Learning Agents on Machine Learning Engineering

Aibench: an industry standard ai benchmark suite

AIBench: an Industry Standard AI Benchmark Suite from Internet Services.

ML Research Benchmark

MLAgentBench: Evaluating Language Agents on Machine Learning Experimentation

MMAU: A Holistic Benchmark of Agent Capabilities Across Diverse Domains

AgentBench: Evaluating LLMs as Agents

BENCHAGENTS: Automated Benchmark Creation with Agent Interaction

ML-Bench: Evaluating Large Language Models and Agents for Machine Learning Tasks on Repository-Level Code

ML-Bench: Large Language Models Leverage Open-source Libraries for Machine Learning Tasks

BenchMARL: Benchmarking Multi-Agent Reinforcement Learning

PMLB: a large benchmark suite for machine learning evaluation and comparison

BLADE: Benchmarking Language Model Agents for Data-Driven Science

BetterBench: Assessing AI Benchmarks, Uncovering Issues, and Establishing Best Practices

The Benchmark Lottery

Scientific Machine Learning Benchmarks

PyBench: Evaluating LLM Agent on various real-world coding tasks

Benchmarking Robustness and Generalization in Multi-Agent Systems: A Case Study on Neural MMO

From Crowdsourced Data to High-Quality Benchmarks: Arena-Hard and BenchBuilder Pipeline

BattleAgentBench: A Benchmark for Evaluating Cooperation and Competition Capabilities of Language Models in Multi-Agent Systems