about
Personalized Soups: Personalized LLM Alignment via Post-Hoc Parameter Merging (arxiv.org)
1 point by birriel on Oct 19, 2023 | hide | past | pdf | discuss on HN

In plain words: Instead of training one model on everyone's averaged preferences, they train a separate version for each trait a user wants, then mix the trained models together to fit that person. This matched conflicting personal tastes better than the usual single averaged model.

Abstract · Personalized Soups: Personalized Large Language Model Alignment via Post-hoc Parameter Merging

While Reinforcement Learning from Human Feedback (RLHF) aligns Large Language Models (LLMs) with general, aggregate human preferences, it is suboptimal for learning diverse, individual perspectives. In this work, we study Reinforcement Learning from Personalized Human Feedback (RLPHF) problem, wherein LLMs are aligned to multiple (sometimes conflicting) preferences by modeling alignment as a Multi-Objective Reinforcement Learning (MORL) problem. Compared to strong single-objective baselines, we show that we can achieve personalized alignment by decomposing preferences into multiple dimensions. These dimensions are defined based on personalizations that are declared as desirable by the user. In this work, we show that they can be efficiently trained independently in a distributed manner and combined effectively post-hoc through parameter merging. The code is available at https://github.com/joeljang/RLPHF.

Joel Jang, Seungone Kim, Bill Yuchen Lin, Yizhong Wang, Jack Hessel, Luke Zettlemoyer, Hannaneh Hajishirzi, Yejin Choi, Prithviraj Ammanabrolu
arXiv:2310.11564 · cs.CL · submitted Oct 17, 2023
abstract · pdf · html · Preprint

add comment on HN