DavidGF's picture
Update README.md
d4bf6fb verified
|
raw
history blame
7.68 kB
metadata
license: other
license_name: gemma-terms-of-use
license_link: https://ai.google.dev/gemma/terms
language:
  - de
  - en
tags:
  - sft
  - dpo
  - laserRMT
  - laser-QLoRa
  - finetune
  - work in progress
  - alpha

SauerkrautLM

VAGO solutions SauerkrautLM-Gemma-7b (alpha)

Introducing SauerkrautLM-Gemma-7b – our Sauerkraut version of the powerful google/gemma-7b ! It is an early stage finetuned model and should be used with caution!

The model SauerkrautLM-Gemma-7b is a joint effort between VAGO solutions and Hyperspace.ai. Much appreciation goes to the tremendous research effort of Fernando Fernandes Neto, David Golchinfar and Eric Hartford on their laserRMT approach. Without their independent research collaboration this model release would not have been possible.

  • Fintuned with SFT
  • Aligned with DPO
  • Using a novel training technique: laser-QLoRA - we partially freeze the model according to a laser-like analysis (Official Paper soon). It allows to evaluate the no free lunch theorem and supports better decision making when optimizing the theorem - created by the LaserRMT research group
  • Optimized with LaserRMT

Table of Contents

  1. Overview of all SauerkrautLM-Gemma-7b models
  2. Model Details
  3. Evaluation
  4. Disclaimer
  5. Contact
  6. Collaborations
  7. Acknowledgement

All SauerkrautLM-Gemma-7b Models

Model HF GPTQ GGUF AWQ
SauerkrautLM-Gemma-7b Link coming soon coming soon coming soon

Model Details

SauerkrautLM-Gemma-7b

Training procedure:

Warning: This finetuned model is in an early stage and we sometimes observed strange behavior. It is still work in progress!

Anyone who has attempted or succeeded in fine-tuning a model is aware of the difficulty in nudging it towards a specific skill, such as mastering new languages, as well as the challenges associated with achieving significant improvements in performance. Experimenting with a novel training strategy and Spherical Linear Interpolation alongside a lasered version of the model itself has proven to be both fascinating and revealing.

Furthermore, we developed one iteration of the model using our entire SFT -Sauerkraut dataset and two additional iterations using subsets of the full dataset—one focused on enhancing MMLU and TQA capabilities, and the other on boosting GSM8K and Winogrande skills.

After optimizing our primary SFT model, we applied a similar strategy to our new DPO Dataset, dividing it into further subsets. We trained one model on the entire dataset again and two more on these specialized subsets.

We actively monitor and assesed the results of each training. Whenever we found a decrease in perplexity on the gsm8k benchmark we intervined. By following this procedure we were able to improve the overall performance, especially in math abilities, without detracting from performance on other benchmarks—a task that is, in general, quite difficult.

This process not only helps in understanding the effectiveness of Spherical Linear Interpolation but also introduces a new method for refining models with enhanced skills through a cycle of targeted data selection (Laser data(x)) + SLERP, followed by a subsequent focus on different data (Laser again on data(y)).

Additionally, we integrated a novel training strategy on the SFT and DPO training process, where we partially freeze the model according to a laser-like analysis aiming to navigate and optimize the trade-offs highlighted by the no free lunch theorem. This innovative training method effectively prevents the significant problem of language models forgetting previously acquired knowledge. This aspect is particularly crucial when attempting to teach the model specific skills, such as a new language, where in general, the model might lose a considerable amount of its prior knowledge and exhibit a decline in overall intelligence.

Detailed information on how the new training strategy works and the advantages it offers over conventional training methods will soon be published in a detailed paper by the LaserRMT research group.

We teached German language skills on this model. As far as we know, it is the first Gemma model with bilingual skills in German and English. Nevertheless, formulations may occur that are not entirely correct (still work in progress).

Prompt Template:

We trained on vicuna prompt template. Please add the following stopping string to your client: "","

" (we did not add the special tokens to the training config)

You are a helpful AI Assistant.

USER: Hello, how are you?
ASSISTANT:

Evaluation

Open LLM Leaderboard:

  • benchmarks were done with the newest version of lm-evaluation-harness on a batch-size of 1:
Metric Value
Avg. 68.92
ARC (25-shot) 59.98
HellaSwag (10-shot) 82.28
MMLU (5-shot) 63.53
TruthfulQA (0-shot) 61.2
Winogrande (5-shot) 80.27
GSM8K (5-shot) 66.26

Dispite the fact that we archived great results on the Open LLM leaderboard benchmarks the model subjectively does not feel as smart as comparable mistral finetunes. Most of its answers are coherent but we observed that the model sometimes answers realy lazy or odd.

Disclaimer

We must inform users that despite our best efforts in data cleansing, the possibility of uncensored content slipping through cannot be entirely ruled out. However, we cannot guarantee consistently appropriate behavior. Therefore, if you encounter any issues or come across inappropriate content, we kindly request that you inform us through the contact information provided. Additionally, it is essential to understand that the licensing of these models does not constitute legal advice. We are not held responsible for the actions of third parties who utilize our models.  

Contact

If you are interested in customized LLMs for business applications, please get in contact with us via our websites. We are also grateful for your feedback and suggestions.  

Collaborations

We are also keenly seeking support and investment for our startups, VAGO solutions and Hyperspace where we continuously advance the development of robust language models designed to address a diverse range of purposes and requirements. If the prospect of collaboratively navigating future challenges excites you, we warmly invite you to reach out to us at VAGO solutions, Hyperspace.computer

Acknowledgement

Many thanks to google for providing such valuable model to the Open-Source community