图信号处理(GSP)提供了一个强大的数学框架,用于分析在图节点上定义的函数。GSP基于图拉普拉斯算子和谱图理论等概念,使我们能够将图操作,特别是GNN内部的操作,视为信号滤波,这与经典信号处理处理时间序列或图像数据的方式非常相似。这种观点对于理解谱GNN以及分析过平滑等现象尤其有帮助。
图上的信号
在GSP中,一个“图信号”只是一个为图中每个节点赋值的函数。如果有一个包含N=∣V∣个节点的图G=(V,E),一个图信号x是RN中的一个向量,其中xi是节点vi处的信号值。
x=[x1,x2,…,xN]T∈RN
这个定义与我们在GNN中使用的节点特征直接一致。所有节点上的单个特征可以被视为一个图信号。因此,节点特征矩阵X∈RN×F可以被看作是F个不同图信号的集合,每个信号对应一个特征维度。
图傅里叶变换
与将时域信号分解为频率分量的经典傅里叶变换类似,图傅里叶变换(GFT)将图信号分解为与图结构相关的分量,特别是其“频率”。这些频率与图拉普拉斯算子L的特征值相关联。
回想一下,归一化图拉普拉斯算子L=I−D−1/2AD−1/2(或未归一化的L=D−A)是一个对称半正定矩阵。它具有一套完整的正交特征向量U=[u1,u2,…,uN],以及对应的实数非负特征值0=λ1≤λ2≤⋯≤λN,存储在对角矩阵Λ中。
定义
图信号x的GFT定义为其在拉普拉斯算子特征向量上的投影:
x^=UTx
这里,x^=[x^1,x^2,…,x^N]T是傅里叶系数向量。每个系数x^i=uiTx衡量信号x与对应特征向量ui的一致性。
逆图傅里叶变换(iGFT)根据其谱系数重构原始信号:
x=Ux^=i=1∑Nx^iui
这表明信号x可以表示为拉普拉斯算子特征向量的线性组合,并由傅里叶系数x^i加权。
频率解释
拉普拉斯算子的特征值λi量化了它们对应特征向量ui在图上的“平滑度”或“变化”。
- 小特征值(λi≈0)对应低频率。相关的特征向量ui在连接的节点间变化缓慢;它们代表平滑信号。与λ1=0相关的特征向量u1通常是一个常数向量(对于连通图),代表最平滑的信号(零变化)。
- 大特征值对应高频率。相关的特征向量ui在相邻节点间快速振荡,代表具有高变化的信号。
因此,GFT系数x^i表示信号在与λi相关频率处分量的强度。集中在低频率(小λi对应大x^i)的信号在图上是平滑的,而具有显著高频分量的信号在邻居之间变化剧烈。
图拉普拉斯算子特征值的典型分布,显示了从零(最平滑)到代表更高频率的较大值的范围。
图滤波
图滤波涉及修改图信号的谱系数。图滤波器通过放大或衰减由拉普拉斯算子特征值定义的特定频率分量来操作。
谱域滤波
线性、移不变图滤波器由一个函数h(λ)定义,称为滤波器的频率响应。在谱域中将此滤波器应用于信号x,意味着将每个傅里叶系数x^i乘以滤波器在对应频率λi处的响应h(λi)。
如果y是滤波后的信号,其GFT y^由下式给出:
y^i=h(λi)x^i
以矩阵形式,这可以写为:
y^=h(Λ)x^
其中h(Λ)是一个对角矩阵,其对角线上是h(λi):h(Λ)=diag(h(λ1),h(λ2),…,h(λN))。
空间域等效性(卷积)
要查看滤波器如何在节点(空间)域中直接作用于信号x,我们可以使用iGFT将谱滤波操作转换回去:
y=Uy^=Uh(Λ)x^=Uh(Λ)UTx
这定义了空间域中的图卷积操作:
y=(Uh(Λ)UT)x=Hx
矩阵H=Uh(Λ)UT表示节点域中的滤波器操作。
这里的一个主要挑战是直接计算此操作需要:
- 计算拉普拉斯算子的完整特征分解(U,Λ),其代价是O(N3)。
- 执行涉及U和UT的矩阵乘法,每个信号向量的代价是O(N2)。
- 由此产生的滤波器矩阵H通常是稠密的,这意味着节点vi处的滤波值yi取决于所有其他节点vj处的信号值xj,这使得该操作是非局部的,对于大型图而言计算成本高昂(O(N2)的应用成本)。
连接图滤波器和GNN层
GSP框架为理解谱GNN提供了基础。这些模型明确或隐式地定义图滤波器并学习其参数。
作为滤波器的谱GNN
早期的谱GNN,例如Bruna等人(2014)提出的模型,直接对滤波器响应h(Λ)进行参数化。一个谱GNN层可以被视为将可学习的图滤波器Hθ=Uhθ(Λ)UT应用于输入节点特征(信号)X(k−1):
X(k)=σ(Uhθ(Λ)UTX(k−1))
这里,hθ(Λ)是一个对角矩阵,其项取决于可学习的参数θ。非线性函数σ随后按元素应用。
为了克服通用谱滤波器Hθ的计算成本和非局部性问题,后续工作集中于设计能够高效计算且空间局部化的滤波器hθ(λ)。
- 多项式滤波器: 一种常见方法是使用特征值的多项式来近似滤波器响应hθ(λ),即hθ(λ)≈∑p=0Pθpλp。由于UΛpUT=(UΛUT)p=Lp,这导致了涉及拉普拉斯算子幂次的滤波器操作:
y≈(p=0∑PθpLp)x
将Lp应用于x只涉及p跳内的节点,这使得滤波器是P-局部化的。ChebNet(使用切比雪夫多项式)和简化的图卷积网络(GCN)(使用一阶近似,P=1)是属于此类别的重要例子。它们学习多项式系数θp。
滤波器属性与GNN行为
GSP有助于分析滤波器结构如何影响GNN行为:
- 低通滤波器: 许多标准GNN层,例如GCN,充当低通滤波器。它们通过平均来自邻居的信息来平滑节点特征。这对于节点分类等任务通常是有益的,因为邻近节点往往具有相似的标签。
- 过平滑: 在许多GNN层中重复应用低通滤波可能导致过平滑。信号中的高频分量(可能包含区分信息)会逐渐衰减。最终,节点表示可能变得过于相似,从而妨碍性能。GSP提供工具,通过分析滤波器的频率响应来量化这种效应。
- 带通或高通滤波器: 尽管作为主要机制不太常见,但设计能够保留或强调更高频率的滤波器可能与局部变化或邻居之间差异很重要的任务相关。
对GNN设计和分析的影响
GSP视角带来多项益处:
- 原理性理解: 它为谱GNN架构为何有效以及它们与经典信号处理的关系提供了理论基础。
- 分析工具: 它允许通过检查层的频率响应来分析GNN属性,例如过平滑。我们可以研究不同的消息传递方案如何隐式地过滤图信号。
- 滤波器设计: GSP原理可以指导设计具有特定所需属性的新GNN层(例如,避免过度平滑或捕获与任务相关的特定频段的层)。
- 连接谱域与空间域: 尽管空间GNN(如GraphSAGE、GAT)并非直接通过谱滤波器定义,但其操作通常可以在GSP框架内进行近似分析,有助于理解它们在信号平滑或特征变换方面的行为。
尽管由于计算成本高昂,任意谱滤波器的直接实现对于大型图仍然具有挑战,但图信号处理的框架对于理解许多高级GNN架构背后的机制以及设计下一代图学习模型是不可或缺的。它提供了一个视角,通过该视角,消息传递的聚合和更新步骤可以被解释为对图上信息的复杂滤波操作。