Rethinking Diverse Human Preference Learning through Principal Component Analysis

Feng Luo, Rui Yang, Hao Sun, Chunyuan Deng, Jiarui Yao +3 more
2/18/2025
cs.AIcs.CL

Official code: amandaluof/DRMs (4 stars, last commit 2025-06-11).

Abstract

Understanding human preferences is crucial for improving foundation models and building personalized AI systems. However, preferences are inherently diverse and complex, making it difficult for traditional reward models to capture their full range. While fine-grained preference data can help, collecting it is expensive and hard to scale. In this paper, we introduce Decomposed Reward Models (DRMs), a novel approach that extracts diverse human preferences from binary comparisons without requiring fine-grained annotations. Our key insight is to represent human preferences as vectors and analyze them using Principal Component Analysis (PCA). By constructing a dataset of embedding differences between preferred and rejected responses, DRMs identify orthogonal basis vectors that capture distinct aspects of preference. These decomposed rewards can be flexibly combined to align with different user needs, offering an interpretable and scalable alternative to traditional reward models. We demonstrate that DRMs effectively extract meaningful preference dimensions (e.g., helpfulness, safety, humor) and adapt to new users without additional training. Our results highlight DRMs as a powerful framework for personalized and interpretable LLM alignment. Our code is available at https://github.com/amandaluof/DRMs.

View on arXivView PDF

Code Implementations(1)

amandaluof/DRMsOfficial100%

[ACL 2024 Findings] Rethinking Diverse Human Preference Learning through Principal Component Analysis

40Shell, PythonMay 26, 20251 years ago

Similar Papers

Cite this paper

@article{luo2025rethinking,
  title  = {Rethinking Diverse Human Preference Learning through Principal Component Analysis},
  author = {Feng Luo and Rui Yang and Hao Sun and Chunyuan Deng and Jiarui Yao and Jingyan Shen and Huan Zhang and Hanjie Chen},
  year   = {2025},
  eprint = {2502.13131},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2502.13131v2}
}

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

Discussion