Beyond the Binary: Capturing Diverse Preferences with Reward Regularization

Abstract

Large language models (LLMs) are increasingly deployed via public-facing interfaces to interact with millions of users, each with diverse preferences. Despite this, preference tuning of LLMs predominantly relies on reward models trained using binary judgments where annotators select the preferred choice out of pairs of model outputs. In this work, we argue that this reliance on binary choices does not capture the broader, aggregate preferences of the target user in real-world tasks. We propose a taxonomy that identifies two dimensions of subjectivity where different users disagree on the preferred output—namely, the \textit{Plurality of Responses to Prompts}, where prompts allow for multiple correct answers, and the \textit{Indistinguishability of Responses}, where candidate outputs are paraphrases of each other. We show that reward models correlate weakly with user preferences in these cases. As a first step to address this issue, we introduce a simple yet effective method that augments existing binary preference datasets with synthetic preference judgments to estimate potential user disagreement. Incorporating these via a margin term as a form of regularization during model training yields predictions that better align with the aggregate user preferences.

Cite

Text

Padmakumar et al. "Beyond the Binary: Capturing Diverse Preferences with Reward Regularization." NeurIPS 2024 Workshops: SoLaR, 2024.

Markdown

[Padmakumar et al. "Beyond the Binary: Capturing Diverse Preferences with Reward Regularization." NeurIPS 2024 Workshops: SoLaR, 2024.](https://mlanthology.org/neuripsw/2024/padmakumar2024neuripsw-beyond/)

BibTeX

@inproceedings{padmakumar2024neuripsw-beyond,
  title     = {{Beyond the Binary: Capturing Diverse Preferences with Reward Regularization}},
  author    = {Padmakumar, Vishakh and Jin, Chuanyang and Kirk, Hannah Rose and He, He},
  booktitle = {NeurIPS 2024 Workshops: SoLaR},
  year      = {2024},
  url       = {https://mlanthology.org/neuripsw/2024/padmakumar2024neuripsw-beyond/}
}