Commit
•
201b7bd
1
Parent(s):
eff57d0
Update README.md
Browse files
README.md
CHANGED
|
@@ -10,4 +10,41 @@ language:
|
|
| 10 |
|
| 11 |
The extrapolated (ExPO) model based on [`internlm2-chat-20b`](https://huggingface.co/internlm/internlm2-chat-20b) and [`internlm/internlm2-chat-20b-sft`](https://huggingface.co/internlm/internlm2-chat-20b-sft), as in the "[Weak-to-Strong Extrapolation Expedites Alignment](https://arxiv.org/abs/2404.16792)" paper.
|
| 12 |
|
| 13 |
-
Specifically, we obtain this model by extrapolating **(alpha = 0.5)** from the weights of the SFT and DPO/RLHF checkpoints, achieving superior alignment with human preference.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
The extrapolated (ExPO) model based on [`internlm2-chat-20b`](https://huggingface.co/internlm/internlm2-chat-20b) and [`internlm/internlm2-chat-20b-sft`](https://huggingface.co/internlm/internlm2-chat-20b-sft), as in the "[Weak-to-Strong Extrapolation Expedites Alignment](https://arxiv.org/abs/2404.16792)" paper.
|
| 12 |
|
| 13 |
+
Specifically, we obtain this model by extrapolating **(alpha = 0.5)** from the weights of the SFT and DPO/RLHF checkpoints, achieving superior alignment with human preference.
|
| 14 |
+
|
| 15 |
+
## Evaluation Results
|
| 16 |
+
|
| 17 |
+
Evaluation results on the **AlpacaEval 2.0** benchmark (you can find the evaluation outputs on the [official GitHub repo](https://github.com/chujiezheng/LLM-Extrapolation/tree/main/results_alpaca)):
|
| 18 |
+
|
| 19 |
+
| | Win Rate (Ori) | LC Win Rate (Ori) | Win Rate (+ ExPO) | LC Win Rate (+ ExPO) |
|
| 20 |
+
| ------------------------------------ | -------------- | ----------------- | ----------------- | -------------------- |
|
| 21 |
+
| `HuggingFaceH4/zephyr-7b-alpha` | 6.7% | 10.0% | **10.6%** | **13.6%** |
|
| 22 |
+
| `HuggingFaceH4/zephyr-7b-beta` | 10.2% | 13.2% | **11.1%** | **14.0%** |
|
| 23 |
+
| `berkeley-nest/Starling-LM-7B-alpha` | 15.0% | 18.3% | **18.2%** | **19.5%** |
|
| 24 |
+
| `Nexusflow/Starling-LM-7B-beta` | 26.6% | 25.8% | **29.6%** | **26.4%** |
|
| 25 |
+
| `snorkelai/Snorkel-Mistral-PairRM` | 24.7% | 24.0% | **28.8%** | **26.4%** |
|
| 26 |
+
| `RLHFlow/LLaMA3-iterative-DPO-final` | 29.2% | 36.0% | **32.7%** | **37.8%** |
|
| 27 |
+
| `internlm/internlm2-chat-1.8b` | 3.8% | 4.0% | **5.2%** | **4.3%** |
|
| 28 |
+
| `internlm/internlm2-chat-7b` | 20.5% | 18.3% | **28.1%** | **22.7%** |
|
| 29 |
+
| `internlm/internlm2-chat-20b` | 36.1% | 24.9% | **46.2%** | **27.2%** |
|
| 30 |
+
| `allenai/tulu-2-dpo-7b` | 8.5% | 10.2% | **11.5%** | **11.7%** |
|
| 31 |
+
| `allenai/tulu-2-dpo-13b` | 11.2% | 15.5% | **15.6%** | **17.6%** |
|
| 32 |
+
| `allenai/tulu-2-dpo-70b` | 15.4% | 21.2% | **23.0%** | **25.7%** |
|
| 33 |
+
|
| 34 |
+
Evaluation results on the **MT-Bench** benchmark (you can find the evaluation outputs on the [official GitHub repo](https://github.com/chujiezheng/LLM-Extrapolation/tree/main/results_mtbench)):
|
| 35 |
+
|
| 36 |
+
| | Original | + ExPO |
|
| 37 |
+
| ------------------------------------ | -------- | -------- |
|
| 38 |
+
| `HuggingFaceH4/zephyr-7b-alpha` | 6.85 | **6.87** |
|
| 39 |
+
| `HuggingFaceH4/zephyr-7b-beta` | 7.02 | **7.06** |
|
| 40 |
+
| `berkeley-nest/Starling-LM-7B-alpha` | 7.82 | **7.91** |
|
| 41 |
+
| `Nexusflow/Starling-LM-7B-beta` | 8.10 | **8.18** |
|
| 42 |
+
| `snorkelai/Snorkel-Mistral-PairRM` | 7.63 | **7.69** |
|
| 43 |
+
| `RLHFlow/LLaMA3-iterative-DPO-final` | 8.08 | **8.45** |
|
| 44 |
+
| `internlm/internlm2-chat-1.8b` | 5.17 | **5.26** |
|
| 45 |
+
| `internlm/internlm2-chat-7b` | 7.72 | **7.80** |
|
| 46 |
+
| `internlm/internlm2-chat-20b` | 8.13 | **8.26** |
|
| 47 |
+
| `allenai/tulu-2-dpo-7b` | 6.35 | **6.38** |
|
| 48 |
+
| `allenai/tulu-2-dpo-13b` | 7.00 | **7.26** |
|
| 49 |
+
| `allenai/tulu-2-dpo-70b` | 7.79 | **8.03** |
|
| 50 |
+
|