Aligning language models with human preferences

Tomasz Korbak
4/18/2024
cs.LGcs.CL

No official code found. Best implementation: tomekkorbak/pretraining-with-human-feedback (184 stars, MIT, last commit 2024-02-13).

Abstract

Language models (LMs) trained on vast quantities of text data can acquire sophisticated skills such as generating summaries, answering questions or generating code. However, they also manifest behaviors that violate human preferences, e.g., they can generate offensive content, falsehoods or perpetuate social biases. In this thesis, I explore several approaches to aligning LMs with human preferences. First, I argue that aligning LMs can be seen as Bayesian inference: conditioning a prior (base, pretrained LM) on evidence about human preferences (Chapter 2). Conditioning on human preferences can be implemented in numerous ways. In Chapter 3, I investigate the relation between two approaches to finetuning pretrained LMs using feedback given by a scoring function: reinforcement learning from human feedback (RLHF) and distribution matching. I show that RLHF can be seen as a special case of distribution matching but distributional matching is strictly more general. In chapter 4, I show how to extend the distribution matching to conditional language models. Finally, in chapter 5 I explore a different root: conditioning an LM on human preferences already during pretraining. I show that involving human feedback from the very start tends to be more effective than using it only during supervised finetuning. Overall, these results highlight the room for alignment techniques different from and complementary to RLHF.

View on arXivView PDF

Code Implementations(1)

MIT

Code accompanying the paper Pretraining Language Models with Human Preferences

18415PythonFeb 20, 20232 years agoMIT
ai-alignmentai-safetydecision-transformersgptlanguage-models+3 more

Cite this paper

@article{korbak2024aligning,
  title  = {Aligning language models with human preferences},
  author = {Tomasz Korbak},
  year   = {2024},
  eprint = {2404.12150},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2404.12150v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2404.12150.svg)](https://codeofpaper.com/paper/2404.12150)

Discussion