Muennighoff
commited on
Commit
•
d7cbab7
1
Parent(s):
f6552b9
Dates instead of 1.7
Browse files
README.md
CHANGED
@@ -113,7 +113,7 @@ For more documentation, see the [GitHub readme](https://github.com/allenai/OLMo?
|
|
113 |
|
114 |
Core model results for the new and original 7B model are found below.
|
115 |
|
116 |
-
| Task | Llama-7b | Llama2-7b | Falcon-7b | Mpt-7b | OLMo-7B | Llama2-13b | **OLMo
|
117 |
|-------------------|----------|-----------|-----------|--------|---------|------------|-------------|
|
118 |
| arc_c | 44.5 | 48.5 | 47.5 | 46.5 | 48.5 | 52.8 | 42.5 |
|
119 |
| arc_e | 67.9 | 69.5 | 70.4 | 70.5 | 65.4 | 73.7 | 67.2 |
|
@@ -131,7 +131,7 @@ Core model results for the new and original 7B model are found below.
|
|
131 |
|
132 |
And for the 1B model:
|
133 |
|
134 |
-
| task | random | [StableLM 2 1.6b](https://huggingface.co/stabilityai/stablelm-2-1_6b)\* | [Pythia 1B](https://huggingface.co/EleutherAI/pythia-1b) | [TinyLlama 1.1B](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1195k-token-2.5T) | OLMo 1B | **OLMo
|
135 |
| ------------- | ------ | ----------------- | --------- | -------------------------------------- | ------- | ---- |
|
136 |
| arc_challenge | 25 | 43.8 | 33.1 | 34.8 | 34.5 | 36.5 |
|
137 |
| arc_easy | 25 | 63.7 | 50.2 | 53.2 | 58.1 | 55.3 |
|
@@ -155,10 +155,10 @@ During the annealing phase we use a higher quality subset of Dolma with a linear
|
|
155 |
|
156 |
### Staged training / annealing
|
157 |
|
158 |
-
In contrast to OLMo
|
159 |
* In the first stage, we trained the model from scratch on the Dolma 1.7 dataset. We set a cosine learning rate schedule with a warmup of 2500 steps, a peak learning rate of 3e-4, and a cosine decay to 3e-5 after 3T tokens. We cut off this stage after 2T tokens, when the learning rate is still high.
|
160 |
* At this point we switch to the second stage, in which we train on a higher-quality subset of Dolma 1.7 (see below) for another 50B tokens, while linearly decaying the learning rate to 0. Our high-quality subset includes (1) using all available Wikipedia, OpenWebMath and Flan data, (2) removing Dolma CC, CC News, and Megawika, and (3) rebalancing remaining sources to achieve approximately equal proportions of each. See exact token counts and relative proportions of this second stage mix below.
|
161 |
-
Both stages contribute equally to the final performance of the OLMo model. After the first stage, OLMo
|
162 |
|
163 |
|
164 |
### Architecture
|
|
|
113 |
|
114 |
Core model results for the new and original 7B model are found below.
|
115 |
|
116 |
+
| Task | Llama-7b | Llama2-7b | Falcon-7b | Mpt-7b | OLMo-7B | Llama2-13b | **OLMo 7B 0424** |
|
117 |
|-------------------|----------|-----------|-----------|--------|---------|------------|-------------|
|
118 |
| arc_c | 44.5 | 48.5 | 47.5 | 46.5 | 48.5 | 52.8 | 42.5 |
|
119 |
| arc_e | 67.9 | 69.5 | 70.4 | 70.5 | 65.4 | 73.7 | 67.2 |
|
|
|
131 |
|
132 |
And for the 1B model:
|
133 |
|
134 |
+
| task | random | [StableLM 2 1.6b](https://huggingface.co/stabilityai/stablelm-2-1_6b)\* | [Pythia 1B](https://huggingface.co/EleutherAI/pythia-1b) | [TinyLlama 1.1B](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1195k-token-2.5T) | OLMo 1B | **OLMo 1B 0724** (ours) |
|
135 |
| ------------- | ------ | ----------------- | --------- | -------------------------------------- | ------- | ---- |
|
136 |
| arc_challenge | 25 | 43.8 | 33.1 | 34.8 | 34.5 | 36.5 |
|
137 |
| arc_easy | 25 | 63.7 | 50.2 | 53.2 | 58.1 | 55.3 |
|
|
|
155 |
|
156 |
### Staged training / annealing
|
157 |
|
158 |
+
In contrast to the first OLMo, we trained OLMo 7B 0424 with a two-stage curriculum:
|
159 |
* In the first stage, we trained the model from scratch on the Dolma 1.7 dataset. We set a cosine learning rate schedule with a warmup of 2500 steps, a peak learning rate of 3e-4, and a cosine decay to 3e-5 after 3T tokens. We cut off this stage after 2T tokens, when the learning rate is still high.
|
160 |
* At this point we switch to the second stage, in which we train on a higher-quality subset of Dolma 1.7 (see below) for another 50B tokens, while linearly decaying the learning rate to 0. Our high-quality subset includes (1) using all available Wikipedia, OpenWebMath and Flan data, (2) removing Dolma CC, CC News, and Megawika, and (3) rebalancing remaining sources to achieve approximately equal proportions of each. See exact token counts and relative proportions of this second stage mix below.
|
161 |
+
Both stages contribute equally to the final performance of the OLMo model. After the first stage, OLMo 7B 0424 already outperforms the older OLMo. The second stage consistently adds 2 to 3 points of performance on top.
|
162 |
|
163 |
|
164 |
### Architecture
|