用少量用户数据实现个性化大模型对齐,支持灵活调整行为模式。
Few-shot Steerable Alignment: Adapting Rewards and LLM Policies with Neural Processes
- 基于可扩展参数条件化的神经过程,从少量选择中推断偏好
- 仅需少量样本即可适应不同用户偏好,避免重复训练多个模型
- 适合需要个性化交互的场景,如客服、创作助手等
随着大语言模型在日常应用中的普及,如何使其与个体用户的多样化偏好对齐成为关键挑战。现有方法通常假设用户目标一致,依赖单一目标微调,但人类偏好受多种不可观测因素影响,常导致偏好数据冲突。现有解决方案多需为特定目标标注昂贵数据集,并训练多个奖励模型或LLM策略,计算成本高且不切实际。本文提出一种少样本可调控对齐框架,通过用户少量选择推断其潜在偏好。我们扩展了Bradley-Terry-Luce模型以处理未观测的异质性因素,并提出其在奖励建模与LLM微调中的实用实现。得益于函数式参数空间条件化设计,经该框架训练的LLM可在推理时根据个体偏好灵活调整输出,生成一系列行为模式。实验验证了方法在数据高效下捕捉并对齐多样人类偏好的有效性。代码已公开:https://github.com/kasia-kobalczyk/few-shot-steerable-alignment。
原文摘要 · Abstract (English)
As large language models (LLMs) become increasingly embedded in everyday applications, ensuring their alignment with the diverse preferences of individual users has become a critical challenge. Currently deployed approaches typically assume homogeneous user objectives and rely on single-objective fine-tuning. However, human preferences are inherently heterogeneous, influenced by various unobservable factors, leading to conflicting signals in preference data. Existing solutions addressing this diversity often require costly datasets labelled for specific objectives and involve training multiple reward models or LLM policies, which is computationally expensive and impractical. In this work, we present a novel framework for few-shot steerable alignment, where users' underlying preferences are inferred from a small sample of their choices. To achieve this, we extend the Bradley-Terry-Luce model to handle heterogeneous preferences with unobserved variability factors and propose its practical implementation for reward modelling and LLM fine-tuning. Thanks to our proposed approach of functional parameter-space conditioning, LLMs trained with our framework can be adapted to individual preferences at inference time, generating outputs over a continuum of behavioural modes. We empirically validate the effectiveness of methods, demonstrating their ability to capture and align with diverse human preferences in a data-efficient manner. Our code is made available at: https://github.com/kasia-kobalczyk/few-shot-steerable-alignment.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。