Evolutionary Contrastive Distillation for Language Model Alignment

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Katz-Samuels, Julian, Li, Zheng, Yun, Hyokun, Nigam, Priyanka, Xu, Yi, Petricek, Vaclav, Yin, Bing, Chilimbi, Trishul
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929535438553088
author Katz-Samuels, Julian
Li, Zheng
Yun, Hyokun
Nigam, Priyanka
Xu, Yi
Petricek, Vaclav
Yin, Bing
Chilimbi, Trishul
author_facet Katz-Samuels, Julian
Li, Zheng
Yun, Hyokun
Nigam, Priyanka
Xu, Yi
Petricek, Vaclav
Yin, Bing
Chilimbi, Trishul
contents The ability of large language models (LLMs) to execute complex instructions is essential for their real-world applications. However, several recent studies indicate that LLMs struggle with challenging instructions. In this paper, we propose Evolutionary Contrastive Distillation (ECD), a novel method for generating high-quality synthetic preference data designed to enhance the complex instruction-following capability of language models. ECD generates data that specifically illustrates the difference between a response that successfully follows a set of complex instructions and a response that is high-quality, but nevertheless makes some subtle mistakes. This is done by prompting LLMs to progressively evolve simple instructions to more complex instructions. When the complexity of an instruction is increased, the original successful response to the original instruction becomes a "hard negative" response for the new instruction, mostly meeting requirements of the new instruction, but barely missing one or two. By pairing a good response with such a hard negative response, and employing contrastive learning algorithms such as DPO, we improve language models' ability to follow complex instructions. Empirically, we observe that our method yields a 7B model that exceeds the complex instruction-following performance of current SOTA 7B models and is competitive even with open-source 70B models.
format Preprint
id arxiv_https___arxiv_org_abs_2410_07513
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Evolutionary Contrastive Distillation for Language Model Alignment
Katz-Samuels, Julian
Li, Zheng
Yun, Hyokun
Nigam, Priyanka
Xu, Yi
Petricek, Vaclav
Yin, Bing
Chilimbi, Trishul
Machine Learning
Artificial Intelligence
Computation and Language
The ability of large language models (LLMs) to execute complex instructions is essential for their real-world applications. However, several recent studies indicate that LLMs struggle with challenging instructions. In this paper, we propose Evolutionary Contrastive Distillation (ECD), a novel method for generating high-quality synthetic preference data designed to enhance the complex instruction-following capability of language models. ECD generates data that specifically illustrates the difference between a response that successfully follows a set of complex instructions and a response that is high-quality, but nevertheless makes some subtle mistakes. This is done by prompting LLMs to progressively evolve simple instructions to more complex instructions. When the complexity of an instruction is increased, the original successful response to the original instruction becomes a "hard negative" response for the new instruction, mostly meeting requirements of the new instruction, but barely missing one or two. By pairing a good response with such a hard negative response, and employing contrastive learning algorithms such as DPO, we improve language models' ability to follow complex instructions. Empirically, we observe that our method yields a 7B model that exceeds the complex instruction-following performance of current SOTA 7B models and is competitive even with open-source 70B models.
title Evolutionary Contrastive Distillation for Language Model Alignment
topic Machine Learning
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2410.07513