Saved in:
| Main Authors: | , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2024
|
| Subjects: | |
| Online Access: | https://arxiv.org/abs/2405.12739 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866917800305491968 |
|---|---|
| author | Lou, Xingzhou Zhang, Junge Xie, Jian Liu, Lifeng Yan, Dong Huang, Kaiqi |
| author_facet | Lou, Xingzhou Zhang, Junge Xie, Jian Liu, Lifeng Yan, Dong Huang, Kaiqi |
| contents | Human preference alignment is critical in building powerful and reliable large language models (LLMs). However, current methods either ignore the multi-dimensionality of human preferences (e.g. helpfulness and harmlessness) or struggle with the complexity of managing multiple reward models. To address these issues, we propose Sequential Preference Optimization (SPO), a method that sequentially fine-tunes LLMs to align with multiple dimensions of human preferences. SPO avoids explicit reward modeling, directly optimizing the models to align with nuanced human preferences. We theoretically derive closed-form optimal SPO policy and loss function. Gradient analysis is conducted to show how SPO manages to fine-tune the LLMs while maintaining alignment on previously optimized dimensions. Empirical results on LLMs of different size and multiple evaluation datasets demonstrate that SPO successfully aligns LLMs across multiple dimensions of human preferences and significantly outperforms the baselines. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2405_12739 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | SPO: Multi-Dimensional Preference Sequential Alignment With Implicit Reward Modeling Lou, Xingzhou Zhang, Junge Xie, Jian Liu, Lifeng Yan, Dong Huang, Kaiqi Machine Learning Human preference alignment is critical in building powerful and reliable large language models (LLMs). However, current methods either ignore the multi-dimensionality of human preferences (e.g. helpfulness and harmlessness) or struggle with the complexity of managing multiple reward models. To address these issues, we propose Sequential Preference Optimization (SPO), a method that sequentially fine-tunes LLMs to align with multiple dimensions of human preferences. SPO avoids explicit reward modeling, directly optimizing the models to align with nuanced human preferences. We theoretically derive closed-form optimal SPO policy and loss function. Gradient analysis is conducted to show how SPO manages to fine-tune the LLMs while maintaining alignment on previously optimized dimensions. Empirical results on LLMs of different size and multiple evaluation datasets demonstrate that SPO successfully aligns LLMs across multiple dimensions of human preferences and significantly outperforms the baselines. |
| title | SPO: Multi-Dimensional Preference Sequential Alignment With Implicit Reward Modeling |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2405.12739 |