摘要
arXiv:2502.06136v2 宣告类型: replace-cross
摘要: 图神经网络(GNNs)已成为学习图形结构数据表示的强大工具。除了实值GNNs,四元数GNNs在处理图形结构数据的任务上也表现出色。为了减少能耗,在保持与原有大小GNNs相当的准确性的同时,我们缩减了模型规模。本文引入了四元数消息传递神经网络(QMPNNs)框架,该框架利用四元数空间计算节点表示。我们的方法提供了一种四分之一原始参数量的一般可推广方法,将四元数表示融入到GNN架构中。此外,我们提出了图彩票(Graph Lottery Tickets)的新视角,重新定义了其在GNNs和QMPNNs上下文中的适用性。我们特别致力于从GNNs的子网络中找到一种初始彩票,能够在训练后达到与原GNN相当的性能。从而进一步减少可训练的模型参数。为了验证我们提出的QMPNN框架和LTH在GNNs和QMPNNs中的有效性,我们在三个基本的图基任务(节点分类、链接预测和图分类)上的实际数据集上评估了其性能。