md_files/科研/动态图神经网络.md

23 KiB
Raw Blame History

动态图神经网络

如何对GAT的权重$W$)和注意力参数($a$)进行增量更新(邻居偶尔变化)

1. 核心思想

  • 局部更新:邻居变化的节点及其直接邻域的权重和注意力参数需要调整,其他部分冻结。
  • 梯度隔离:反向传播时,仅计算受影响节点的梯度,避免全局参数震荡。

2. 数学实现步骤

(1) 识别受影响的节点

设邻居变化后的新邻接矩阵为 $\tilde{A}$,原邻接矩阵为 $A$,受影响节点集合 \mathcal{V}_{\text{affected}} 包括:

  • 新增或删除边的两端节点(直接受影响)。
  • 这些节点的1-hop邻居间接受影响根据GAT层数决定

(2) 损失函数局部化

仅对 \mathcal{V}_{\text{affected}} 中的节点计算损失:


\mathcal{L}_{\text{incremental}} = \sum_{i \in \mathcal{V}_{\text{affected}}} \ell(y_i, \hat{y}_i)

其中 \ell 为交叉熵损失,y_i 为标签,\hat{y}_i 为模型输出。

(3) 梯度计算与参数更新

  • 梯度掩码
    反向传播时,非受影响节点的梯度强制置零:

    
    \nabla_{W,a} \mathcal{L}_{\text{incremental}} = \left\{ 
    \begin{array}{ll}
    \nabla_{W,a} \ell(y_i, \hat{y}_i) & \text{if } i \in \mathcal{V}_{\text{affected}} \\
    0 & \text{otherwise}
    \end{array} \right.
    
  • 参数更新
    使用优化器如Adam仅更新有梯度的参数

    
    W \leftarrow W - \eta \nabla_W \mathcal{L}_{\text{incremental}}, \quad a \leftarrow a - \eta \nabla_a \mathcal{L}_{\text{incremental}}
    

    其中 \eta 为较小的学习率(防止过拟合)。

(4) 注意力权重的动态适应

GAT的注意力机制会自动适应新邻居


\alpha_{ij} = \text{softmax}\left(\text{LeakyReLU}\left(a^T [W h_i \| W h_j]\right)\right)

由于 Wa 已局部更新,新邻居 j \in \tilde{\mathcal{N}}(i) 的权重 \alpha_{ij} 会重新计算。

3. 适用场景

  • 低频变化:如社交网络每天新增少量边、论文引用网络月度更新。
  • 局部变化:每次变化仅影响图中少量节点(<10%)。

若邻居高频变化如秒级更新需改用动态GNN如TGAT或时间序列建模。

EvolveGCN

EvolveGCN-H

1. EvolveGCN-H核心思想

EvolveGCN-H 通过 GRU门控循环单元 动态更新 GCN 每一层的权重矩阵 $W_t^{(l)}$,将权重矩阵视为 GRU 的 隐藏状态,并利用当前时间步的 节点嵌入(特征) 作为输入来驱动演化。
关键特点

  • 输入依赖:利用节点嵌入 H_t^{(l)} 指导权重更新。
  • 时序建模:通过 GRU 隐式捕捉参数演化的长期依赖。

2. 动态更新流程(以第 l 层为例)

输入

  1. 当前节点嵌入矩阵 $H_t^{(l)} \in \mathbb{R}^{n \times d}$
  2. 上一时间步的权重矩阵 $W_{t-1}^{(l)} \in \mathbb{R}^{d \times d'}$
  3. 邻接矩阵 $A_t \in \mathbb{R}^{n \times n}$

输出

  1. 更新后的权重矩阵 $W_t^{(l)} \in \mathbb{R}^{d \times d'}$。
  2. 下一层节点嵌入 $H_t^{(l+1)} \in \mathbb{R}^{n \times d'}$。

3. 动态更新示意图

Time Step t-1                          Time Step t
+-------------------+                 +-------------------+
|  Weight Matrix    |                 |  Weight Matrix    |
|    W_{t-1}^{(l)}  |  --(GRU更新)--> |    W_t^{(l)}      |
+-------------------+                 +-------------------+
         ^                                      ^
         |                                      |
+-------------------+                 +-------------------+
|  Node Embeddings  |                 |  Node Embeddings |
|    H_t^{(l)}      |  --(GCN计算)--> |    H_t^{(l+1)}    |
+-------------------+                 +-------------------+
         ^                                      ^
         |                                      |
+-------------------+                 +-------------------+
|  邻接矩阵 A_t     |                 |  邻接矩阵 A_{t+1} |
| (显式输入)        |                 | (下一时间步输入)  |
+-------------------+                 +-------------------+

\begin{align*}
W_t^{(l)} &<= H_t^{(l)} + W_{t-1}^{(l)} \\
H_t^{(l+1)} &<= A_t + H_t^{(l)} + W_t^{(l)}
\end{align*}

4. 具体步骤分解

步骤 1节点嵌入聚合Summarize

由于 GRU 的输入需与隐藏状态 W_{t-1}^{(l)} 的列维度匹配(即 $d'$),需将 H_t^{(l)}n \times d 压缩为 d' \times d


Z_t = \text{Summarize}(H_t^{(l)}, d')

将会舍弃部分节点

实现方式(论文方案):

  1. 计算得分:

    
    y_t = H_t^{(l)} p / \|p\| \quad (p \in \mathbb{R}^d \text{为可学习参数})
    

    学一个“打分器”参数 $p$$H_t^{(l)} p$相当于对每个节点的嵌入向量和 $p$做点积,得到一个分数。比如在社交网络中,p 可能代表“活跃度”,得分高的用户更活跃。

  2. 选取 Top-d' 个节点(按 y_t 排序),加权求和:

    
    Z_t = [H_t^{(l)} \circ \tanh(y_t)]_{\text{top-}d'} \quad (\circ \text{为逐元素乘})
    
    • 输出 $Z_t \in \mathbb{R}^{d' \times d}$。

举个例子

假设:

  • 有3个节点$n=3$),嵌入维度 $d=2$选Top-2个节点$d'=2$)。

  • 节点嵌入:

    
    H_t^{(l)} = \begin{bmatrix} 1 & 0.5 \\ 0.3 & 2 \\ -1 & 1 \end{bmatrix}, \quad p = [1, 0]
    

    p 只关注嵌入的第一维,比如“用户发帖数量”)

  1. 计算分数

    
     y_t = H_t^{(l)} p = [1 \cdot 1 + 0.5 \cdot 0, \ 0.3 \cdot 1 + 2 \cdot 0, \ -1 \cdot 1 + 1 \cdot 0] = [1, 0.3, -1] 
    

    Top-2节点是第1、第2个节点分数1和0.3)。

  2. 加权聚合

    
     Z_t = \begin{bmatrix} [1, 0.5] \circ \tanh(1) \\ [0.3, 2] \circ \tanh(0.3) \end{bmatrix} = \begin{bmatrix} 0.76 & 0.38 \\ 0.09 & 0.58 \end{bmatrix} 
    

    (假设 \tanh(1) \approx 0.76, $\tanh(0.3) \approx 0.29$

  3. 输出Z_t2 \times 2 矩阵可以直接喂给GRU。

步骤 2GRU 更新权重矩阵

W_t^{(l)} = \text{GRU}(Z_t^T, W_{t-1}^{(l)})

标准GRU的输入、隐藏、输出都是向量但这里都是矩阵

当前时间步的输入:Z_t^T

上一时间步的隐藏状态:W_{t-1}^{(l)}

更新隐藏状态: W_t^{(l)}

标准GRU 论文中的矩阵GRU 作用
W_{xz} (输入→更新门) W_Z 将当前输入 Z_t^T 映射到更新门。
W_{hz} (隐藏→更新门) U_Z 将上一隐藏状态 W_{t-1}^{(l)} 映射到更新门。
b_z (更新门偏置) B_Z 更新门的偏置项。
W_{xh} (输入→候选状态) W_H 将当前输入 Z_t^T 映射到候选状态。
W_{hh} (隐藏→候选状态) U_H 将调制后的隐藏状态 (R_t \circ W_{t-1}^{(l)}) 映射到候选状态。
b_h (候选状态偏置) B_H 候选状态的偏置项。

GRU 的矩阵形式计算

  1. 重置门 $R_t$:控制历史信息的遗忘程度

    
    R_t = \sigma(Z_t^T W_Z + W_{t-1}^{(l)} U_Z + B_Z)
    
  2. 更新门 $Z_t$:控制新旧状态混合比例

    
    U_t = \sigma(Z_t^T W_U + W_{t-1}^{(l)} U_U + B_U)
    
  3. 候选状态 $\widetilde{W}_t$

    
    \widetilde{W}_t = \tanh(Z_t^T W_H + (R_t \circ W_{t-1}^{(l)}) U_H + B_H)
    
  4. 最终权重更新

    
    W_t^{(l)} = (1 - U_t) \circ W_{t-1}^{(l)} + U_t \circ \widetilde{W}_t
    
    • 输出 $W_t^{(l)} \in \mathbb{R}^{d \times d'}$。
步骤 3GCN 生成下一层嵌入

使用更新的 W_t^{(l)} 执行标准 GCN 操作:


H_t^{(l+1)} = \sigma\left(\widehat{A}_t H_t^{(l)} W_t^{(l)}\right)
  • \widehat{A}_t 为归一化邻接矩阵(含自环)。

5. 关键设计细节

  1. 权重共享
    • 所有时间步共享同一 GRU 的参数($W_, U_, B_*$),确保模型尺寸不随时间增长。
  2. 层独立性
    • 每一层 GCN 的权重矩阵独立演化(不同层有各自的 GRU
  3. 特征与结构的协同
    • 节点嵌入 H_t^{(l)} 既包含特征信息,也隐含历史结构信息(通过多层 GCN 传播),因此 GRU 能间接感知结构变化。

6. 所需提前训练的权重

参数类型 符号 维度 作用
GCN 初始权重 W_0^{(l)} \mathbb{R}^{d \times d'} 初始时刻各层 GCN 的初始参数
GRU 输入变换矩阵 W_Z, W_R, W_H \mathbb{R}^{d \times d'} 将输入 Z_t^T 映射到门控
GRU 隐藏变换矩阵 U_Z, U_R, U_H \mathbb{R}^{d' \times d'} W_{t-1}^{(l)} 映射到门控
GRU 偏置项 B_Z, B_R, B_H \mathbb{R}^{d'} 门控和候选状态的偏置
Summarize 参数 p \mathbb{R}^d 对特征进行打分,不同层$p$不一样
任务相关参数 例如 MLP 权重 任务相关 链接预测、节点分类等输出层

EvolveGCN-O

1. 核心思想

EvolveGCN-O 通过 LSTM 直接演化 GCN 的权重矩阵 $W_t^{(l)}$,将权重矩阵视为 LSTM的输出(下一时间步的输入),不依赖节点嵌入
关键特点

  • 结构驱动:仅通过历史权重 W_{t-1}^{(l)} 预测当前权重,完全基于图结构的动态变化。
  • 轻量化:无需处理节点嵌入,计算效率更高。

2. 动态更新流程(第$l$层)

输入

  1. 上一时间步权重 W_{t-1}^{(l)} \in \mathbb{R}^{d \times d'}
  2. 邻接矩阵 $A_t \in \mathbb{R}^{n \times n}$仅用于GCN计算

输出

  1. 更新后权重 W_t^{(l)} \in \mathbb{R}^{d \times d'}
  2. 下一层节点嵌入 H_t^{(l+1)} \in \mathbb{R}^{n \times d'}

3.具体步骤分解

步骤1LSTM更新权重矩阵

矩阵版LSTM计算

  1. 遗忘门:
    F_t = \sigma(W_F W_{t-1}^{(l)} + U_F C_{t-1} + B_F)
  2. 输入门:
    I_t = \sigma(W_I W_{t-1}^{(l)} + U_I C_{t-1} + B_I)
  3. 候选状态:
    \widetilde{C}_t = \tanh(W_C W_{t-1}^{(l)} + U_C C_{t-1} + B_C)
  4. 细胞状态更新:
    C_t = F_t \circ C_{t-1} + I_t \circ \widetilde{C}_t
  5. 输出门:
    O_t = \sigma(W_O W_{t-1}^{(l)} + U_O C_{t-1} + B_O)
  6. 最终权重输出:
    W_t^{(l)} = O_t \circ \tanh(C_t)
步骤2GCN生成下一层嵌入

H_t^{(l+1)} = \sigma(\widehat{A}_t H_t^{(l)} W_t^{(l)})

4. 与EvolveGCN-H对比

特性 EvolveGCN-H EvolveGCN-O
RNN类型 GRU LSTM
演化依据 节点嵌入+历史权重 仅历史权重
计算复杂度 需Summarize
适用场景 特征动态性强(如社交网络) 结构变化主导(如路由器拓扑)

TGAT

Bochner定理

Bochner定理为TGAT的连续时间编码提供了理论基础使其能够将任意时间差映射为可学习的向量表示从而在动态图中有效捕捉时序依赖。这一方法超越了传统离散编码的局限性是TGAT的核心创新之一。

时间编码的构造

1Bochner 定理的数学形式

根据Bochner定理任意正定时间核函数 f(\Delta t) 可表示为:


f(\Delta t) = \mathbb{E}_{\omega \sim \mu} \left[ e^{i \omega \Delta t} \right],

其中 e^{i \omega \Delta t} = \cos(\omega \Delta t) + i \sin(\omega \Delta t) 是复指数函数。

2蒙特卡洛近似 & 实值化

由于直接计算期望复杂TGAT 用蒙特卡洛采样近似:

  1. 从分布 \mu 中采样 k 个频率 $\omega_1, \omega_2, \ldots, \omega_k$。

  2. 用这些频率构造实值编码(取复指数的实部和虚部,即 \cos 和 $\sin$

    
    \phi(\Delta t) = \sqrt{\frac{1}{k}} \left[ \cos(\omega_1 \Delta t), \sin(\omega_1 \Delta t), \ldots, \cos(\omega_k \Delta t), \sin(\omega_k \Delta t) \right].
    

    这里 k 是采样频率的数量,决定了编码的维度 $d=2k$(每组 \cos + \sin 占 2 维)。

例子

假设将时间$t$编码为一个2维向量实际论文中维度更高比如100维

时间编码函数为:


\Phi_d(t) = \sqrt{\frac{1}{d}} \left[ \cos(\omega_1 t), \sin(\omega_1 t), \cos(\omega_2 t), \sin(\omega_2 t), \ldots \right]

其中:

  • $d$是维度(这里$d=2$即2维向量
  • \omega_1, \omega_2, \ldots 是从某个分布$p(\omega)$中随机采样的频率参数(可以理解为“时间刻度”)。
  1. 随机初始化频率
    假设我们随机设定两个频率:

    • \omega_1 = 0.5
    • \omega_2 = 1.0
  2. 编码时间点

    • 对时间$t=2$,计算:

      
      \Phi_2(2) = \sqrt{\frac{1}{2}} \left[ \cos(0.5 \cdot 2), \sin(0.5 \cdot 2) \right] = \sqrt{0.5} \left[ \cos(1.0), \sin(1.0) \right] \approx [0.54, 0.84]
      
    • 对时间$t=5$,计算:

      
      \Phi_2(5) = \sqrt{0.5} \left[ \cos(2.5), \sin(2.5) \right] \approx [-0.35, 0.94]
      
  3. 时间差异的体现

    • 两个时间点$t=2$和$t=5$的编码向量不同,且它们的点积(相似度)会反映时间跨度$|5-2|=3$

      
      \Phi_2(2) \cdot \Phi_2(5) \approx 0.54 \cdot (-0.35) + 0.84 \cdot 0.94 \approx 0.59
      
    • 如果$t=2$和$t=4$的编码点积更大,说明$\Delta t=2$比$\Delta t=3$的交互更相关。

为什么这样设计?

  • 谐波基函数cos(ωt)sin(ωt)是周期性函数,能捕捉时间的周期性模式(比如白天/夜晚的周期性行为,但不会强制引入周期性)。
  • 内积反映时间差通过Bochner定理两个时间编码的内积Φ(t₁)·Φ(t₂)近似于一个时间核函数𝒦(t₁-t₂),时间差越小,内积越大(相似性越高)。
  • 可学习性:频率参数{ωᵢ}可以通过训练优化(例如让模型自动学习“短期交互”和“长期交互”的不同时间尺度)。

工作原理

1.时间约束的邻居集合

时间约束:对于目标节点 v_0 在时间 $t$,其邻居 \mathcal{N}(v_0; t) 仅包含与 v_0 在时间 t 之前发生过交互的节点(即 $t_i < t$)。


\mathcal{N}(v_0; t) = \{ v_i \mid \exists (v_0, v_i, t_i) \in \mathcal{E}, t_i < t \}

实际实现中TGAT通过以下两种策略避免邻居集合无限膨胀

1固定时间窗口Sliding Time Window

  • 仅保留目标时间 t 的最近 \Delta T 时间内的交互节点例如过去7天的邻居

  • 数学表达

    
    \mathcal{N}(v_0; t) = \{ v_i \mid (v_0, v_i, t_i) \in \mathcal{E}, t - \Delta T \leq t_i < t \}
    

2邻居采样Neighborhood Sampling

  • 即使在一个时间窗口内如果邻居数量过多例如社交网络中的活跃用户TGAT会随机采样固定数量的邻居如最多20个
  • 随机采样可行是因为时间编码注意力权重会自动学习为近期交互分配更高权重。

2.时间编码的邻居特征矩阵


Z(t) = \left[ \tilde{h}_0^{(l-1)}(t) \| \Phi_{d_T}(0), \tilde{h}_1^{(l-1)}(t_1) \| \Phi_{d_T}(t-t_1), \ldots \right]^\top

输入

  • 目标节点 v_0 在时间 t 的上一层特征:$\tilde{h}_0^{(l-1)}(t)$。
  • 邻居节点 v_i 在历史时间 t_i 的特征:$\tilde{h}_1^{(l-1)}(t_1), \tilde{h}_2^{(l-1)}(t_2), \ldots$。
  • 时间编码函数 $\Phi_{d_T}$:将时间差 t-t_i 映射为向量

3.注意力权重计算

通过Query-Key-Value机制计算邻居权重


\alpha_i = \text{softmax} \left( \frac{(q(t)^\top K_i(t))}{\sqrt{d_h}} \right), \quad q(t) = [Z(t)]_0 W_Q, \quad K_i(t) = [Z(t)]_i W_K
  • $q(t)$目标节点的Query。
  • $K_i(t)$邻居节点的Key。
  • 动态性:权重 \alpha_i 依赖当前时间 t 和邻居交互时间 $t_i$。

4.多头注意力聚合


h(t) = \text{Attn}(q(t), K(t), V(t)) = \sum_{i=1}^N \alpha_i V_i(t), \quad V_i(t) = [Z(t)]_i W_V

$V_i(t)$邻居节点的Value含时间编码的特征。

5.节点嵌入更新


\tilde{h}_0^{(l)}(t) = \text{FFN}(h(t) \| x_0)
  • $h(t)$:聚合后的邻居表示。
  • $x_0$:目标节点的原始特征。
  • FFNFeed-Forward Network进一步融合时空信息。

动态推理:对任意新时间 $t$,只需输入当前邻接关系和节点特征,通过前向传播计算嵌入。

总结:每个节点,动态融合节点自身及其邻居在不同时间点的嵌入,其中这些嵌入都显式编码了时间信息。

不同层之间的嵌入维度可以不同,但同一层内所有时间步的嵌入维度必须保持一致。

每个节点运行的模型一样的。实时性。

节点是否需要交互(是否需要全局信息) //

不知道全局信息(只知道邻居) 训练出来的模型是否一样?(能否协同。)

举例

假设目标节点 v_0 在时间 t=10 需要聚合历史邻居,其中:

$t=10$时$v_0$的特征:\tilde{h}_0(t=10)=[0.1, 0.2, 0.3, 0.4, 0.5]

  • 在时间 $t_1=8$v_0 与邻居 \{v_{1}, v_{2}\} 交互。

    • v_1 的特征:\tilde{h}_1(t=8) = [0.6, 0.7, 0.8, 0.9, 1.0]

    • v_2 的特征:\tilde{h}_2(t=8) = [0.2, 0.3, 0.4, 0.5, 0.6]

  • 在时间 $t_2=5$v_0 与邻居 \{v_{3}\} 交互。

    • v_3 的特征:\tilde{h}_3(t=5) = [1.1, 1.2, 1.3, 1.4, 1.5]

时间编码d_T=2, $\omega_1=1$

  • \Phi_{d_T}(10-8=2) = [\cos(2), \sin(2)] \approx [-0.42, 0.91]
  • \Phi_{d_T}(10-5=5) = [\cos(5), \sin(5)] \approx [0.28, -0.96]
  • \Phi_{d_T}(0) = [1, 0]

将每个邻居的特征与对应时间编码拼接:


Z(10) = \begin{bmatrix}
\tilde{h}_0(10) \| \Phi_{d_T}(0) \\
\tilde{h}_1(8) \| \Phi_{d_T}(2) \\
\tilde{h}_2(8) \| \Phi_{d_T}(2) \\
\tilde{h}_3(5) \| \Phi_{d_T}(5)
\end{bmatrix} = \begin{bmatrix}
0.1, 0.2, 0.3, 0.4, 0.5, 1, 0 \\
0.6, 0.7, 0.8, 0.9, 1.0, -0.42, 0.91 \\
0.2, 0.3, 0.4, 0.5, 0.6, -0.42, 0.91 \\
1.1, 1.2, 1.3, 1.4, 1.5, 0.28, -0.96
\end{bmatrix}_{4 \times 7}

注意力权重计算:

1.投影矩阵参数(简化示例,设 d_h = 3,真实情况由训-练得来):


W_Q = W_K = \begin{bmatrix}
0.1 & 0.2 & 0.3 \\
0.4 & 0.5 & 0.6 \\
0.7 & 0.8 & 0.9 \\
1.0 & 1.1 & 1.2 \\
1.3 & 1.4 & 1.5 \\
0.2 & 0.3 & 0.4 \\
0.5 & 0.6 & 0.7
\end{bmatrix}_{7 \times 3}, \quad
W_V = \begin{bmatrix}
0.3 & 0.4 & 0.5 \\
0.6 & 0.7 & 0.8 \\
0.9 & 1.0 & 1.1 \\
1.2 & 1.3 & 1.4 \\
1.5 & 1.6 & 1.7 \\
0.1 & 0.2 & 0.3 \\
0.4 & 0.5 & 0.6
\end{bmatrix}_{7 \times 3}

2.计算 Query/Key/Value

  • Query(目标节点 $v_0$

q(10) = [Z(10)]_0 W_Q = [0.1, 0.2, 0.3, 0.4, 0.5, 1, 0] \cdot W_Q = [2.38, 2.76, 3.14]
  • Keys(邻居节点):

    
    K(10) = [Z(10)]_{1:3} W_K = \begin{bmatrix}
    0.6 & 0.7 & 0.8 & 0.9 & 1.0 & -0.42 & 0.91 \\
    0.2 & 0.3 & 0.4 & 0.5 & 0.6 & -0.42 & 0.91 \\
    1.1 & 1.2 & 1.3 & 1.4 & 1.5 & 0.28 & -0.96
    \end{bmatrix} \cdot W_K = \begin{bmatrix}
    3.22 & 3.68 & 4.14 \\
    1.82 & 2.18 & 2.54 \\
    4.18 & 4.64 & 5.10
    \end{bmatrix}
    
  • Values(邻居节点):

    
    V(10) = [Z(10)]_{1:3} W_V = \begin{bmatrix}
    3.52 & 3.92 & 4.32 \\
    2.12 & 2.42 & 2.72 \\
    4.48 & 4.88 & 5.28
    \end{bmatrix}
    

3.计算注意力权重

  • 计算未归一化分数(缩放因子 $\sqrt{d_h} = \sqrt{3} \approx 1.732$

\text{scores} = \frac{q(10) K(10)^\top}{\sqrt{3}} = \frac{[2.38, 2.76, 3.14] \cdot \begin{bmatrix}3.22 & 1.82 & 4.18 \\ 3.68 & 2.18 & 4.64 \\ 4.14 & 2.54 & 5.10\end{bmatrix}}{1.732}\approx [18.46, 10.56, 21.40]
  • Softmax 归一化:
    
    \alpha = \text{softmax}([18.46, 10.56, 21.40]) = [2.4 \times 10^{-2}, 1.6 \times 10^{-5}, 0.976]
    

4.聚合邻居信息


h(10) = \sum_{i=1}^3 \alpha_i V_i = 0.024 \cdot \begin{bmatrix}3.52 \\ 3.92 \\ 4.32\end{bmatrix} + 1.6 \times 10^{-5} \cdot \begin{bmatrix}2.12 \\ 2.42 \\ 2.72\end{bmatrix} + 0.976 \cdot \begin{bmatrix}4.48 \\ 4.88 \\ 5.28\end{bmatrix}\approx [4.48, 4.88, 5.28]

5.节点嵌入更新

1.拼接操作


z=[h(10) \| x_0] = [4.48, 4.88, 5.28, 0.1, 0.2, 0.3, 0.4, 0.5]

2. FFN参数设置

FFN 包含两层:


\tilde{h}_0^{(l)}(10)=\text{FFN}(z) = \text{ReLU}(z W_0 + b_0) W_1 + b_1

需提前训练的参数

参数名称 符号 维度 作用 来源模块
时间编码频率参数 \omega_1, \ldots, \omega_d d \times 1 控制时间编码的基频率,用于生成 $\Phi_{d_T}(t)$。 功能性时间编码Bochner定理
Query投影矩阵 W_Q (d + d_T) \times d_h 将目标节点特征和时间编码映射为Query向量。 自注意力机制
Key投影矩阵 W_K (d + d_T) \times d_h 将邻居节点特征和时间编码映射为Key向量。 自注意力机制
Value投影矩阵 W_V (d + d_T) \times d_h 将邻居节点特征和时间编码映射为Value向量。 自注意力机制
FFN第一层权重 W_0^{(l)} (d_h + d_0) \times d_f 前馈网络的第一层线性变换,融合邻居聚合特征和原始特征。 前馈网络FFN
FFN第一层偏置 b_0^{(l)} d_f \times 1 第一层的偏置项。 前馈网络FFN
FFN第二层权重 W_1^{(l)} d_f \times d 前馈网络的第二层线性变换,生成最终节点表示。 前馈网络FFN
FFN第二层偏置 b_1^{(l)} d \times 1 第二层的偏置项。 前馈网络FFN