核心概念
本文提出了一種名為 FCLG 的新型聯邦圖嵌入方法,該方法利用雙層對比學習機制(內部對比和外部對比)來學習分散式圖數據中的圖級表徵,並有效解決了聯邦學習中的非獨立同分布問題,在圖級聚類任務中展現出優於現有方法的性能。
摘要
文獻資訊
- 標題:圖級表徵的聯邦對比學習
- 作者:Xiang Li, Gagan Agrawal, Rajiv Ramnath, Ruoming Jin
研究目標
本研究旨在解決在聯邦學習環境下,如何在沒有圖級標籤的情況下學習圖級表徵,並克服由於數據異構性帶來的挑戰。
方法
本研究提出了一種名為 FCLG 的新型聯邦圖嵌入方法,該方法採用雙層對比學習機制:
- 內部對比學習: 在每個客戶端上,利用圖擴散技術生成增強圖,並通過對比不同視圖下的圖級表徵來學習每個圖的獨特特徵。
- 外部對比學習: 在服務器端,通過對比來自不同客戶端的局部模型和全局模型的圖級表徵,來減輕局部模型漂移,並提高全局模型的泛化性能。
主要發現
- FCLG 在四個公開圖數據集上的圖級聚類任務中,相較於其他聯邦學習基線方法,取得了顯著的性能提升。
- 在非獨立同分布數據集上,FCLG 的聚類準確率提升了 2-7%;在獨立同分布數據集上,提升了 2-4%。
- 實驗結果表明,FCLG 對數據異構性和客戶端數量具有良好的魯棒性。
主要結論
FCLG 是一種有效的聯邦圖嵌入方法,能夠在無監督的情況下學習高質量的圖級表徵,並有效解決聯邦學習中的非獨立同分布問題。
意義
本研究為聯邦圖學習提供了一種新的思路,並為解決數據隱私和安全問題提供了一種可行的解決方案。
局限性和未來研究方向
- 未來可以進一步探索更先進的圖擴散技術和對比學習策略,以進一步提高模型性能。
- 可以將 FCLG 應用於其他圖級任務,例如圖分類和鏈路預測。
統計資料
在非獨立同分布數據集 PROTEINS 上,FCLG 的聚類準確率提升了 7%。
在非獨立同分布數據集 ENZYMES 和 DHFR 上,FCLG 的聚類準確率提升了約 5%。
在非獨立同分布數據集 NCI1 上,FCLG 的聚類準確率提升了 2%。