本文研究如何在異質性圖上提高圖神經網絡(GNN)的泛化能力。異質性圖是指相似節點之間的連接較少,這對GNN的性能造成挑戰,因為GNN的優秀表現通常來自於聚合同質性信息。
為了在異質性圖上找到同質性,需要增加GNN的深度以擴大範疇(受體場)。但是,統一擴大範疇會導致性能下降,因為現實世界的圖通常在節點之間存在同質性差異。理想的方法是個性化範疇,允許節點有不同的範疇大小。
現有的個性化範疇方法通常在GNN架構中添加節點自適應權重,雖然表達能力更強,但仍然會嚴重過擬合。我們提出將個性化範疇建模為一個單獨的範疇分類問題,以克服GNN在節點分類中的過擬合問題。具體來說,我們預測每個節點的最佳GNN深度。我們的理論和實證分析表明,準確預測深度可以顯著提高泛化能力。
我們進一步提出了Adaptive Scope (AS),這是一種輕量級的多層感知機方法,只參與GNN推理階段。AS編碼結構模式並預測深度,以選擇每個節點最佳的模型進行預測。實驗結果顯示,AS對於各種GNN架構和數據集都具有很強的靈活性,同時顯著提高了準確率。
Til et annet språk
fra kildeinnhold
arxiv.org
Viktige innsikter hentet fra
by Gangda Deng,... klokken arxiv.org 09-12-2024
https://arxiv.org/pdf/2409.06998.pdfDypere Spørsmål