解决联邦学习中大模型对齐难题,提升隐私保护下的个性化效果。
FedPDPO: Federated Personalized Direct Preference Optimization for Large Language Model Alignment
- 采用低秩适配器实现高效通信,客户端保持预训练模型冻结。
- 引入显式奖励头与瓶颈适配器,显著提升非独立同分布数据下的性能。
- 适合需要隐私保护的个性化大模型对齐场景,如医疗、金融领域。
在联邦学习中对齐大型语言模型与人类偏好面临挑战,因数据分散、隐私敏感且高度非独立同分布。直接偏好优化(DPO)是强化学习人类反馈的高效替代方案,但在联邦设置下受非独立同分布数据影响,隐式奖励泛化能力差。为此,我们提出FedPDPO:一种用于大模型偏好的个性化联邦框架。该框架采用参数高效微调架构,客户端保留冻结的预训练模型骨干,并附加低秩适配器(LoRA),实现通信高效聚合。为应对非独立同分布异构性,设计:(1)共享全局LoRA适配器与客户端特定的LLM头部;(2)基于客户端显式奖励头的个性化DPO训练策略,补充隐式奖励并缓解异构性;(3)瓶颈适配器以平衡全局与本地特征。提供理论分析,证明其概率基础与合理性。多组偏好数据集上的大量实验表明,该方法在联邦域内与跨域设置下均达到领先性能,平均准确率提升最高达4.80%。
原文摘要 · Abstract (English)
Aligning large language models (LLMs) with human preferences in federated learning (FL) is challenging due to decentralized, privacy-sensitive, and highly non-IID preference data. Direct Preference Optimization (DPO) offers an efficient alternative to reinforcement learning with human feedback (RLHF), but its direct application in FL suffers from severe performance degradation under non-IID data and limited generalization of implicit rewards. To bridge this gap, we propose FedPDPO (Federated Personalized Direct Preference Optimization), a personalized federated framework for preference alignment of LLMs. It adopts a parameter-efficient fine-tuning architecture where each client maintains a frozen pretrained LLM backbone augmented with a Low-Rank Adaptation (LoRA) adapter, enabling communication-efficient aggregation. To address non-IID heterogeneity, we devise (1) the globally shared LoRA adapter with the personalized client-specific LLM head. Moreover, we introduce (2) a personalized DPO training strategy with a client-specific explicit reward head to complement implicit rewards and further alleviate non-IID heterogeneity, and (3) a bottleneck adapter to balance global and local features. We provide theoretical analysis establishing the probabilistic foundation and soundness. Extensive experiments on multiple preference datasets demonstrate state-of-the-art performance, achieving up to 4.80% average accuracy improvements in federated intra-domain and cross-domain settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。