Robust recipes for to align language models with human and AI preferences

The Alignment Handbook

Robust recipes to align language models with human and AI preferences.

What is this?

Just one year ago, chatbots were out of fashion and most people hadn't heard about techniques like Reinforcement Learning from Human Feedback (RLHF) to align language models with human preferences. Then, OpenAI broke the internet with ChatGPT and Meta followed suit by releasing the Llama series of language models which enabled the ML community to build their very own capable chatbots. This has led to a rich ecosystem of datasets and models that have mostly focused on teaching language models to follow instructions through supervised fine-tuning (SFT).

However, we know from the InstructGPT and Llama2 papers that significant gains in helpfulness and safety can be had by augmenting SFT with human (or AI) preferences. At the same time, aligning language models to a set of preferences is a fairly novel idea and there are few public resources available on how to train these models, what data to collect, and what metrics to measure for best downstream performance.

The Alignment Handbook aims to fill that gap by providing the community with a series of robust training recipes that span the whole pipeline.

Links 🔗

• Zephyr 7B models, datasets, and demos

Contents

The initial release of the handbook will focus on the following techniques:

• Supervised fine-tuning: teach language models to follow instructions and tips on how to collect and curate your own training dataset.
• Reward modeling: teach language models to distinguish model responses according to human or AI preferences.
• Rejection sampling: a simple, but powerful technique to boost the performance of your SFT model.
• Direct preference optimisation (DPO): a powerful and promising alternative to PPO.

Getting started

To run the code in this project, first create a Python virtual environment using e.g. Conda:

conda create -n handbook python=3.10 && conda activate handbook

Next, install PyTorch v2.1.0. Since this hardware-dependent, we direct you to the PyTorch Installation Page.

Once PyTorch is installed, you can install the remaining package dependencies as follows:

pip install .

Next, log into your Hugging Face account as follows:

huggingface-cli login

Finally, install Git LFS so that you can push models to the Hugging Face Hub:

sudo apt-get install git-lfs

Citation

If you find the content of this repo useful in your work, please cite it as follows:

@misc{alignment_handbook2023,
  author = {Lewis Tunstall and Edward Beeching and Nathan Lambert and Nazneen Rajani and Alexander M. Rush and Thomas Wolf},
  title = {The Alignment Handbook},
  year = {2023},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/huggingface/alignment-handbook}}
}