图傅立叶变换的理解及例子

前言

本文是图神经网络系列的一部分内容，主要对图傅立叶变换 (Graph Fourier Transform,GFT) 进行讲解，用公式梳理了GFT的流程并举一个例子便于理解。

其他关于更全面的知识讲解，见博客《关于谱图理论-图傅里叶变换-谱卷积等谱图领域知识的理解》。

GFT流程

GFT公式为：
$$\hat{f}=U^{T}f$$，其中 $U$ 是单位特征向量矩阵， $U^T$ 的每一行是单位特征向量; $f:V\to \mathbb{R}$ 是定义在图 $G$ 顶点上的信号，称之为图信号。

展开为矩阵的形式：
$$\left(\begin{array}{c}\hat{f}({\lambda }_1)\\ \hat{f}({\lambda }_2)\\ ⋮\\ \hat{f}({\lambda }_l)\end{array}\right)=\left[\begin{array}{cccc}{u}_1(1)& u_1(2)& \cdots & u_1(n)\\ u_2(1)& u_2(2)& \cdots & u_2(n)\\ ⋮& ⋮& & ⋮\\ u_n(1)& u_2(2)& \cdots & u_n(n)\end{array}\right]\cdot \left[\begin{array}{c}f(1)\\ f(2)\\ ⋮\\ f(n)\end{array}\right]$$ 对等式右边的矩阵进行计算，可得：
$$\left(\begin{array}{c}\hat{f}({\lambda }_1)\\ \hat{f}({\lambda }_2)\\ ⋮\\ \hat{f}({\lambda }_n)\end{array}\right)=\left[\begin{array}{c}{\sum }_{i=1}^{n}f(i)u_1(i)\\ {\sum }_{i=1}^{n}f(i)u_2(i)\\ ⋮\\ {\sum }_{i=1}^{n}f(i)u_n(i)\end{array}\right]$$ 可以理解为图信号 $f$ 的傅里叶变换的每个元素 $\hat{f}({\lambda }_l)$，都等于相应特征值下对应特征向量 $u_l$ 与 图信号 $f$ 的内积。

GFT例子

设图 $G$ 的形式如下：

图	度矩阵	邻接矩阵	拉普拉斯矩阵
	$\left[\begin{array}{cccc}2& 0& 0& 0\\ 0& 2& 0& 0\\ 0& 0& 3& 0\\ 0& 0& 0& 1\end{array}\right]$	$\left[\begin{array}{cccc}0& 1& 1& 0\\ 1& 0& 1& 0\\ 1& 1& 0& 1\\ 0& 0& 1& 0\end{array}\right]$	$\left[\begin{array}{cccc}2& -1& -1& 0\\ & & & \\ -1& 2& -1& 0\\ & & & \\ -1& -1& 3& -1\\ & & & \\ 0& 0& -1& 1\end{array}\right]$

• 创建拉普拉斯矩阵 $L$

import numpy as np
L = np.array(
        [[ 2., -1., -1.,  0.],
        [-1.,  2., -1.,  0.],
        [-1., -1.,  3., -1.],
        [ 0.,  0., -1.,  1.]])

• 求 $L$ 对应的特征值和特征向量
  即求 $({\lambda }_1,{\lambda }_2,{\lambda }_3,{\lambda }_4)$ 和 $U$

vals_ori,vecs_ori = np.linalg.eig(L)
print(vals_ori.round(3))
print(vecs_ori.round(3))

##输出
"""
[4. 3. 0. 1.]
[[-0.289  0.707 -0.5   -0.408]
 [-0.289 -0.707 -0.5   -0.408]
 [ 0.866  0.    -0.5    0.   ]
 [-0.289 -0.    -0.5    0.816]]
"""

由于拉普拉斯图谱(Laplacian Graph Spectrum) 要求 $0={\lambda }_0\le {\lambda }_1\le \cdots \le {\lambda }_{N-1}$，单位特征向量矩阵 $U$ 按照这个顺序排列，因此我们可以得到按从小到大排序的特征值序列及对应的单位特征向量矩阵。

# 求特征值从小到大排序的索引
lambda_order_index = [x[0] for x in sorted(enumerate(vals_ori), key=lambda x:x[-1])]
#　调整特征向量矩阵中的特征向量位置
U = (vecs_ori.T)[lambda_order_index].T
print(U.round(3))
## 输出
"""
[[-0.5   -0.408  0.707 -0.289]
 [-0.5   -0.408 -0.707 -0.289]
 [-0.5    0.     0.     0.866]
 [-0.5    0.816 -0.    -0.289]]
"""

• 图信号 $f$
  由于图信号 $f$ 是关于图顶点的函数，为简单期间，$f$ 是将顶点映射为顶点的值的函数，即：
  $$f=\left[\begin{array}{c}f(1)\\ f(2)\\ f(3)\\ f(4)\end{array}\right]=\left[\begin{array}{c}1\\ 2\\ 3\\ 4\end{array}\right]$$所以，$f$ 的代码如下

f_vec=np.array([[1],[2],[3],[4]])

• 求 $f$ 的GFT
  即实现：$\hat{f}=U^{T}f$

f_hat = np.matmul(U.T, f_vec)
print(f_hat.round(3))
"""
[[-5.   ]
 [ 2.041]
 [-0.707]
 [ 0.577]]
"""

$■$