SPO: Multi-Dimensional Preference Sequential Alignment With Implicit Reward Modeling

Xingzhou Lou,Junge Zhang,Jian Xie,Lifeng Liu,Dong Yan,Kaiqi Huang
2024-10-11
Abstract: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.
Machine Learning
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to achieve multi - dimensional alignment of human preferences in large - language models (LLMs). Current methods either overlook the multi - dimensional nature of human preferences (e.g., usefulness and harmlessness) or face complexity issues when managing multiple reward models. To this end, the paper proposes the **Sequential Preference Optimization (SPO)** method, which achieves multi - dimensional alignment with human preferences by fine - tuning LLMs round by round. SPO avoids explicit reward modeling and directly optimizes the model to align with subtle human preferences. The paper not only theoretically derives the closed - form optimal strategy and loss function of SPO, but also shows through gradient analysis how SPO fine - tunes LLMs while maintaining the alignment of previously optimized dimensions. Experimental results show that SPO can successfully achieve multi - dimensional alignment of human preferences on LLMs of different scales and significantly outperforms the baseline methods.