摘要
arXiv:2412.11293v2 动态图嵌入类型:替换跨学科
摘要:动态图嵌入作为一种重要技术,已被广泛应用于不同领域的复杂时变网络建模。虽然基于变换器的模型在捕捉时间图数据中的远距离依赖方面显示出潜力,但由于计算复杂度为二次,它们面临可扩展性挑战。这项研究提出了使用变换器和最近提出的Mamba架构(一种具有线性复杂度的状态空间模型)进行动态图嵌入方法的比较分析。我们引入了三个新的模型:TransformerG2G增强的图卷积网络、$\mathcal{DG}$-Mamba和结合图同构网络边缘卷积的$\mathcal{GDG}$-Mamba。在多个基准数据集上的实验表明,Mamba基模型在链接预测任务中达到了与基于变换器的方法相当或更好的性能,同时在较长序列上提供了显著的计算效率提升。值得注意的是,在具有高时间变异性数据集(如UCI、比特币和真实世界挖掘)上,$\mathcal{DG}$-Mamba变体始终优于基于变换器的模型,同时在更稳定的图形(如SBM)上保持竞争力。我们通过分析注意力权重和状态矩阵,提供了对学习到的时间依赖性的见解,揭示了模型捕捉复杂时间模式的能力。通过有效地将状态空间模型与图神经网络相结合,我们的研究解决了之前方法的关键限制,并为高效的时变图表示学习研究做出了贡献。这些发现为动态图嵌入的扩展到更大、更复杂的实际网络提供了有希望的方向,可能在社会网络分析、金融建模和生物系统动力学等领域开辟新的应用。