README

The AI Scientist-v2: Workshop-Level Automated
    Scientific Discovery via Agentic Tree Search
  



  📚 [Paper] |
  📝  [Blog Post] |
  📂  [ICLR2025 Workshop Experiment]


Fully autonomous scientific research systems are becoming increasingly capable, with AI playing a pivotal role in transforming how scientific discoveries are made.
We are excited to introduce The AI Scientist-v2, a generalized end-to-end agentic system that has generated the first workshop paper written entirely by AI and accepted through peer review.

This system autonomously generates hypotheses, runs experiments, analyzes data, and writes scientific manuscripts. Unlike its predecessor (AI Scientist-v1), the AI Scientist-v2 removes reliance on human-authored templates, generalizes across Machine Learning (ML) domains, and employs a progressive agentic tree search, guided by an experiment manager agent.

 Note:
 The AI Scientist-v2 doesn’t necessarily produce better papers than v1, especially when a strong starting template is available. v1 follows well-defined templates, leading to high success rates, while v2 takes a broader, more exploratory approach with lower success rates. v1 works best for tasks with clear objectives and a solid foundation, whereas v2 is designed for open-ended scientific exploration.

 Caution!
 This codebase will execute Large Language Model (LLM)-written code. There are various risks and challenges associated with this autonomy, including the potential use of dangerous packages, uncontrolled web access, and the possibility of spawning unintended processes. Ensure that you run this within a controlled sandbox environment (e.g., a Docker container). Use at your own discretion.

 Table of Contents

1.  Requirements
       Installation
       Supported Models and API Keys
2.  Generate Research Ideas
3.  Run AI Scientist-v2 Paper Generation Experiments
4.  Citing The AI Scientist-v2
5.  Frequently Asked Questions
6.  Acknowledgement

 Requirements

This code is designed to run on Linux with NVIDIA GPUs using CUDA and PyTorch.

 Installation

bash
 Create a new conda environment
conda create -n ai_scientist python=3.11
conda activate ai_scientist

 Install PyTorch with CUDA support (adjust pytorch-cuda version for your setup)
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

 Install PDF and LaTeX tools
conda install anaconda::poppler
conda install conda-forge::chktex

 Install Python package requirements
pip install -r requirements.txt


Installation usually takes no more than one hour.

 Supported Models and API Keys

 OpenAI Models

By default, the system uses the OPENAI_API_KEY environment variable for OpenAI models.

 Gemini Models

By default, the system uses the GEMINI_API_KEY environment variable for Gemini models through OpenAI API.

 Claude Models via AWS Bedrock

To use Claude models provided by Amazon Bedrock, install the necessary additional packages:
bash
pip install anthropic[bedrock]

Next, configure valid