🤖 AI Summary
To address the challenge of accurately modeling users’ implicit multi-dimensional preferences from sparse pairwise comparative feedback in large language model (LLM) response personalization, this paper proposes AMPLe—a novel framework for adaptive multi-dimensional preference learning. AMPLe innovatively refines Bayesian posterior updating to suppress noise and bias inherent in pairwise comparisons, and introduces a generalized binary search–based active querying strategy to minimize user annotation effort. By jointly modeling multi-dimensional preferences and grounding query selection in theoretical guarantees, AMPLe achieves over 40% improvement in feedback efficiency for language generation tasks, producing high-quality personalized responses with only a small number of comparative judgments. The implementation is publicly available.
📝 Abstract
Large language models (LLMs) have shown remarkable success, but aligning them with human preferences remains a core challenge. As individuals have their own, multi-dimensional preferences, recent studies have explored multi-dimensional personalization, which aims to enable models to generate responses personalized to explicit preferences. However, human preferences are often implicit and thus difficult to articulate, limiting the direct application of this approach. To bridge this gap, we propose Active Multi-dimensional Preference Learning (AMPLe), designed to capture implicit user preferences from interactively collected comparative feedback. Building on Bayesian inference, our work introduces a modified posterior update procedure to mitigate estimation bias and potential noise in comparisons. Also, inspired by generalized binary search, we employ an active query selection strategy to minimize the number of required comparisons by a user. Through theoretical analysis and experiments on language generation tasks, we demonstrate feedback efficiency and effectiveness of our framework in personalizing model responses. Our code is publicly available at https://github.com/ml-postech/AMPLe .