Hindsight Experience Replay has become the ubiquitous method for goal-conditioned reinforcement learning, but leaves open the question of which goal to relabel with.
In this work, accepted at ICML, we propose instead simply Learning Everything All at Once (LEO).
1/
(๐งต) Happy to release AIRS-Bench, a benchmark to test the autonomous machine learning abilities of AI research agents ๐ค
AIRS-Bench includes 20 tasks sourced from machine learning papers that assess the autonomous research abilities of LLM agents throughout the full research lifecycle, from hypothesis generation ๐ก and implementation ๐ ๏ธ to experimentation ๐งช and analysis ๐
Each task is extracted from a paper with a state-of-the-art result and consists of a:
๐ problem description (e.g. text similarity)
๐๏ธ a dataset (e.g. SICK) and
๐ a metric (e.g. Spearman correlation) to optimise over
The agent is then given a GPU and 24 hours to develop and submit a Python solution that matches or exceeds the paper SOTA ๐
Read on for baseline results and examples of agents surpassing human SOTA ๐
๐ฑWe open-source the AIRS-Bench task definitions and evaluation code to accelerate in autonomous scientific research:
๐ป GitHub: https://t.co/UXzNXyGdU5
๐ ArXiv: https://t.co/badN0jq0IA
๐ค HF paper: https://t.co/6FIWxF0Bsw
๐ Meta AI website: https://t.co/wcIWLrlYBU
Huge shoutout to the team from Meta FAIR who painstakingly crafted, debugged and inspected every single of these tasks and its runs across more than a dozen of agents @alisia_lupidi, @_tomwithanh, @BhavulGauri, @basselralomari, @albertomariape, Alexis Audran-Reiss, Muna Aghamelu, Nicolas Baldwin, @LuciaCKun, @GagnonAudet, Chee Hau Leow, Sandra Lefdal, Abhinav Moudgil, Saba Nazir, Emanuel Tewolde, Isabel Urrego, @mahnerak, @ishitamed, @EdanToledo and @rybolos, @alex_h_miller, @j_foerst, @yorambac for their leadership and support
AIRS-Bench is out!
AIRS-Bench is a suite of 20 challenging ML tasks designed to evaluate LLM agents as AI Research Scientists spanning the full scientific method: from hypothesis generation and experimental design to result validation.
Paper: https://t.co/vW6IvccSkt
๐จTL;DR: Benchmarking for AI Scientists just got better!๐จ
Everyone is excited about AI Scientists, but we don't have a large scale benchmark that evaluates automated (or augmented) AI research systems on the home turf of the machine learning community: Machine Learning benchmarks.
Meet AIRS-Bench, our attempt at filling this gap. We hope AIRS-Bench will help the community to improve the signal-to-noise ratio in the era of research agents and is an important step towards turning ML benchmarks into standardised tasks for AI research agents. This has implications beyond AI scientists and will also help address the replication crisis in ML.
The team has invested countless hours (human and GPU) selecting/constructing the tasks, running baseline agents, analysing the outcomes, and hardening the benchmark.
We are excited for the community to both expand on our initial task set and benchmark new agentic systems!
Introducing - AIRS Bench, a benchmark for โAI Researcher Agentโ. Agents attempt 20 open ML problems starting from zero code (full research loop). And yes, they beat SOTA in few cases (read more below!) https://t.co/npx0JbRYPo