CLAISep 26, 2024

Self-supervised Preference Optimization: Enhance Your Language Model with Preference Degree Awareness

arXiv:2409.17791v124 citationsh-index: 7Has Code
Originality Incremental advance
AI Analysis

This work addresses a key limitation in aligning language models with human preferences, offering an incremental improvement over existing methods like DPO for researchers and practitioners in AI alignment.

The paper tackles the problem of large language models overlooking varying preference degrees in responses during preference optimization, proposing a Self-supervised Preference Optimization (SPO) framework that integrates a self-supervised preference degree loss with alignment loss, resulting in significant performance boosts and state-of-the-art results on two widely used datasets.

Recently, there has been significant interest in replacing the reward model in Reinforcement Learning with Human Feedback (RLHF) methods for Large Language Models (LLMs), such as Direct Preference Optimization (DPO) and its variants. These approaches commonly use a binary cross-entropy mechanism on pairwise samples, i.e., minimizing and maximizing the loss based on preferred or dis-preferred responses, respectively. However, while this training strategy omits the reward model, it also overlooks the varying preference degrees within different responses. We hypothesize that this is a key factor hindering LLMs from sufficiently understanding human preferences. To address this problem, we propose a novel Self-supervised Preference Optimization (SPO) framework, which constructs a self-supervised preference degree loss combined with the alignment loss, thereby helping LLMs improve their ability to understand the degree of preference. Extensive experiments are conducted on two widely used datasets of different tasks. The results demonstrate that SPO can be seamlessly integrated with existing preference optimization methods and significantly boost their performance to achieve state-of-the-art performance. We also conduct detailed analyses to offer comprehensive insights into SPO, which verifies its effectiveness. The code is available at https://github.com/lijian16/SPO.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes