Konsep Inti
FedGraph 是一個為了解決聯邦圖學習中實際部署和基準測試問題而設計的研究庫,它支持各種最先進的演算法,並包含用於系統性能評估的分析工具,重點關注訓練期間的通信和計算成本。
論文資訊
Yuhang Yao, Yuan Li, Xinyi Fan, Junhao Li, Kay Liu, Weizhao Jin, Srivatsan Ravi, Philip S. Yu, & Carlee Joe-Wong. (2024). FedGraph: A Research Library and Benchmark for Federated Graph Learning. arXiv preprint arXiv:2410.06340.
研究目標
本研究旨在開發一個名為 FedGraph 的研究庫,用於解決聯邦圖學習中實際部署和基準測試的挑戰。
方法
FedGraph 建立在 PyTorch Geometric (PyG)、Ray 和 PyTorch 之上,旨在簡化聯邦環境下的圖神經網路訓練。它支持模擬和真實聯邦環境下的各種聯邦訓練方法,並支持客戶端與中央伺服器之間的通信,以進行模型更新和資訊聚合。
主要發現
FedGraph 支持一系列最先進的聯邦圖學習方法,包括 FedAvg、FedProx、GCFL、FedSage+ 和 FedGCN。
它提供了分析工具,用於評估系統性能,重點關注訓練期間的通信和計算成本。
FedGraph 支持三種主要的圖問題:節點分類、鏈路預測和圖分類,並為每個任務提供各種演算法和資料集。
主要結論
FedGraph 是一個綜合性的研究庫,可以促進聯邦圖學習的實際應用和未來演算法的設計。
意義
本研究通過提供一個易於使用且功能豐富的平台,為聯邦圖學習領域做出了貢獻,促進了該領域的研究和開發。
局限性和未來研究
未來的研究方向包括支持更多種類的圖神經網路模型、探索更先進的聯邦優化演算法,以及在更大規模的真實世界資料集上評估 FedGraph 的性能。
Statistik
FedGCN 在 Citeseer 資料集上使用 5 個客戶端進行訓練,隨著資料分佈變得更加獨立同分佈,預訓練通信時間和訓練時間會增加。