用少量用户数据实时调整模型偏好,效果远超传统方法。
Test-Time Alignment via Hypothesis Reweighting
- 通过重加权多个预测头实现快速个性化
- 仅需1-5个标注样本,准确率提升20%
- 适合需要即时适配的推荐与生成场景
基于聚合偏好训练的奖励模型往往无法捕捉个体用户的价值观,而现有微调或长上下文条件化方法在实时个性化中成本过高。我们提出假设重加权(HyRe),仅需目标用户或领域1-5个已标注样本,即可通过重加权集成成员实现实时个性化。该方法基于观察:不同预测头可捕捉偏好数据的不同有效解释,重加权能显著优于均匀平均。HyRe训练单个网络,含多个预测头以捕捉偏好数据的多种合理解释,再使用贝叶斯更新提升与目标用户偏好最匹配的头权重。整个过程仅需一次前向传播,计算开销低于1%,适用于推理时个性化。我们在多样化目标偏好分布上评估了HyRe,每个目标分布仅需5组偏好对,即在2B和8B规模下超越RewardBench上现有最优奖励模型,并在32项个性化任务中提升奖励模型准确率20%。
原文摘要 · Abstract (English)
Reward models trained on aggregate preferences often fail to capture individual users' values, but existing adaptation methods such as fine-tuning or long-context conditioning are too costly for real-time personalization. We propose Hypothesis Reweighting (HyRe), which enables real-time personalization by reweighting ensemble members using just 1-5 labeled examples from the target user or domain. Our method builds on the empirical observation that when different heads capture different valid interpretations of preference data, reweighting them can substantially outperform uniform averaging. HyRe trains a single network with multiple prediction heads that capture different valid interpretations of preference data, then uses a Bayesian update to upweight the heads that best match the target user's preferences. This requires only a single forward pass with negligible (<1%) computational overhead, making it practical for inference-time personalization. We evaluate HyRe across diverse target preference distributions. With as few as five preference pairs per target distribution, HyRe surpasses state-of-the-art reward models on RewardBench at 2B and 8B scale and improves reward model accuracy by 20% across 32 personalization tasks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。