Semi-global sequential recommendation via EM-like federated training

Li Li,Zhuohuang Zhang,Chenxi Huang,Jianwei Zhang
DOI: https://doi.org/10.1016/j.eswa.2024.123460
IF: 8.5
2024-02-18
Expert Systems with Applications
Abstract:Sequential recommendation has garnered significant attention due to its effectiveness in assisting individuals in finding items of interest. The sequential recommendation methods typically employ a centralized training approach, which can raise privacy concerns. A potential solution is to maintain user data on local devices and train a recommendation model using federated training. Traditional federated recommendation methods aim to learn a highly generalized and globally shared model for all users. However, user preferences can vary significantly, posing challenges for the global model to meet the personalized requirements of users. Recently, personalized federated recommendations have emerged to provide users with customized models. Although existing personalized federated recommendation methods are effective, they tend to acquire purely local models, leading to potential limitations in generalizability. To strike a better balance between generalization and personalization, we propose a semi-global solution called FedEM. FedEM adopts a generative approach to address the problem of sequential recommendation and draws insights from the Expectation–Maximization (EM) algorithm, which iteratively infers optimal latent variables and estimates parameters through maximum likelihood. To implement FedEM, we first unveil the latent variables of users and acquire a uniform global model through collective wisdom. Subsequently, we construct user profiles from multiple perspectives and employ the Gaussian Mixture Model (GMM) to cluster users with similar preferences into groups. Finally, each group independently trains personalized models based on the maximum group likelihood. This process is akin to executing the EM algorithm once. Experimental results on four real-world datasets demonstrate the effectiveness of FedEM. The code is available at: https://github.com/MuziLee-x/FedEM .
computer science, artificial intelligence,engineering, electrical & electronic,operations research & management science
What problem does this paper attempt to address?