May, 2023

利用本地化提高图神经网络的表达能力

TL;DR本文提出局部 Weisfeiler-Leman(WL)算法以提高表达能力和降低计算复杂度,并解决所有 k 的局部 k-WL 的子图计数问题。研究证明,本地 k-WL 可以表达更多模式,但在表达能力上最多与 (k + 1) -WL 一样,比应用于整个图的 k-WL 更具有时间和空间效率。用局部 k-WL 等效的两个图的子图和诱导子图的计数是不变的。该算法也引入了局部和递归的 Layer k-WL 以及一个可扩展片段技术,可用于任意 k,以保证使用仅 1-WL 的确切计数,同时比 Papp 和 Wattenhofer(2022a)提到的 GNN 层次更具表达能力。