chrisociepa commited on
Commit
9f30543
1 Parent(s): 5605e98

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +217 -0
README.md CHANGED
@@ -1,3 +1,220 @@
1
  ---
2
  license: cc-by-nc-4.0
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: cc-by-nc-4.0
3
+ language:
4
+ - pl
5
+ library_name: transformers
6
+ tags:
7
+ - finetuned
8
+ inference:
9
+ parameters:
10
+ temperature: 0.6
11
+ widget:
12
+ - messages:
13
+ - role: user
14
+ content: Co przedstawia polskie godło?
15
  ---
16
+
17
+ <p align="center">
18
+ <img src="https://huggingface.co/speakleash/Bielik-7B-Instruct-v0.1/raw/main/speakleash_cyfronet.png">
19
+ </p>
20
+
21
+ # Bielik-7B-Instruct-v0.1
22
+
23
+ The Bielik-7B-Instruct-v0.1 is an instruct fine-tuned version of the [Bielik-7B-v0.1](https://huggingface.co/speakleash/Bielik-7B-v0.1). Forementioned model stands as a testament to the unique collaboration between the open-science/open-souce project SpeakLeash and the High Performance Computing (HPC) center: ACK Cyfronet AGH. Developed and trained on Polish text corpora, which has been cherry-picked and processed by the SpeakLeash team, this endeavor leverages Polish large-scale computing infrastructure, specifically within the PLGrid environment, and more precisely, the HPC centers: ACK Cyfronet AGH. The creation and training of the Bielik-7B-Instruct-v0.1 was propelled by the support of computational grant number PLG/2024/016951, conducted on the Helios supercomputer, enabling the use of cutting-edge technology and computational resources essential for large-scale machine learning processes. As a result, the model exhibits an exceptional ability to understand and process the Polish language, providing accurate responses and performing a variety of linguistic tasks with high precision.
24
+
25
+ ## Model
26
+
27
+ The [SpeakLeash](https://speakleash.org/) team is working on their own set of instructions in Polish, which is continuously being expanded and refined by annotators. A portion of these instructions, which had been manually verified and corrected, has been utilized for training purposes. Moreover, due to the limited availability of high-quality instructions in Polish, publicly accessible collections of instructions in English were used - [OpenHermes-2.5](https://huggingface.co/datasets/teknium/OpenHermes-2.5) and [orca-math-word-problems-200k](https://huggingface.co/datasets/microsoft/orca-math-word-problems-200k), which accounted for half of the instructions used in training. The instructions varied in quality, leading to a deterioration in model’s performance. To counteract this while still allowing ourselves to utilize forementioned datasets,several improvements were introduced:
28
+ * Weighted tokens level loss - a strategy inspired by [offline reinforcement learning](https://arxiv.org/abs/2005.01643) and [C-RLFT](https://arxiv.org/abs/2309.11235)
29
+ * Adaptive learning rate inspired by the study on [Learning Rates as a Function of Batch Size](https://arxiv.org/abs/2006.09092)
30
+ * Masked user instructions
31
+
32
+ Bielik-7B-Instruct-v0.1 has been trained with the use of an original open source framework called [ALLaMo](https://github.com/chrisociepa/allamo) implemented by [Krzysztof Ociepa](https://www.linkedin.com/in/krzysztof-ociepa-44886550/). This framework allows users to train language models with architecture similar to LLaMA and Mistral in fast and efficient way.
33
+
34
+ ### Model description:
35
+
36
+ * **Developed by:** [SpeakLeash](https://speakleash.org/)
37
+ * **Language:** Polish
38
+ * **Model type:** causal decoder-only
39
+ * **Finetuned from:** [Bielik-7B-v0.1](https://huggingface.co/speakleash/Bielik-7B-v0.1)
40
+ * **License:** CC BY NC 4.0 (non-commercial use)
41
+ * **Model ref:** speakleash:e38140bea0d48f1218540800bbc67e89
42
+
43
+ ## Training
44
+
45
+ * Framework: [ALLaMo](https://github.com/chrisociepa/allamo)
46
+ * Visualizations: [W&B](https://wandb.ai)
47
+
48
+ <p align="center">
49
+ <img src="https://huggingface.co/speakleash/Bielik-7B-Instruct-v0.1/raw/main/sft_train_loss.png">
50
+ </p>
51
+ <p align="center">
52
+ <img src="https://huggingface.co/speakleash/Bielik-7B-Instruct-v0.1/raw/main/sft_train_ppl.png">
53
+ </p>
54
+ <p align="center">
55
+ <img src="https://huggingface.co/speakleash/Bielik-7B-Instruct-v0.1/raw/main/sft_train_lr.png">
56
+ </p>
57
+
58
+ ### Training hyperparameters:
59
+
60
+ | **Hyperparameter** | **Value** |
61
+ |-----------------------------|------------------|
62
+ | Micro Batch Size | 1 |
63
+ | Batch Size | up to 4194304 |
64
+ | Learning Rate (cosine, adaptive) | 7e-6 -> 6e-7 |
65
+ | Warmup Iterations | 50 |
66
+ | All Iterations | 55440 |
67
+ | Optimizer | AdamW |
68
+ | β1, β2 | 0.9, 0.95 |
69
+ | Adam_eps | 1e−8 |
70
+ | Weight Decay | 0.05 |
71
+ | Grad Clip | 1.0 |
72
+ | Precision | bfloat16 (mixed) |
73
+
74
+
75
+ ### Instruction format
76
+
77
+ In order to leverage instruction fine-tuning, your prompt should be surrounded by `[INST]` and `[/INST]` tokens. The very first instruction should start with the beginning of a sentence token. The generated completion will be finished by the end-of-sentence token.
78
+
79
+ E.g.
80
+ ```
81
+ prompt = "<s>[INST] Jakie mamy pory roku? [/INST]"
82
+ completion = "W Polsce mamy 4 pory roku: wiosna, lato, jesień i zima.</s>"
83
+ ```
84
+
85
+ This format is available as a [chat template](https://huggingface.co/docs/transformers/main/chat_templating) via the `apply_chat_template()` method:
86
+
87
+ ```python
88
+ from transformers import AutoModelForCausalLM, AutoTokenizer
89
+
90
+ device = "cuda" # the device to load the model onto
91
+
92
+ model_name = "speakleash/Bielik-7B-Instruct-v0.1"
93
+
94
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
95
+ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16)
96
+
97
+ messages = [
98
+ {"role": "user", "content": "Jakie mamy pory roku w Polsce?"},
99
+ {"role": "assistant", "content": "W Polsce mamy 4 pory roku: wiosna, lato, jesień i zima."},
100
+ {"role": "user", "content": "Która jest najcieplejsza?"}
101
+ ]
102
+
103
+ input_ids = tokenizer.apply_chat_template(messages, return_tensors="pt")
104
+
105
+ model_inputs = input_ids.to(device)
106
+ model.to(device)
107
+
108
+ generated_ids = model.generate(model_inputs, max_new_tokens=1000, do_sample=True)
109
+ decoded = tokenizer.batch_decode(generated_ids)
110
+ print(decoded[0])
111
+ ```
112
+
113
+ ## Evaluation
114
+
115
+
116
+ Models have been evaluated on [Open PL LLM Leaderboard](https://huggingface.co/spaces/speakleash/open_pl_llm_leaderboard) 5-shot. The benchmark evaluates models in NLP tasks like sentiment analysis, categorization, text classification but does not test chatting skills. Here are presented:
117
+ - Average - average score among all tasks normalized by baseline scores
118
+ - Reranking - reranking task, commonly used in RAG
119
+ - Reader (Generator) - open book question answering task, commonly used in RAG
120
+ - Perplexity (lower is better) - as a bonus, does not correlate with other scores and should not be used for model comparison
121
+
122
+
123
+
124
+ | | Average | RAG Reranking | RAG Reader | Perplexity |
125
+ |--------------------------------------------------------------------------------------|----------:|--------------:|-----------:|-----------:|
126
+ | **7B parameters models:** | | | | |
127
+ | Baseline (majority class) | 0.00 | 53.36 | - | - |
128
+ | Voicelab/trurl-2-7b | 18.85 | 60.67 | 77.19 | 1098.88 |
129
+ | meta-llama/Llama-2-7b-chat-hf | 21.04 | 54.65 | 72.93 | 4018.74 |
130
+ | mistralai/Mistral-7B-Instruct-v0.1 | 26.42 | 56.35 | 73.68 | 6909.94 |
131
+ | szymonrucinski/Curie-7B-v1 | 26.72 | 55.58 | 85.19 | 389.17 |
132
+ | HuggingFaceH4/zephyr-7b-beta | 33.15 | 71.65 | 71.27 | 3613.14 |
133
+ | HuggingFaceH4/zephyr-7b-alpha | 33.97 | 71.47 | 73.35 | 4464.45 |
134
+ | internlm/internlm2-chat-7b-sft | 36.97 | 73.22 | 69.96 | 4269.63 |
135
+ | internlm/internlm2-chat-7b | 37.64 | 72.29 | 71.17 | 3892.50 |
136
+ | [Bielik-7B-Instruct-v0.1](https://huggingface.co/speakleash/Bielik-7B-Instruct-v0.1) | 39.28 | 61.89 | **86.00** | 277.92 |
137
+ | mistralai/Mistral-7B-Instruct-v0.2 | 40.29 | 72.58 | 79.39 | 2088.08 |
138
+ | teknium/OpenHermes-2.5-Mistral-7B | 42.64 | 70.63 | 80.25 | 1463.00 |
139
+ | openchat/openchat-3.5-1210 | 44.17 | 71.76 | 82.15 | 1923.83 |
140
+ | speakleash/mistral_7B-v2/spkl-all_sft_v2/e1_base/spkl-all_2e6-e1_70c70cc6 | 45.44 | 71.27 | 91.50 | 279.24 |
141
+ | Nexusflow/Starling-LM-7B-beta | 45.69 | 74.58 | 81.22 | 1161.54 |
142
+ | openchat/openchat-3.5-0106 | 47.32 | 74.71 | 83.60 | 1106.56 |
143
+ | berkeley-nest/Starling-LM-7B-alpha | **47.46** | **75.73** | 82.86 | 1438.04 |
144
+ | | | | | |
145
+ | **Models with different sizes:** | | | | |
146
+ | Azurro/APT3-1B-Instruct-v1 (1B) | -13.80 | 52.11 | 12.23 | 739.09 |
147
+ | Voicelab/trurl-2-13b-academic (13B) | 29.45 | 68.19 | 79.88 | 733.91 |
148
+ | upstage/SOLAR-10.7B-Instruct-v1.0 (10.7B) | 46.07 | 76.93 | 82.86 | 789.58 |
149
+ | | | | | |
150
+ | **7B parameters pretrained and continously pretrained models:** | | | | |
151
+ | OPI-PG/Qra-7b | 11.13 | 54.40 | 75.25 | 203.36 |
152
+ | meta-llama/Llama-2-7b-hf | 12.73 | 54.02 | 77.92 | 850.45 |
153
+ | internlm/internlm2-base-7b | 20.68 | 52.39 | 69.85 | 3110.92 |
154
+ | [Bielik-7B-v0.1](https://huggingface.co/speakleash/Bielik-7B-v0.1) | 29.38 | 62.13 | **88.39** | 123.31 |
155
+ | mistralai/Mistral-7B-v0.1 | 30.67 | 60.35 | 85.39 | 857.32 |
156
+ | internlm/internlm2-7b | 33.03 | 69.39 | 73.63 | 5498.23 |
157
+ | alpindale/Mistral-7B-v0.2-hf | 33.05 | 60.23 | 85.21 | 932.60 |
158
+ | speakleash/mistral-apt3-7B/spi-e0_hf | 35.50 | 62.14 | **87.48** | 132.78 |
159
+
160
+ SpeakLeash models have one of the best scores in the RAG Reader task.
161
+ We have managed to increase Average score by almost 9 pp. in comparison to Mistral-7B-v0.1.
162
+ In our subjective evaluations of chatting skills SpeakLeash models perform better than other models with higher Average scores.
163
+
164
+
165
+
166
+ ## Limitations and Biases
167
+
168
+ Bielik-7B-Instruct-v0.1 is a quick demonstration that the base model can be easily fine-tuned to achieve compelling and promising performance. It does not have any moderation mechanisms. We're looking forward to engaging with the community in ways to make the model respect guardrails, allowing for deployment in environments requiring moderated outputs.
169
+
170
+ Bielik-7B-Instruct-v0.1 can produce factually incorrect output, and should not be relied on to produce factually accurate data. Bielik-7B-Instruct-v0.1 was trained on various public datasets. While great efforts have been taken to clear the training data, it is possible that this model can generate lewd, false, biased or otherwise offensive outputs.
171
+
172
+ ## License
173
+
174
+ Because of an unclear legal situation, we have decided to publish the model under CC BY NC 4.0 license - it allows for non-commercial use. The model can be used for scientific purposes and privately, as long as the license conditions are met.
175
+
176
+ ## Citation
177
+ Please cite this model using the following format:
178
+
179
+ ```
180
+ @misc{Bielik7Bv01,
181
+ title = {Introducing Bielik-7B-Instruct-v0.1: Instruct Polish Language Model},
182
+ author = {Ociepa, Krzysztof and Flis, Łukasz and Wróbel, Krzysztof and Kondracki, Sebastian and {SpeakLeash Team} and {Cyfronet Team}},
183
+ year = {2024},
184
+ url = {https://huggingface.co/speakleash/Bielik-7B-Instruct-v0.1},
185
+ note = {Accessed: 2024-04-01}, % change this date
186
+ urldate = {2024-04-01} % change this date
187
+ }
188
+ ```
189
+
190
+ ## Responsible for training the model
191
+
192
+ * [Krzysztof Ociepa](https://www.linkedin.com/in/krzysztof-ociepa-44886550/)<sup>SpeakLeash</sup> - team leadership, conceptualizing, data preparation, process optimization and oversight of training
193
+ * [Łukasz Flis](https://www.linkedin.com/in/lukasz-flis-0a39631/)<sup>Cyfronet AGH</sup> - coordinating and supervising the training
194
+ * [Krzysztof Wróbel](https://www.linkedin.com/in/wrobelkrzysztof/)<sup>SpeakLeash</sup> - benchmarks
195
+ * [Sebastian Kondracki](https://www.linkedin.com/in/sebastian-kondracki/)<sup>SpeakLeash</sup> - coordinating and preparation of instructions
196
+ * [Maria Filipkowska](https://www.linkedin.com/in/maria-filipkowska/)<sup>SpeakLeash</sup> - preparation of instructions
197
+ * [Paweł Kiszczak](https://www.linkedin.com/in/paveu-kiszczak/)<sup>SpeakLeash</sup> - preparation of instructions
198
+ * [Adrian Gwoździej](https://www.linkedin.com/in/adrgwo/)<sup>SpeakLeash</sup> - data quality and instructions cleaning
199
+ * [Igor Ciuciura](https://www.linkedin.com/in/igor-ciuciura-1763b52a6/)<sup>SpeakLeash</sup> - instructions cleaning
200
+ * [Jacek Chwiła](https://www.linkedin.com/in/jacek-chwila/)<sup>SpeakLeash</sup> - instructions cleaning
201
+
202
+ The model could not have been created without the commitment and work of the entire SpeakLeash team, whose contribution is invaluable. Thanks to the hard work of many individuals, it was possible to gather a large amount of content in Polish and establish collaboration between the open-science SpeakLeash project and the HPC center: ACK Cyfronet AGH. Individuals who contributed to the creation of the model through their commitment to the open-science SpeakLeash project:
203
+ [Grzegorz Urbanowicz](https://www.linkedin.com/in/grzegorz-urbanowicz-05823469/),
204
+ [Szymon Baczyński](https://www.linkedin.com/in/szymon-baczynski/),
205
+ [Paweł Cyrta](https://www.linkedin.com/in/cyrta),
206
+ [Jan Maria Kowalski](https://www.linkedin.com/in/janmariakowalski/),
207
+ [Karol Jezierski](https://www.linkedin.com/in/karol-jezierski/),
208
+ [Kamil Nonckiewicz](https://www.linkedin.com/in/kamil-nonckiewicz/),
209
+ [Izabela Babis](https://www.linkedin.com/in/izabela-babis-2274b8105/),
210
+ [Nina Babis](https://www.linkedin.com/in/nina-babis-00055a140/),
211
+ [Waldemar Boszko](https://www.linkedin.com/in/waldemarboszko),
212
+ [Remigiusz Kinas](https://www.linkedin.com/in/remigiusz-kinas/),
213
+ and many other wonderful researchers and enthusiasts of the AI world.
214
+
215
+ Members of the ACK Cyfronet AGH team:
216
+ [Szymon Mazurek](https://www.linkedin.com/in/sz-mazurek-ai/).
217
+
218
+ ## Contact Us
219
+
220
+ If you have any questions or suggestions, please use the discussion tab. If you want to contact us directly, join our [Discord SpeakLeash](https://discord.gg/3G9DVM39).