Self-Augmented Preference Optimization: Off-Policy Paradigms for Language Model Alignment

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yin, Yueqin, Wang, Zhendong, Xie, Yujia, Chen, Weizhu, Zhou, Mingyuan
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911896638062592
author Yin, Yueqin
Wang, Zhendong
Xie, Yujia
Chen, Weizhu
Zhou, Mingyuan
author_facet Yin, Yueqin
Wang, Zhendong
Xie, Yujia
Chen, Weizhu
Zhou, Mingyuan
contents Traditional language model alignment methods, such as Direct Preference Optimization (DPO), are limited by their dependence on static, pre-collected paired preference data, which hampers their adaptability and practical applicability. To overcome this limitation, we introduce Self-Augmented Preference Optimization (SAPO), an effective and scalable training paradigm that does not require existing paired data. Building on the self-play concept, which autonomously generates negative responses, we further incorporate an off-policy learning pipeline to enhance data exploration and exploitation. Specifically, we employ an Exponential Moving Average (EMA) model in conjunction with a replay buffer to enable dynamic updates of response segments, effectively integrating real-time feedback with insights from historical data. Our comprehensive evaluations of the LLaMA3-8B and Mistral-7B models across benchmarks, including the Open LLM Leaderboard, IFEval, AlpacaEval 2.0, and MT-Bench, demonstrate that SAPO matches or surpasses established offline contrastive baselines, such as DPO and Odds Ratio Preference Optimization, and outperforms offline self-play methods like SPIN. Our code is available at https://github.com/yinyueqin/SAPO
format Preprint
id arxiv_https___arxiv_org_abs_2405_20830
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Self-Augmented Preference Optimization: Off-Policy Paradigms for Language Model Alignment
Yin, Yueqin
Wang, Zhendong
Xie, Yujia
Chen, Weizhu
Zhou, Mingyuan
Computation and Language
Machine Learning
Traditional language model alignment methods, such as Direct Preference Optimization (DPO), are limited by their dependence on static, pre-collected paired preference data, which hampers their adaptability and practical applicability. To overcome this limitation, we introduce Self-Augmented Preference Optimization (SAPO), an effective and scalable training paradigm that does not require existing paired data. Building on the self-play concept, which autonomously generates negative responses, we further incorporate an off-policy learning pipeline to enhance data exploration and exploitation. Specifically, we employ an Exponential Moving Average (EMA) model in conjunction with a replay buffer to enable dynamic updates of response segments, effectively integrating real-time feedback with insights from historical data. Our comprehensive evaluations of the LLaMA3-8B and Mistral-7B models across benchmarks, including the Open LLM Leaderboard, IFEval, AlpacaEval 2.0, and MT-Bench, demonstrate that SAPO matches or surpasses established offline contrastive baselines, such as DPO and Odds Ratio Preference Optimization, and outperforms offline self-play methods like SPIN. Our code is available at https://github.com/yinyueqin/SAPO
title Self-Augmented Preference Optimization: Off-Policy Paradigms for Language Model Alignment
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2405.20830