Skip to content
declare-labPublic

About

Agentic Quality-Diversity Search for Research Idea Generation

Resources

Stars

21 stars

Watchers

0 watching

Forks

Repository files navigation

IDEAgent: Quality-Diversity search for scientific ideation

📝 Changelog

  • 2026-10-20: Added verifiable/: IDEAgent on AlphaResearch and ALE-Bench, the four baselines under the same program condition, and the Yield evaluation under the three-judge diversity jury.

📖 Abstract

Large Language Models (LLMs) have significantly advanced the automation of scientific discovery. However, existing systems share a core limitation: they optimize ideas independently for either Quality or Diversity. This often results in closely related ideas or large sets of trivial, unsound, or unclear concepts. We instead argue that research ideation should be framed as a conjunction of Quality and Diversity objectives, i.e., a Quality-Diversity (QD) search. Accordingly, we introduce IDEAgent, a multi-agent framework that tracks idea evolution through lineages. We jointly drive Quality using multi-objective feedback for repair and refinement, while Diversity is achieved through lightweight sequential memory and explicit comparison against completed ideas, historical variants, and rejected proposals. To evaluate this joint objective, we develop Yield, a metric that computes the largest mutually diverse set of ideas satisfying predefined quality thresholds. Across 32 topics spanning 8 CS domains, IDEAgent outperforms the best baseline by 3.89× on Yield, while achieving Yield of at least 2 on 8× as many topics. We further analyze quality improvements, showing that repair and refinement improve logical rigor and clarity while preserving non-obviousness. Lastly, we extend IDEAgent to two algorithm discovery tasks for verifiable quality judgment and find that it still consistently outperforms all baselines, and yields the largest set of high-quality and diverse programs that beat a trivial/standard baseline.

🗂️ What Is Here

.
├── ideagent/          # runtime package: IDEAgent's search/repair/refinement
│                      # loop, quality + diversity evaluators, shared infra
├── baselines/
│   ├── simple/        # stateless-independent, single-shot batch, and
│   │                  # sequential-memory baselines
│   └── nova/          # NOVA-style closed-book baseline
├── verifiable/        # IDEAgent on verifiable domains (AlphaResearch,
│                      # ALE-Bench): ideas with programs that are run and scored
├── scripts/           # runners and evaluators (run_ideagent.py,
│                      # eval_ideas.py, select_lineage_representatives.py)
├── configs/           # canonical YAML configs for each pipeline
└── data/
    ├── bkgd_papers.jsonl  # topic input data
    └── examples/          # sample generations from IDEAgent and every baseline

⚙️ Setup

pip install -e .

Set the relevant API keys as environment variables or in a .env file (OPENAI_API_KEY, GEMINI_API_KEY, ANTHROPIC_API_KEY — see ideagent/clients.py).

🚀 Run IDEAgent (main system)

python scripts/run_ideagent.py --config configs/ideagent.yaml

🧪 Run a Baseline

# Stateless-independent or single-shot batch
python baselines/simple/run_simple_baseline.py --config baselines/simple/stateless.yaml
python baselines/simple/run_simple_baseline.py --config baselines/simple/single_shot.yaml

# Sequential-memory
python baselines/simple/sequential_memory.py --config baselines/simple/sequential_memory.yaml

# NOVA closed-book
python baselines/nova/run_nova_closed_book.py --config baselines/nova/nova_closed_book.yaml

📊 Evaluate

Runs both quality and diversity eval in one command:

python scripts/eval_ideas.py \
    --config configs/comp_quality_eval_opus47.yaml configs/diversity_eval_opus47.yaml \
    --input <final_idea_cores.jsonl> \
    --quality-output <quality_report.jsonl> \
    --diversity-output <diversity_report.jsonl> \
    --resume

Pass --skip-quality or --skip-diversity to run only one half. --start-idx/ --end-idx slice the topic list for sharding a run. --max-workers overrides both evaluators' concurrency.

🏆 Results

Open-ended ideation

Mean Yield across 32 topics spanning 8 CS domains. Yield(τ) is the size of the largest mutually-diverse (pairwise diversity ≥ 7) subset of an archive that is simultaneously sound (≥ 7), clear (≥ 6), and non-obvious (≥ τ); Yield(τ=7) is the primary metric reported in the paper.

Method Yield (τ=7, primary) Yield (τ=6)
Sequential-Memory 0.19 0.53
One-Shot 0.00 0.00
Stateless 0.25 0.81
NOVA 0.28 1.16
IDEAgent 1.09 2.31

IDEAgent outperforms the best baseline (NOVA) by 3.89× on Yield(τ=7).

Verifiable search

Means over the problems of each benchmark (9 for AlphaResearch, 7 for ALE-Bench). $Q_i$ is 100 × an idea's progress from the trivial baseline to the reference (the human-best for AlphaResearch, the contest winner for ALE-Bench); $Q_{\max}$ and $Q_{\text{avg}}$ are the highest and the mean $Q_i$ among a method's ten final ideas. $Y_k$ is the size of the largest set of mutually distinct ideas, under the three-judge jury, that beat the baseline with $Q_i \ge k$. See verifiable/README.md.

MethodAlphaResearchALE-Bench
QmaxQavgY0Y50Y75Y100QmaxQavgY0Y25Y50Y75
Sequential-Memory92.646.14.003.111.670.5638.318.62.291.711.140.29
One-Shot53.216.92.781.000.440.3314.86.21.140.430.430.00
Stateless93.547.84.113.111.780.5645.122.72.291.710.570.00
NOVA75.434.44.562.670.780.2237.318.02.292.001.000.00
IDEAgent106.254.55.674.443.110.6756.422.63.432.571.570.57

💡 Examples

data/examples/ holds sample generations from IDEAgent and every baseline, one JSONL file per method, over the same fixed set of topics for direct comparison.

🔬 Verifiable Search

verifiable/ runs IDEAgent where an idea's quality can be measured: each idea comes with the complete program that implements it, which is run and scored on AlphaResearch (9 mathematical construction problems) and ALE-Bench (7 AtCoder Heuristic Contest problems). Diversity is judged by a three-judge jury. See verifiable/README.md for the setup, runs and evaluation.

📄 Citation

@misc{gumma2026ideagentagenticqualitydiversitysearch,
      title={IDEAgent: Agentic Quality-Diversity Search for Research Idea Generation}, 
      author={Varun Gumma and Navonil Majumder and Soumitra Sinhahajari and Soujanya Poria},
      year={2026},
      eprint={2607.22375},
      archivePrefix={arXiv},
      primaryClass={cs.AI},
      url={https://arxiv.org/abs/2607.22375}, 
}

About

Agentic Quality-Diversity Search for Research Idea Generation

Resources

Stars

21 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages