摘要
图神经网络 (GNN) 在各种图学习任务中展现出优异的性能,但在应用于大规模图时面临着巨大的计算挑战。缓解这些挑战的一种有效方法是图稀疏化,它涉及移除不必要的边以减少计算开销。然而,以前的图稀疏化方法通常依赖于单一的全局稀疏化设置和统一的剪枝标准,无法为每个节点的复杂局部上下文提供定制的稀疏化方案。在本文中,我们引入了混合图 (MoG),利用混合专家 (MoE) 的概念,为每个节点动态选择定制的剪枝解决方案。具体来说,MoG 包含多个稀疏化专家,每个专家都具有独特的稀疏化水平和剪枝标准,并为每个节点选择合适的专家。随后,MoG 对不同专家在 Grassmann 流形上生成的稀疏图进行混合,以推导出最佳的稀疏图。MoG 的一个显著特性是其完全局部的性质,因为它取决于每个节点的具体情况。对四个大型 OGB 数据集和两个超像素数据集进行了广泛的实验,配备了五个 GNN 主干,结果表明 MoG (I) 在更高的稀疏化水平 ($8.67\%\sim 50.85\%$) 下识别子图,性能与稠密图相当或更好,(II) 在 GNN 推理中实现了 $1.47-2.62\times$ 的加速,性能下降微不足道,以及 (III) 提升了“顶尖学生” GNN 的性能 ($1.02\%\uparrow$ 在 RevGNN+\textsc{ogbn-proteins} 上和 $1.74\%\uparrow$ 在 DeeperGCN+\textsc{ogbg-ppa} 上)。