提出可动态演化的图交互机制,更好捕捉节点关系随时间变化的模式。
Time-varying Interaction Graph ODE for Dynamic Graph Representation Learning

- 用可学习的交互基函数分解图神经ODE演化过程
- 在6个动态图数据集上属性预测效果优于现有方法
- 适合研究动态网络建模与可解释性分析的学者
图神经微分方程(Graph Neural ODE)将神经ODE与图神经网络的消息传递机制结合,为图表示学习提供连续时间建模方法。然而,在动态图场景中,现有图神经ODE通常采用统一的消息传递机制,假设节点间交互在任意时刻共享相同的消息传递函数,难以捕捉交互模式的多样性与时变特性。为此,我们提出时变交互图常微分方程(TI-ODE)。其核心思想是将图ODE的演化函数分解为一组可学习的交互基函数,每个基函数对应一种特定类型的节点间交互。这些基函数通过随时间变化的可学习权重动态组合,使节点交互模式能够自适应地随时间演化。在六个动态图数据集上的实验表明,TI-ODE持续优于现有方法,在属性预测任务中达到领先性能;在Covid数据集上的实验进一步验证了TI-ODE的可解释性与泛化能力。此外,我们从理论和实证两方面证明,相较于使用统一消息传递机制的模型,TI-ODE具有更优鲁棒性。
原文摘要 · Abstract (English)
Graph neural Ordinary Differential Equations (ODE) combine neural ODE with the message passing mechanism of Graph Neural Networks (GNN), providing a continuous-time modeling method for graph representation learning. However, in dynamic graph scenarios, existing graph neural ODEs typically employ a unified message passing mechanism, assuming that inter-node interactions share the same message passing function at any time, which makes it challenging to capture the diversity and time-varying nature of inter-node interaction patterns. To address this, we propose Time-varying Interaction Graph Ordinary Differential Equations (TI-ODE). The core idea of TI-ODE is to decompose the evolution function of a graph ODE into a set of learnable interaction basis functions, where each basis function corresponds to a distinct type of inter-node interaction. These basis functions are dynamically combined through time-dependent learnable weights, enabling inter-node interaction patterns to adaptively evolve over time. Experimental results on six dynamic graph datasets demonstrate that TI-ODE consistently outperforms existing methods and achieves state-of-the-art performance on attribute prediction tasks, and experiments on the \textit{Covid} dataset further verify the interpretability and generalizability of our TI-ODE. Furthermore, we demonstrate both theoretically and empirically that TI-ODE exhibits superior robustness compared to models utilizing a unified message-passing mechanism.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。