Skip to content
Β 
Β 

Latest commit

Β 

History

17 Commits

Folders and files

NameName
Last commit message
Last commit date
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

OpenDiscoveryTrace

Process Traces for Evaluating AI Scientist Workflows

Best Dataset Award Oral Presentation OpenReview Dataset on HF License: CC BY 4.0

πŸ† Best Dataset Award β€” ICML 2026 Workshop on AI for Science: AI Scientists – Tools, Co-authors, or Founders? (Dataset Proposal Competition)

🎀 Oral presentation β€” Dataset & AI Scientist Highlights, Block 1 Β· Saturday, July 11, 2026 Β· 10:30–10:50 KST Β· Hall C, ICML 2026

πŸ“„ Paper: openreview.net/forum?id=EHT3wVhCUZ

πŸ€— Dataset: huggingface.co/datasets/aayambansall/OpenDiscoveryTrace


Overview

Existing benchmarks for AI scientific agents evaluate only final outputs. OpenDiscoveryTrace captures the full reasoning process β€” every thought, tool call, error, revision, and confidence estimate β€” as models work through scientific tasks.

558 trajectories across 7 models (3 frontier, 4 open-weight), 124 tasks in 4 science domains (drug discovery, materials science, genomics, scientific literature analysis).

Headline Result

On 363 LLM-judged frontier trajectories, all three frontier models achieve comparable success rates (GPT-5.4: 88.6%, Gemini 3.1 Pro: 89.3%, Claude Opus 4.6: 83.9%), yet Claude Opus 4.6 makes 30Γ— more errors than GPT-5.4 (2.5 vs. 0.08 per trajectory, p < 10⁻⁴, Cliff's Ξ΄ = 0.613). The error profiles are qualitatively different: 66.7% tool misuse for Claude versus 83.6% reasoning errors for GPT-5.4. Output-only benchmarks miss this entirely.


Links

Resource Link
Paper (OpenReview) openreview.net/forum?id=EHT3wVhCUZ
Paper (PDF in this repo) paper/paper.pdf Β· full-length version
Full dataset (HuggingFace) huggingface.co/datasets/aayambansall/OpenDiscoveryTrace
Workshop AI Scientists – Tools, Co-authors, or Founders? @ ICML 2026
Workshop schedule ai4sciencecommunity.github.io/icml26/schedule
Dataset Proposal Competition ai4sciencecommunity.github.io/icml26/dataset
ICML virtual page icml.cc/virtual/2026/workshop/54099

Repository Structure

OpenDiscoveryTrace/
β”‚
β”œβ”€β”€ paper/                          # Submission
β”‚   β”œβ”€β”€ paper.pdf                   #   Competition proposal (2pp + appendix)
β”‚   β”œβ”€β”€ paper.tex                   #   LaTeX source
β”‚   β”œβ”€β”€ references.bib              #   Bibliography
β”‚   β”œβ”€β”€ table_comparison.tex        #   Benchmark comparison table
β”‚   β”œβ”€β”€ figures/                    #   Figures used in paper
β”‚   └── supplementary/              #   Full-length analysis paper
β”‚
β”œβ”€β”€ src/
β”‚   β”œβ”€β”€ harness/                    # Trajectory generation
β”‚   β”‚   β”œβ”€β”€ agent_harness.py        #   Main harness: runs models through tasks
β”‚   β”‚   └── run_opensource.py       #   Open-weight model runner (local GPU)
β”‚   β”œβ”€β”€ analysis/                   # Analysis pipelines
β”‚   β”‚   β”œβ”€β”€ analyze_trajectories.py #   Core stats + figures
β”‚   β”‚   └── reviewer_analysis.py    #   Extended analysis (taxonomy, baselines)
β”‚   └── baselines/                  # Benchmark task baselines
β”‚       └── implement_four.py       #   IAA, sequence models, live retrieval
β”‚
β”œβ”€β”€ data/
β”‚   β”œβ”€β”€ task_bank.json              # 200 scientific tasks (124 executed)
β”‚   └── samples/                    # Sample trajectories (full set on HuggingFace)
β”‚       β”œβ”€β”€ frontier/               #   3 tasks Γ— 3 frontier models
β”‚       └── open_weight/            #   1 task Γ— Qwen2.5-1.5B
β”‚
β”œβ”€β”€ results/
β”‚   β”œβ”€β”€ statistics/                 # Raw analysis outputs (JSON)
β”‚   β”‚   β”œβ”€β”€ analysis_results.json
β”‚   β”‚   β”œβ”€β”€ reviewer_analysis_results.json
β”‚   β”‚   └── four_additions_results.json
β”‚   └── figures/                    # All generated figures (PDF)
β”‚
β”œβ”€β”€ research_notes/                 # Research process documentation
β”‚   β”œβ”€β”€ literature-review.md        #   60+ papers across 5 facets
β”‚   β”œβ”€β”€ reasoning.md                #   Hypothesis deliberation
β”‚   β”œβ”€β”€ methodology.md              #   Pre-analysis plan
β”‚   └── synthesis.md                #   Results interpretation
β”‚
β”œβ”€β”€ assets/                         # Award certificate
β”œβ”€β”€ LICENSE                         # CC-BY-4.0
β”œβ”€β”€ requirements.txt
└── README.md

Quick Start

Browse a trajectory

import json

with open("data/samples/frontier/dd_e01_gpt-5.4.json") as f:
    traj = json.load(f)

print(f"Task:    {traj['prompt'][:80]}...")
print(f"Model:   {traj['model']}")
print(f"Steps:   {traj['metadata']['total_steps']}")
print(f"Errors:  {traj['metadata']['total_failures']}")
print(f"Claim:   {traj['outcome']['final_claim'][:120]}...")

Run analysis

pip install -r requirements.txt
python src/analysis/analyze_trajectories.py

Generate new trajectories

# Frontier models (requires OPENAI_API_KEY / ANTHROPIC_API_KEY / GEMINI_API_KEY)
python src/harness/agent_harness.py --model gpt-5.4 --max-tasks 10

# Open-weight models (requires GPU, no API keys)
python src/harness/run_opensource.py

Trace Schema

Each step in a trajectory records 9 fields:

Field Description
step_id Sequential step index
timestamp ISO 8601 UTC
phase Scientific workflow phase (literature review β†’ hypothesis β†’ experiment β†’ execution β†’ analysis β†’ conclusion)
thought Model's reasoning
action Tool call details (type, tool name, input, output)
observation Processed result of the action
error Error state (occurred, type, message)
revision_trigger What prompted a strategy change
confidence Self-reported certainty [0, 1]

Full JSON schema in paper/paper.tex Appendix A.


Models

Model Type Trajectories
GPT-5.4 Frontier 124
Claude Opus 4.6 Frontier 124
Gemini 3.1 Pro Frontier 124
Qwen2.5-7B-Instruct Open-weight (single-response) 30
Mistral-7B-v0.3 Open-weight (single-response) 30
Phi-3.5-mini-instruct Open-weight (single-response) 30
Qwen2.5-1.5B-Instruct Open-weight (single-response) 30
Qwen2.5-7B-Instruct Open-weight (tool-scaffolded, full multi-step harness) 6
Frontier models Live-retrieval variants (PubMed / PubChem) 60
Total 558

Frontier trajectories are fully balanced across 4 domains Γ— 3 difficulty levels.


Benchmark Tasks

  1. Trajectory Outcome Prediction β€” predict success from step features
  2. Error Localization β€” identify the step where reasoning went wrong
  3. Claim Verification β€” verify correctness of final claims
  4. Autonomy Classification β€” classify L1–L4 autonomy levels
  5. Process Quality Scoring β€” multi-axis trajectory quality

Baselines (logistic regression, random forest, LSTM, Transformer) in results/statistics/ and Appendix H of the paper.


Full Dataset

Sample trajectories are included in data/samples/. The complete dataset is hosted on HuggingFace:

huggingface.co/datasets/aayambansall/OpenDiscoveryTrace


License

Code, data, and paper are released under CC-BY-4.0. See LICENSE.


Citation

@inproceedings{bansal2026opendiscoverytrace,
  title     = {OpenDiscoveryTrace: Process Traces for Evaluating
               AI Scientist Workflows},
  author    = {Bansal, Aayam and Balaji, Keertan},
  booktitle = {ICML 2026 Workshop on AI for Science: AI Scientists --
               Tools, Co-authors, or Founders?},
  year      = {2026},
  url       = {https://openreview.net/forum?id=EHT3wVhCUZ},
  note      = {Best Dataset Award, Dataset Proposal Competition}
}

Best Dataset Award

Best Dataset Award for ICML 2026 AI for Science Workshop

Best Dataset Award, ICML 2026 AI for Science Workshop.

About

Process traces for evaluating AI scientist workflows | πŸ† ICML 2026 AI4Science Best Dataset

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages