通过低秩分解实现高效个性化图学习,显著降低通信开销。
Communication-Efficient Personalized Federal Graph Learning via Low-Rank Decomposition
- 将模型分为低秩共性部分与稀疏私有部分,分途优化。
- 在16个数据集上比最先进方法准确率提升5.64%,通信量减少18.58倍。
- 适合隐私敏感、数据异构的分布式图学习场景。
联邦图学习(FGL)使各客户端在本地处理私有图数据并与中心服务器交互,从而保护隐私。然而,客户端的图数据通常非独立同分布,单一模型难以在所有客户端上表现良好。此外,通信开销是FGL的主要瓶颈。为此,我们提出一种高效的个性化联邦图学习算法CEFGL。该方法将模型参数分解为低秩通用部分和稀疏私有部分,采用双通道编码器分别学习稀疏的本地知识与共享的低秩全局知识。同时,在通信周期间进行多轮本地随机梯度下降,并集成高效压缩技术。实验表明,该方法在16个数据集上均取得最优分类性能。具体而言,相较于最先进的方法FedStar(以GIN为基础模型),在跨数据集设置CHEM上,准确率提升5.64%,通信比特数减少18.58倍,通信时间减少1.65倍。
原文摘要 · Abstract (English)
Federated graph learning (FGL) has gained significant attention for enabling heterogeneous clients to process their private graph data locally while interacting with a centralized server, thus maintaining privacy. However, graph data on clients are typically non-IID, posing a challenge for a single model to perform well across all clients. Another major bottleneck of FGL is the high cost of communication. To address these challenges, we propose a communication-efficient personalized federated graph learning algorithm, CEFGL. Our method decomposes the model parameters into low-rank generic and sparse private models. We employ a dual-channel encoder to learn sparse local knowledge in a personalized manner and low-rank global knowledge in a shared manner. Additionally, we perform multiple local stochastic gradient descent iterations between communication phases and integrate efficient compression techniques into the algorithm. The advantage of CEFGL lies in its ability to capture common and individual knowledge more precisely. By utilizing low-rank and sparse parameters along with compression techniques, CEFGL significantly reduces communication complexity. Extensive experiments demonstrate that our method achieves optimal classification accuracy in a variety of heterogeneous environments across sixteen datasets. Specifically, compared to the state-of-the-art method FedStar, the proposed method (with GIN as the base model) improves accuracy by 5.64\% on cross-datasets setting CHEM, reduces communication bits by a factor of 18.58, and reduces the communication time by a factor of 1.65.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。