论文标题
具有深度平衡模型的统一标签的图形神经网络
Unifying Label-inputted Graph Neural Networks with Deep Equilibrium Models
论文作者
论文摘要
图形神经网络(GNN)在非欧盟数据中学习的成功引起了许多子主题,例如标签输入的GNN(LGNN)和隐式GNN(IGNN)。 LGNN,明确输入GNN中的监督信息(又称标签),集成了标签传播以实现卓越的性能,但与其传播距离和适应性之间的困境。 IGNN,通过迭代网络无限时间来输出平衡点,利用整个图中的信息来捕获长期依赖性,但其网络的限制以确保平衡的存在。这项工作通过在IGNN理论中解释LGNN并将流行的LGNN缩小为INGN的形式来统一两个子域。统一促进了两个子域之间的交流,并激发了更多的研究。具体而言,引入了IGNN的隐式分化,以通过恒定记忆区分其无限范围标签的传播,从而使传播既遥远又适应。此外,LGNN的蒙版标签策略已被证明能够以网络敏捷的方式保证IGNN的良好性,从而使其网络更加复杂,从而更具表现力。提出了结合LGNN和IGNN的优势,提出了标签输入的隐式GNN(LI-GNN)。它可以广泛应用于任何特定的GNN,以提高其性能。在两个合成和六个现实世界数据集上进行的节点分类实验证明了其有效性。代码可从https://github.com/cf020031308/li-gnn获得
The success of Graph Neural Networks (GNN) in learning on non-Euclidean data arouses many subtopics, such as Label-inputted GNN (LGNN) and Implicit GNN (IGNN). LGNN, explicitly inputting supervising information (a.k.a. labels) in GNN, integrates label propagation to achieve superior performance, but with the dilemma between its propagating distance and adaptiveness. IGNN, outputting an equilibrium point by iterating its network infinite times, exploits information in the entire graph to capture long-range dependencies, but with its network constrained to guarantee the existence of the equilibrium. This work unifies the two subdomains by interpreting LGNN in the theory of IGNN and reducing prevailing LGNNs to the form of IGNN. The unification facilitates the exchange between the two subdomains and inspires more studies. Specifically, implicit differentiation of IGNN is introduced to LGNN to differentiate its infinite-range label propagation with constant memory, making the propagation both distant and adaptive. Besides, the masked label strategy of LGNN is proven able to guarantee the well-posedness of IGNN in a network-agnostic manner, granting its network more complex and thus more expressive. Combining the advantages of LGNN and IGNN, Label-inputted Implicit GNN (LI-GNN) is proposed. It can be widely applied to any specific GNN to boost its performance. Node classification experiments on two synthesized and six real-world datasets demonstrate its effectiveness. Code is available at https://github.com/cf020031308/LI-GNN