让大模型同时满足多个冲突目标,像跳舞一样平衡不同需求。
Pareto Multi-Objective Alignment for Language Models
- 将多目标对齐转化为可快速求解的凸优化问题,突破传统方法效率瓶颈。
- 在125M到7B参数模型上实现毫秒级优化,比传统方法快上千倍。
- 理论保证找到最优权衡点,适合需要灵活适应用户偏好的实际应用。
大型语言模型在现实应用中常需平衡多个相互冲突的目标,如信息量与简洁性、帮助性与创造性。现有对齐方法主要基于强化学习人类反馈(RLHF),仅优化单一奖励函数,导致行为僵化,无法捕捉人类偏好的复杂性与多样性。为解决这一问题,我们提出帕累托多目标对齐(PAMA),一种理论严谨且计算高效的算法,专为语言模型多目标对齐设计。与计算开销巨大的多目标优化(MOO)方法相比,PAMA将多目标RLHF转化为具有闭式解的凸优化问题,显著提升可扩展性。传统MOO方法复杂度为O(n²×d),其中d为模型参数量(通常达数十亿),难以直接优化;而PAMA将复杂度降至O(n),n为目标数量,使优化可在毫秒内完成。我们证明了PAMA收敛至帕累托驻点,即无法在不损害其他目标的前提下改进任一目标。在125M至7B参数的语言模型上进行的大量实验验证了其鲁棒有效的多目标对齐能力,充分展现其理论优势。PAMA为此前被认为不可行的多目标对齐问题提供高效解决方案,为实现多样化人类价值观对齐的通用型真实世界部署铺平道路。
原文摘要 · Abstract (English)
Large language models (LLMs) are increasingly deployed in real-world applications that require careful balancing of multiple, often conflicting, objectives, such as informativeness versus conciseness, or helpfulness versus creativity. However, current alignment methods, primarily based on RLHF, optimize LLMs toward a single reward function, resulting in rigid behavior that fails to capture the complexity and diversity of human preferences. This limitation hinders the adaptability of LLMs to practical scenarios, making multi-objective alignment (MOA) a critical yet underexplored area. To bridge this gap, we propose Pareto Multi-Objective Alignment (PAMA), a principled and computationally efficient algorithm designed explicitly for MOA in LLMs. In contrast to computationally prohibitive multi-objective optimization (MOO) methods, PAMA transforms multi-objective RLHF into a convex optimization with a closed-form solution, significantly enhancing scalability. Traditional MOO approaches suffer from prohibitive O(n^2*d) complexity, where d represents the number of model parameters, typically in the billions for LLMs, rendering direct optimization infeasible. PAMA reduces this complexity to O(n) where n is the number of objectives, enabling optimization to be completed within milliseconds. We provide theoretical guarantees that PAMA converges to a Pareto stationary point, where no objective can be improved without degrading at least one other. Extensive experiments across language models ranging from 125M to 7B parameters demonstrate PAMA's robust and effective MOA capabilities, aligning with its theoretical advantages. PAMA provides a highly efficient solution to the MOA problem that was previously considered intractable, offering a practical and theoretically grounded approach to aligning LLMs with diverse human values, paving the way for versatile and adaptable real-world AI deployments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。