Update README with verified ScienceAccelBench results

#2
Files changed (1) hide show
  1. README.md +11 -26
README.md CHANGED
@@ -25,15 +25,15 @@ The training data is sourced from [AItonomy/ScienceIDE](https://huggingface.co/d
25
 
26
  ## Quick start
27
 
28
- Use Transformers 5.16.1, PyTorch and Accelerate. The example loads the model, configuration and tokenizer directly from this repository. Sign in with `hf auth login` to access this private repository.
29
 
30
  ```python
31
  from transformers import AutoTokenizer, AutoModelForImageTextToText
32
 
33
  model_id = "AItonomy/PhAI-IDE-9B"
34
- tokenizer = AutoTokenizer.from_pretrained(model_id, token=True)
35
  model = AutoModelForImageTextToText.from_pretrained(
36
- model_id, dtype="bfloat16", device_map="auto", token=True,
37
  )
38
  inputs = tokenizer.apply_chat_template(
39
  [{"role": "user", "content": "Explain how to verify a numerical simulation."}],
@@ -45,32 +45,17 @@ print(tokenizer.decode(output[0, inputs["input_ids"].shape[1]:], skip_special_to
45
 
46
  BF16 weights occupy approximately 18.82 GB, plus runtime memory and KV cache.
47
 
48
- ## Benchmark performance
49
 
50
- ### Score improvements
51
 
52
- Scores (%); gains in percentage points.
53
 
54
- | Benchmark | Qwen/Qwen3.5-9B | PhAI-IDE-9B | Gain (pp) |
55
- | --- | ---: | ---: | ---: |
56
- | BBH word-sorting | 25.60 | **60.40** | **+34.80** |
57
- | AQuA-RAT | 86.61 | **88.98** | **+2.36** |
58
- | MATH-500 | 91.40 | **92.20** | **+0.80** |
59
- | OpenBookQA (128-sample subset) | 33.59 | **35.16** | **+1.56** |
60
- | GSM8K (128-sample subset) | 89.84 | **92.97** | **+3.12** |
61
-
62
- ### Comparison with published models
63
-
64
- Scores (%); evaluation settings vary by source.
65
-
66
- | Benchmark | Reference model | Parameters | Published score | PhAI-IDE-9B result | Source |
67
- | --- | --- | ---: | ---: | ---: | --- |
68
- | GSM8K | Llama-3.1-8B-Instruct | 8B | 84.5 | **92.97** | [Meta model card](https://huggingface.co/meta-llama/Llama-3.1-8B-Instruct#instruction-tuned-models) |
69
- | GSM8K | Qwen2.5-7B-Instruct | 7B | 91.6 | **92.97** | [Qwen2.5 report, Table 8](https://arxiv.org/html/2412.15115v2#S5.SS2.SSS1) |
70
- | GSM8K | Gemma-2-9B-IT | 9B | 76.7 | **92.97** | [Qwen2.5 report, Table 8](https://arxiv.org/html/2412.15115v2#S5.SS2.SSS1) |
71
- | GSM8K | SciTulu-7B | 7B | 29.5 | **92.97** | [SciRIFF report, Table 7](https://arxiv.org/html/2406.07835v2#A3) |
72
- | BBH word-sorting | Llama-3.1-8B-Instruct | 8.03B | 51.2 | **60.40** | [BenchLM independent results](https://huggingface.co/spaces/steampunque/benchlm/blob/d45e8600172857935610426f797a4429f2f136d6/README.md) |
73
- | BBH word-sorting | Qwen2.5-7B-Instruct | 7.62B | 15.6 | **60.40** | [BenchLM independent results](https://huggingface.co/spaces/steampunque/benchlm/blob/d45e8600172857935610426f797a4429f2f136d6/README.md) |
74
 
75
  ## Training procedure
76
 
 
25
 
26
  ## Quick start
27
 
28
+ Use Transformers 5.16.1, PyTorch and Accelerate. The example loads the model, configuration and tokenizer directly from this repository.
29
 
30
  ```python
31
  from transformers import AutoTokenizer, AutoModelForImageTextToText
32
 
33
  model_id = "AItonomy/PhAI-IDE-9B"
34
+ tokenizer = AutoTokenizer.from_pretrained(model_id)
35
  model = AutoModelForImageTextToText.from_pretrained(
36
+ model_id, dtype="bfloat16", device_map="auto",
37
  )
38
  inputs = tokenizer.apply_chat_template(
39
  [{"role": "user", "content": "Explain how to verify a numerical simulation."}],
 
45
 
46
  BF16 weights occupy approximately 18.82 GB, plus runtime memory and KV cache.
47
 
48
+ ## ScienceAccelBench performance
49
 
50
+ Across all 27 ScienceAccelBench environments, the task-pooled pass rate increases from **17.09% to 20.89% (+3.80 percentage points)** on **158 valid paired tasks**.
51
 
52
+ Task-held-out, localized scientific-code repair on familiar codebases, with original numerical verification. Pass rates (%); gains in percentage points.
53
 
54
+ | Environment | Tasks | Qwen3.5-9B | PhAI-IDE-9B | Gain (pp) |
55
+ | --- | ---: | ---: | ---: | ---: |
56
+ | laps | 16 | 31.25 | **50.00** | **+18.75** |
57
+ | mitgcm-biogeo | 8 | 0.00 | **12.50** | **+12.50** |
58
+ | pluto-rmhd-resrmhd | 7 | 0.00 | **28.57** | **+28.57** |
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
59
 
60
  ## Training procedure
61