Self-Supervised Alignment with Mutual Information: Learning to Follow Principles without Preference Labels

TL;DR

SAMI is a novel iterative algorithm that fine-tunes pretrained language models to follow behavioral principles without requiring preference labels, demonstrations, or human oversight, improving alignment in dialogue and summarization tasks.

Contribution

Introduces SAMI, a method for aligning language models to principles using mutual information, eliminating the need for preference labels or demonstrations.

Findings

SAMI improves model performance on dialogue and summarization tasks.

SAMI surpasses instruction-finetuned baselines in win rates.

SAMI generalizes to diverse principles and larger models.

Abstract

When prompting a language model (LM), users often expect the model to adhere to a set of behavioral principles across diverse tasks, such as producing insightful content while avoiding harmful or biased language. Instilling such principles (i.e., a constitution) into a model is resource-intensive, technically challenging, and generally requires human preference labels or examples. We introduce SAMI, an iterative algorithm that finetunes a pretrained language model (without requiring preference labels or demonstrations) to increase the conditional mutual information between constitutions and self-generated responses given queries from a dataset. On single-turn dialogue and summarization, a SAMI-trained mistral-7b outperforms the initial pretrained model, with win rates between 66% and 77%. Strikingly, it also surpasses an instruction-finetuned baseline (mistral-7b-instruct) with win rates between 55% and 57% on single-turn dialogue. SAMI requires a model that writes the principles. To avoid dependence on strong models for writing principles, we align a strong pretrained model (mixtral-8x7b) using constitutions written by a weak instruction-finetuned model (mistral-7b-instruct), achieving a 65% win rate on summarization. Finally, we investigate whether SAMI generalizes to diverse summarization principles (e.g., "summaries should be scientific") and scales to stronger models (llama3-70b), finding that it achieves win rates of up to 68% for learned and 67% for held-out principles compared to the base model. Our results show that a pretrained LM can learn to follow constitutions without using preference labels, demonstrations, or human oversight.