Many Preferences, Few Policies: Towards Scalable Language Model Personalization
This addresses the scalability challenge in LLM personalization for users, though it is incremental as it builds on existing portfolio optimization concepts.
The paper tackles the impracticality of maintaining a separate LLM per user for personalization by developing PALM, a method to select a small portfolio of LLMs that captures diverse user preferences, achieving theoretical guarantees on portfolio size and approximation quality with empirical validation.
The holy grail of LLM personalization is a single LLM for each user, perfectly aligned with that user's preferences. However, maintaining a separate LLM per user is impractical due to constraints on compute, memory, and system complexity. We address this challenge by developing a principled method for selecting a small portfolio of LLMs that captures representative behaviors across heterogeneous users. We model user preferences across multiple traits (e.g., safety, humor, brevity) through a multi-dimensional weight vector. Given reward functions across these dimensions, our algorithm PALM (Portfolio of Aligned LLMs) generates a small portfolio of LLMs such that, for any weight vector, the portfolio contains a near-optimal LLM for the corresponding scalarized objective. To the best of our knowledge, this is the first result that provides theoretical guarantees on both the size and approximation quality of LLM portfolios for personalization. It characterizes the trade-off between system cost and personalization, as well as the diversity of LLMs required to cover the landscape of user preferences. We provide empirical results that validate these guarantees and demonstrate greater output diversity over common baselines.