English

FedPDPO: Federated Personalized Direct Preference Optimization for Large Language Model Alignment

Machine Learning 2026-03-23 v1 Computation and Language

Abstract

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.

Keywords

Cite

@article{arxiv.2603.19741,
  title  = {FedPDPO: Federated Personalized Direct Preference Optimization for Large Language Model Alignment},
  author = {Kewen Zhu and Liping Yi and Zhiming Zhao and Zhuang Qi and Han Yu and Qinghua Hu},
  journal= {arXiv preprint arXiv:2603.19741},
  year   = {2026}
}

Comments

under review

R2 v1 2026-07-01T11:29:28.642Z