arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.01249cs.LGstat.ML
收藏

基于非线性谱滤波器的图等变机器学习

Equivariant Machine Learning on Graphs with Nonlinear Spectral Filters

Viterbi Faculty of Electrical and Computer Engineering, Technion(以色列理工学院维特比电气与计算机工程学院);Faculty of Mathematics, Technion(以色列理工学院数学学院)

Ya-Wei Eileen Lin, Ronen Talmon, Ron Levie

AI总结:

针对标准谱图神经网络激活函数破坏图函数平移等变性的问题,提出基于可迁移新型谱域的非线性谱滤波器,该滤波器具备通用逼近性,在节点和图分类任务上性能优于现有谱GNN。

摘要:

等变机器学习是一种设计符合问题对称性的深度学习模型的方法,旨在降低模型复杂度并提升泛化能力。本文聚焦于将作为图像卷积网络基础的平移等变性,扩展到通用图的场景。与图像不同,图没有自然的域平移概念,因此我们将图函数平移视为对称群:即与图平移算子对易的酉算子。值得注意的是,这类对称性作用于信号空间而非直接作用于空间域。我们发现,标准谱图神经网络(GNN)的每个线性滤波器层都与图函数平移对易,但激活函数会破坏这种对称性。为此,我们提出了对图函数平移完全等变的非线性谱滤波器(NLSFs),并证明其具有通用逼近特性。所提出的NLSFs基于一种可在不同图之间迁移的新型谱域形式。我们在节点分类和图分类基准测试中,验证了NLSFs相较于现有谱GNN的更优性能。

英文摘要:

Equivariant machine learning is an approach for designing deep learning models that respect the symmetries of the problem, with the aim of reducing model complexity and improving generalization. In this paper, we focus on an extension of shift equivariance, which is the basis of convolution networks on images, to general graphs. Unlike images, graphs do not have a natural notion of domain translation. Therefore, we consider the graph functional shifts as the symmetry group: the unitary operators that commute with the graph shift operator. Notably, such symmetries operate in the signal space rather than directly in the spatial space. We remark that each linear filter layer of a standard spectral graph neural network (GNN) commutes with graph functional shifts, but the activation function breaks this symmetry. Instead, we propose nonlinear spectral filters (NLSFs) that are fully equivariant to graph functional shifts and show that they have universal approximation properties. The proposed NLSFs are based on a new form of spectral domain that is transferable between graphs. We demonstrate the superior performance of NLSFs over existing spectral GNNs in node and graph classification benchmarks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑