Enhancing federated learning-based social recommendations with graph attention networks

被引:0
|
作者
Xu, Zhihui [1 ]
Li, Bing [1 ]
Cao, Wenming [1 ]
机构
[1] Chongqing Jiaotong Univ, Sch Math & Stat, Chongqing 400074, Peoples R China
基金
中国国家自然科学基金;
关键词
Social recommendation; Federal learning; Graph neural networks; Privacy protection;
D O I
10.1016/j.neucom.2024.129045
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Social recommendations are crucial in information services, particularly for mitigating information overload on the Internet. Graph Attention Networks (GATs) have emerged as an effective solution for the social recommendation problem, harnessing the benefits of representation learning from user-item and user-user graphs. However, traditional GAT-based social recommendation models often necessitate the centralized storage and aggregation of all the users' social connections and item interactions. This centralization raises significant privacy concerns. To address these issues, we propose a federated learning-based social recommendation system using GATs. Specifically, we first design a structured-enhanced graph clustering algorithm to obtain soft clustering assignments for items. Based on these assignments, the similarity between pairwise items and their corresponding correlations are computed. Second, we employ graph attention networks to infer the local heterogeneity of social relationships, user-item interactions, and item relationships within the federated learning framework for each client. Using the soft clustering assignments, we derive hard clustering indicators and sample pseudo items. Local gradients of embedding and model parameters are then computed and uploaded to the central server for aggregation under the local differential privacy mechanism. Experimental results over real-world data sets indicate that the proposed model is more effective in privacy protection while maintaining acceptable performance compared with the central learning framework. Additionally, the proposed model is more stable in addressing the cold-start issue for items than state-of-the-art federated learning-based social recommendation models.
引用
收藏
页数:11
相关论文
共 50 条
  • [41] Deep Reinforcement Learning-based Quantization for Federated Learning
    Zheng, Sihui
    Dong, Yuhan
    Chen, Xiang
    2023 IEEE WIRELESS COMMUNICATIONS AND NETWORKING CONFERENCE, WCNC, 2023,
  • [42] Federated Learning for IoT Networks: Enhancing Efficiency and Privacy
    Zahri, Sofia
    Bennouri, Hajar
    Chehri, Abdellah
    Abdelmoniem, Ahmed M.
    2023 IEEE 9TH WORLD FORUM ON INTERNET OF THINGS, WF-IOT, 2023,
  • [43] Graph Learning-Based Cooperative Spectrum Sensing in Cognitive Radio Networks
    Jiang, Tao
    Jin, Ming
    Guo, Qinghua
    Tian, Ye
    Liu, Juan
    IEEE WIRELESS COMMUNICATIONS LETTERS, 2023, 12 (01) : 138 - 142
  • [44] Graph Structure Learning-Based Compression Method for Convolutional Neural Networks
    Wang, Tao
    Zheng, Xiangwei
    Zhang, Lifeng
    Zhang, Yuang
    ALGORITHMS AND ARCHITECTURES FOR PARALLEL PROCESSING, ICA3PP 2023, PT II, 2024, 14488 : 130 - 146
  • [45] Hyper-Graph Attention Based Federated Learning Methods for Use in Mental Health Detection
    Ahmed, Usman
    Lin, Jerry Chun-Wei
    Srivastava, Gautam Srivastava
    IEEE JOURNAL OF BIOMEDICAL AND HEALTH INFORMATICS, 2023, 27 (02) : 768 - 777
  • [46] Social Networks based Robust Federated Learning for Encrypted Traffic Classification
    Zeng, Yong
    Wang, Zhe
    Guo, Xiaoya
    Shi, Kaichao
    Liu, Zhihong
    Zhu, Xiaoyan
    Ma, Jianfeng
    ICC 2023-IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS, 2023, : 4937 - 4942
  • [47] Graph Attention Networks for Neural Social Recommendation
    Mu, Nan
    Zha, Daren
    He, Yuanye
    Tang, Zhihao
    2019 IEEE 31ST INTERNATIONAL CONFERENCE ON TOOLS WITH ARTIFICIAL INTELLIGENCE (ICTAI 2019), 2019, : 1320 - 1327
  • [48] DFedSN: Decentralized federated learning based on heterogeneous data in social networks
    Yikuan Chen
    Li Liang
    Wei Gao
    World Wide Web, 2023, 26 : 2545 - 2568
  • [49] DFedSN: Decentralized federated learning based on heterogeneous data in social networks
    Chen, Yikuan
    Liang, Li
    Gao, Wei
    WORLD WIDE WEB-INTERNET AND WEB INFORMATION SYSTEMS, 2023, 26 (05): : 2545 - 2568
  • [50] Point clouds learning with attention-based graph convolution networks
    Xie, Zhuyang
    Chen, Junzhou
    Peng, Bo
    NEUROCOMPUTING, 2020, 402 : 245 - 255