Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

A Federated Framework for LLM-based Recommendation

About

Large Language Models (LLMs) have empowered generative recommendation systems through fine-tuning user behavior data. However, utilizing the user data may pose significant privacy risks, potentially leading to ethical dilemmas and violations of data protection regulations. To address the privacy concerns, Federated Learning for Recommendation (Fed4Rec) has been identified as a promising solution. However, directly applying Fed4Rec in the LLM context introduces two challenges: 1) exacerbated client performance imbalance, which ultimately impacts the system's long-term effectiveness, and 2) substantial client resource costs, posing a high demand for clients' both computational and storage capability to locally train and infer LLMs. To tackle these challenges, we propose a federated framework for LLM-based recommendation (shorted as FELLRec). Generally, FELLRec designs two key strategies. 1) Dynamic balance strategy, which designs dynamic parameter aggregation and learning speed for different clients, aiming to ensure balanced performance across clients. 2) Flexible storage strategy, which selectively retains certain sensitive LLM layers on the client side, while offloading other layers to the server, aiming to preserve privacy while saving resources. Experiment results show that FELLRec can achieve a more balanced client performance and improved overall performance in a computational and storage-efficient way while safeguarding user privacy well.

Jujia Zhao, Wenjie Wang, Chen Xu, See-Kiong Ng, Tat-Seng Chua• 2024

Related benchmarks

TaskDatasetResultRank
Federated Cross-Domain RecommendationGoodReads Comics
Hit Rate @58.4
14
Federated Cross-Domain RecommendationGoodReads Crime
H@53.16
14
Federated Cross-Domain RecommendationGoodReads Children
H@54.74
14
Federated Cross-Domain RecommendationAmazon Beauty (test)
H@51.73
10
Federated Cross-Domain RecommendationAmazon Clothing (test)
Hits@50.85
10
Federated Cross-Domain RecommendationGoodReads Crime, Comics & Children Average
Avg H@55.43
10
RecommendationAmazon Electronics & Phones (test)
Avg Local Train Time (s)850
5
Showing 7 of 7 rows

Other info

Follow for update