摘要
arXiv:2502.06136v1 交叉类型公告
摘要:图神经网络(GNNs)已经成为了学习图结构化数据表示的强大工具。除了实值GNNs外,四元数GNNs在图结构化数据的任务上也表现出色。为了减少能源消耗,我们在保持与原大小GNNs相当的准确度的同时减小了模型大小。本文介绍了四元数消息传递神经网络(QMPNNs),一种利用四元数空间计算节点表示的框架。我们的方法提供了一种将四元数表示引入GNN架构中的通用方法,参数量仅为原模型的四分之一。此外,我们提出了图兑奖券(Graph Lottery Tickets)的新视角,重新定义了其在GNNs和QMPNNs中的适用性。我们具体的目标是从GNNs的子网络中找到初始化兑奖券,使其在训练后能达到与原GNN相当的性能,从而进一步减少可训练模型参数量。为了验证我们提出的QMPNN框架和LTH(剪枝)方法在GNNs和QMPNNs上的有效性,我们在三个基本的图任务——节点分类、链接预测和图分类——上对基于真实世界数据集的性能进行了评估。