arXiv:2511.16992cs.LG2025-11

FIRM让大模型在保护隐私的前提下,高效对齐人类价值观。

FIRM: Federated In-client Regularized Multi-objective Alignment for Large Language Models

  • 客户端本地优化并正则化多目标,避免传输多个梯度。
  • 只需传一组参数,通信量大幅降低,训练更平稳。
  • 首次提供收敛保证,适合需隐私保护的多目标对齐场景。

将大语言模型(LLM)与人类价值观对齐通常需权衡帮助性与无害性等多重冲突目标。传统训练计算开销大,集中式处理又引发严重数据隐私问题。联邦学习(FL)提供了替代方案,但现有联邦多目标优化(FMOO)方法因需向服务器传输多组梯度,通信瓶颈严重,难以扩展至大模型。本文提出FIRM(联邦客户端正则化多目标对齐),通过客户端本地求解正则化多目标优化问题,直接缓解客户端分歧漂移,无需传输多梯度。客户端仅需上传一组适配参数,极大提升通信效率。我们证明算法可收敛至帕累托驻点,首次为该联邦多目标对齐场景提供有限时间收敛保证。实验表明,相比基线,FIRM训练更平稳,客户端分歧漂移减少,奖励权衡更优。我们还提出目标偏好引入方法,通过实证帕累托图验证其能平滑响应指定偏好。

原文摘要 · Abstract (English)

Aligning Large Language Models (LLMs) with human values often involves balancing multiple, conflicting objectives such as helpfulness and harmlessness. Training these models is computationally intensive, and centralizing the process raises significant data privacy concerns. Federated Learning (FL) offers a compelling alternative, but existing Federated Multi-Objective Optimization (FMOO) methods face severe communication bottlenecks as their reliance on transmitting multiple gradients to a server is unscalable for large models. We introduce FIRM (Federated In-client Regularized Multi-objective alignment), a novel algorithm that achieves both client disagreement drift mitigation and communication efficiency. In FIRM, each client locally solves a regularized multi-objective optimization problem. By directly mitigating client disagreement drift through in-client regularization, our method eliminates the need for the multi-gradient transmissions common in prior works. Consequently, clients need only to transmit a single set of adapted parameters, maintaining high communication efficiency. We prove that our algorithm converges to Pareto-stationary points and, to our knowledge, provide the first finite-time convergence guarantees for this federated multi-objective alignment setting. Empirically, we show that FIRM leads to smoother training dynamics, reduced client disagreement drift, and improved reward trade-offs compared to baselines. We further propose a method to incorporate a preference over the objectives and report empirical Pareto plots, demonstrating that FIRM can smoothly adapt trade-offs between objectives in response to specified preferences.

联邦学习多目标优化大模型对齐隐私保护

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。