Plug-and-Play Parameter-Efficient Tuning of Embeddings for Federated Recommendation

Authors

  • Haochen Yuan Harbin Institute of Technology
  • Yang Zhang University of North Texas
  • Xiang He Harbin Institute of Technology
  • Quan Z. Sheng Macquarie University
  • Zhongjie Wang Harbin Institute of Technology

DOI:

https://doi.org/10.1609/aaai.v40i19.38660

Abstract

With the rise of cloud-edge collaboration, recommendation services are increasingly trained in distributed environments. Federated Recommendation (FR) enables such multi-end collaborative training while preserving privacy by sharing model parameters instead of raw data. However, the large number of parameters, primarily due to the massive item embeddings, significantly hampers communication efficiency. While existing studies mainly focus on improving the efficiency of FR models, they largely overlook the issue of embedding parameter overhead. To address this gap, we propose a FR training framework with Parameter-Efficient Fine-Tuning (PEFT) based embedding designed to reduce the volume of embedding parameters that need to be transmitted. Our approach offers a lightweight, plugin-style solution that can be seamlessly integrated into existing FR methods. In addition to incorporating common PEFT techniques such as LoRA and Hash-based encoding, we explore the use of Residual Quantized Variational Autoencoders (RQ-VAE) as a novel PEFT strategy within our framework. Extensive experiments across various FR model backbones and datasets demonstrate that our framework significantly reduces communication overhead while improving accuracy.

Downloads

Published

2026-03-14

How to Cite

Yuan, H., Zhang, Y., He, X., Sheng, Q. Z., & Wang, Z. (2026). Plug-and-Play Parameter-Efficient Tuning of Embeddings for Federated Recommendation. Proceedings of the AAAI Conference on Artificial Intelligence, 40(19), 16235–16243. https://doi.org/10.1609/aaai.v40i19.38660

Issue

Section

AAAI Technical Track on Data Mining & Knowledge Management III