Add paper link
Browse files
README.md
CHANGED
@@ -1,109 +1,112 @@
|
|
1 |
-
---
|
2 |
-
license: llama3.1
|
3 |
-
datasets:
|
4 |
-
- nvidia/OpenMathInstruct-2
|
5 |
-
language:
|
6 |
-
- en
|
7 |
-
base_model:
|
8 |
-
- meta-llama/Llama-3.1-8B-Instruct
|
9 |
-
model-index:
|
10 |
-
- name: Control-LLM-Llama3.1-8B-Math16
|
11 |
-
results:
|
12 |
-
- task:
|
13 |
-
type: math-evaluation
|
14 |
-
dataset:
|
15 |
-
type: parquet
|
16 |
-
name: Math, Math Hard, GSM8K
|
17 |
-
dataset_kwargs:
|
18 |
-
data_files: "https://github.com/linkedin/ControlLLM/blob/main/src/controlllm/inference/llm_eval_harness/additional_tasks/math/joined_math.parquet"
|
19 |
-
metrics:
|
20 |
-
- name: exact_match,none
|
21 |
-
type: exact_match
|
22 |
-
value: 0.6205678398534606
|
23 |
-
stderr: 0.005249520342473376
|
24 |
-
verified: false
|
25 |
-
- name: exact_match,none (gsm8k_0shot_instruct)
|
26 |
-
type: exact_match
|
27 |
-
value: 0.8968915845337376
|
28 |
-
stderr: 0.008376436987507811
|
29 |
-
verified: false
|
30 |
-
- name: exact_match,none (meta_math_0shot_instruct)
|
31 |
-
type: exact_match
|
32 |
-
value: 0.6166
|
33 |
-
stderr: 0.006876797660918556
|
34 |
-
verified: false
|
35 |
-
- name: exact_match,none (meta_math_hard_0shot_instruct)
|
36 |
-
type: exact_match
|
37 |
-
value: 0.36027190332326287
|
38 |
-
stderr: 0.013198755610252931
|
39 |
-
verified: false
|
40 |
-
- task:
|
41 |
-
type: original-capability
|
42 |
-
dataset:
|
43 |
-
type: meta/Llama-3.1-8B-Instruct-evals
|
44 |
-
name: Llama-3.1-8B-Instruct-evals Dataset
|
45 |
-
dataset_path: "meta-llama/llama-3.1-8_b-instruct-evals"
|
46 |
-
dataset_name: "Llama-3.1-8B-Instruct-evals__arc_challenge__details"
|
47 |
-
metrics:
|
48 |
-
- name: exact_match,strict-match
|
49 |
-
type: exact_match
|
50 |
-
value: 0.6001372485281902
|
51 |
-
stderr: 0.002821514831773572
|
52 |
-
verified: false
|
53 |
-
- name: exact_match,strict-match (meta_arc_0shot_instruct)
|
54 |
-
type: exact_match
|
55 |
-
value: 0.8248927038626609
|
56 |
-
stderr: 0.011139722235859526
|
57 |
-
verified: false
|
58 |
-
- name: exact_match,strict-match (meta_gpqa_0shot_cot_instruct)
|
59 |
-
type: exact_match
|
60 |
-
value: 0.3080357142857143
|
61 |
-
stderr: 0.021836780796366417
|
62 |
-
verified: false
|
63 |
-
- name: exact_match,strict-match (meta_mmlu_0shot_instruct)
|
64 |
-
type: exact_match
|
65 |
-
value: 0.7159948725252813
|
66 |
-
stderr: 0.00380556397209409
|
67 |
-
verified: false
|
68 |
-
- name: exact_match,strict-match (meta_mmlu_pro_5shot_instruct)
|
69 |
-
type: exact_match
|
70 |
-
value: 0.45403922872340424
|
71 |
-
stderr: 0.004539171007529716
|
72 |
-
verified: false
|
73 |
-
---
|
74 |
-
# Control-LLM-Llama3.1-8B-Math16
|
75 |
-
This is a fine-tuned model of Llama-3.1-8B-Instruct for mathematical tasks on OpenMath2 dataset.
|
76 |
-
|
77 |
-
##
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
| **
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
- **
|
94 |
-
- **
|
95 |
-
- **
|
96 |
-
- **
|
97 |
-
- **
|
98 |
-
- **
|
99 |
-
- **
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: llama3.1
|
3 |
+
datasets:
|
4 |
+
- nvidia/OpenMathInstruct-2
|
5 |
+
language:
|
6 |
+
- en
|
7 |
+
base_model:
|
8 |
+
- meta-llama/Llama-3.1-8B-Instruct
|
9 |
+
model-index:
|
10 |
+
- name: Control-LLM-Llama3.1-8B-Math16
|
11 |
+
results:
|
12 |
+
- task:
|
13 |
+
type: math-evaluation
|
14 |
+
dataset:
|
15 |
+
type: parquet
|
16 |
+
name: Math, Math Hard, GSM8K
|
17 |
+
dataset_kwargs:
|
18 |
+
data_files: "https://github.com/linkedin/ControlLLM/blob/main/src/controlllm/inference/llm_eval_harness/additional_tasks/math/joined_math.parquet"
|
19 |
+
metrics:
|
20 |
+
- name: exact_match,none
|
21 |
+
type: exact_match
|
22 |
+
value: 0.6205678398534606
|
23 |
+
stderr: 0.005249520342473376
|
24 |
+
verified: false
|
25 |
+
- name: exact_match,none (gsm8k_0shot_instruct)
|
26 |
+
type: exact_match
|
27 |
+
value: 0.8968915845337376
|
28 |
+
stderr: 0.008376436987507811
|
29 |
+
verified: false
|
30 |
+
- name: exact_match,none (meta_math_0shot_instruct)
|
31 |
+
type: exact_match
|
32 |
+
value: 0.6166
|
33 |
+
stderr: 0.006876797660918556
|
34 |
+
verified: false
|
35 |
+
- name: exact_match,none (meta_math_hard_0shot_instruct)
|
36 |
+
type: exact_match
|
37 |
+
value: 0.36027190332326287
|
38 |
+
stderr: 0.013198755610252931
|
39 |
+
verified: false
|
40 |
+
- task:
|
41 |
+
type: original-capability
|
42 |
+
dataset:
|
43 |
+
type: meta/Llama-3.1-8B-Instruct-evals
|
44 |
+
name: Llama-3.1-8B-Instruct-evals Dataset
|
45 |
+
dataset_path: "meta-llama/llama-3.1-8_b-instruct-evals"
|
46 |
+
dataset_name: "Llama-3.1-8B-Instruct-evals__arc_challenge__details"
|
47 |
+
metrics:
|
48 |
+
- name: exact_match,strict-match
|
49 |
+
type: exact_match
|
50 |
+
value: 0.6001372485281902
|
51 |
+
stderr: 0.002821514831773572
|
52 |
+
verified: false
|
53 |
+
- name: exact_match,strict-match (meta_arc_0shot_instruct)
|
54 |
+
type: exact_match
|
55 |
+
value: 0.8248927038626609
|
56 |
+
stderr: 0.011139722235859526
|
57 |
+
verified: false
|
58 |
+
- name: exact_match,strict-match (meta_gpqa_0shot_cot_instruct)
|
59 |
+
type: exact_match
|
60 |
+
value: 0.3080357142857143
|
61 |
+
stderr: 0.021836780796366417
|
62 |
+
verified: false
|
63 |
+
- name: exact_match,strict-match (meta_mmlu_0shot_instruct)
|
64 |
+
type: exact_match
|
65 |
+
value: 0.7159948725252813
|
66 |
+
stderr: 0.00380556397209409
|
67 |
+
verified: false
|
68 |
+
- name: exact_match,strict-match (meta_mmlu_pro_5shot_instruct)
|
69 |
+
type: exact_match
|
70 |
+
value: 0.45403922872340424
|
71 |
+
stderr: 0.004539171007529716
|
72 |
+
verified: false
|
73 |
+
---
|
74 |
+
# Control-LLM-Llama3.1-8B-Math16
|
75 |
+
This is a fine-tuned model of Llama-3.1-8B-Instruct for mathematical tasks on OpenMath2 dataset.
|
76 |
+
|
77 |
+
## Linked Paper
|
78 |
+
This model is associated with the paper: [Control-LLM](https://arxiv.org/abs/2501.10979).
|
79 |
+
|
80 |
+
## Evaluation Results
|
81 |
+
Here is an overview of the evaluation results and findings:
|
82 |
+
|
83 |
+
### Benchmark Results Table
|
84 |
+
The table below summarizes evaluation results across mathematical tasks and original capabilities.
|
85 |
+
|
86 |
+
| **Model** | **MH** | **M** | **G8K** | **M-Avg** | **ARC** | **GPQA** | **MLU** | **MLUP** | **O-Avg** | **Overall** |
|
87 |
+
|-------------------|--------|--------|---------|-----------|---------|----------|---------|----------|-----------|-------------|
|
88 |
+
| Llama3.1-8B-Inst | 23.7 | 50.9 | 85.6 | 52.1 | 83.4 | 29.9 | 72.4 | 46.7 | 60.5 | 56.3 |
|
89 |
+
| **Control LLM*** | 36.0 | 61.7 | **89.7**| 62.5 | 82.5 | 30.8 | **71.6**| 45.4 | **57.6** | **60.0** |
|
90 |
+
|
91 |
+
---
|
92 |
+
### Explanation:
|
93 |
+
- **MH**: MathHard
|
94 |
+
- **M**: Math
|
95 |
+
- **G8K**: GSM8K
|
96 |
+
- **M-Avg**: Math - Average across MathHard, Math, and GSM8K
|
97 |
+
- **ARC**: ARC benchmark
|
98 |
+
- **GPQA**: General knowledge QA
|
99 |
+
- **MLU**: MMLU (Massive Multitask Language Understanding)
|
100 |
+
- **MLUP**: MMLU Pro
|
101 |
+
- **O-Avg**: Original Capability - Average across ARC, GPQA, MMLU, and MLUP
|
102 |
+
- **Overall**: Combined average across all tasks
|
103 |
+
|
104 |
+
### Catastrophic Forgetting on OpenMath
|
105 |
+
The following plot illustrates and compares catastrophic forgetting mitigation during training
|
106 |
+
|
107 |
+

|
108 |
+
|
109 |
+
### Alignment Result
|
110 |
+
The plot below highlights the alignment result of the model trained with Control LLM.
|
111 |
+
|
112 |
+

|