FedRL: A Reinforcement Learning Federated Recommender System for Efficient Communication Using Reinforcement Selector and Hypernet Generator

Yicheng Di,Hongjian Shi,Ruhui Ma,Honghao Gao,Yuan Liu,Weiyu Wang
DOI: https://doi.org/10.1145/3682076
2024-07-29
Abstract:The field of recommender systems aims to predict users' latent interests by analyzing their preferences and behaviors. However, privacy concerns about user data collection lead to challenges such as incomplete initial information and data sparsity. Federated learning has emerged to address these privacy issues in recommender systems. However, federated recommender systems face heterogeneity among edge devices regarding data features and sample sizes. Moreover, differences in computational and storage capabilities introduce communication overhead and processing delays during parameter aggregation at the third-party server. This paper introduces a framework named A Reinforcement Learning Federated Recommender System for Efficient Communication Using Reinforcement Selector and Hypernet Generator (FedRL) to address the proposed issues. The Reinforcement Selector (RLS) dynamically selects participating edge devices and helps maximize their use of local data resources. Meanwhile, the Hypernet Generator (HNG) optimizes communication bandwidth consumption during the federated learning parameter transmission, enabling rapid deployment and updates of new model architectures or hyperparameters. Furthermore, the framework incorporates item attributes as content embeddings in edge devices' recommender models, enriching them with global information. Real-world dataset experiments demonstrate that the proposed solution balances recommender quality and communication efficiency. The code for this work is publicly available on GitHub: https://github.com/diyicheng/FedRL .
What problem does this paper attempt to address?