Implemented based on PyTorch
-
The copyright of the code in this project belongs to the authors of the DeFedGCN paper (Submitted to INFOCOM 2024) and can only be used for academic research. If you need to quote or reprint, please indicate the source and the original link.
-
The final interpretation right of this copyright and disclaimer belongs to the authors.
Local data partition for clients.
Define the basic parameters for the implementation of DeFedGCN.
Define the basic LightGCN model in DeFedGCN. For more details about origninal LightGCN,please refer to LightGCN: Simplifying and Powering Graph Convolution Network for Recommendation
The entire implementation of DeFedGCN, including the training and aggregation processes.
Local training on clinets local data (sub-graphs)
Define methods of data loading.
Evaluation for DeFedGCN and corresponding attacks.
Attack experiment main program entrance, attack experiment preliminary preparation
Define the basic attack model. For more details, please refer to Model Inversion Attacks against Graph Neural Networks
Model inversation attack proceedings.
Sub-graph expansion in DeFedGCN
Generates the communication topology for all clients