ULMA: Unified Language Model Alignment with Human Demonstration and Point-wise Preference

Tianchi Cai, Xierui Song, Jiyan Jiang, Fei Teng, Jinjie Gu +1 more
12/5/2023
cs.LGcs.CL

No official code found. Best implementation: Unified-Language-Model-Alignment/src (14 stars, Apache-2.0, last commit 2023-10-07).

Abstract

Aligning language models to human expectations, e.g., being helpful and harmless, has become a pressing challenge for large language models. A typical alignment procedure consists of supervised fine-tuning and preference learning. Most preference learning methods, such as RLHF and DPO, depend on pairwise preference data, which inadequately address scenarios where human feedback is point-wise, leading to potential information loss and suboptimal performance. Addressing this gap, we introduce Point-wise Direct Preference Optimization, a novel preference learning method designed to harness point-wise feedback effectively. Our work also uncovers a novel connection between supervised fine-tuning and point-wise preference learning, culminating in Unified Language Model Alignment, a single-step method that unifies the alignment with human demonstrations and point-wise preferences. Extensive experiments on point-wise preference datasets with binary or continuous labels validate the effectiveness of our methods. Our code and a new dataset with high-quality demonstration samples on harmlessness are released.

View on arXivView PDF

Code Implementations(1)

Apache-2.0
141Shell, PythonSep 29, 20233 years agoApache-2.0

Similar Papers

Cite this paper

@article{cai2023ulma,
  title  = {ULMA: Unified Language Model Alignment with Human Demonstration and Point-wise Preference},
  author = {Tianchi Cai and Xierui Song and Jiyan Jiang and Fei Teng and Jinjie Gu and Guannan Zhang},
  year   = {2023},
  eprint = {2312.02554},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2312.02554v2}
}

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/2312.02554.svg)](https://codeofpaper.com/paper/2312.02554)

Discussion