用流体动力学思想改进图神经网络,解决特征混淆问题
Graph Navier Stokes Networks

- 引入纳维-斯托克斯方程思想,用动态速度场实现图上的对流传播
- 在12个真实数据集上分类精度超越现有方法,且有效缓解过平滑
- 适合处理异质性高、同质性不一的复杂图结构数据
图神经网络(GNN)已成为深度学习的核心,现有方法大多基于图信号处理和扩散方程建模消息传递,但存在深度增加时节点特征趋于相似的过平滑问题。受纳维-斯托克斯方程启发,本文提出图纳维-斯托克斯网络(GNSN),通过在图结构上定义动态速度场来引入对流机制,突破传统扩散式消息传递的局限。该模型自适应平衡对流与扩散,实现了更高效、直接的消息传播。在12个真实世界数据集上的大量实验表明,GNSN在分类准确率上持续优于当前最优基线,且显著缓解了过平滑问题。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have emerged as a cornerstone of deep learning, with most existing methods rooted in graph signal processing and diffusion equations to model message passing. However, these approaches inherently suffer from the oversmoothing problem, where node features become indistinguishable as the network depth increases. Inspired by the Navier Stokes equations, we introduce Graph Navier Stokes Networks (GNSN), a novel architecture that transcends conventional diffusion-based message passing by incorporating convection into graph structures. GNSN defines a dynamic velocity field on the graph to govern convection, enabling more efficient and direct message propagation. By adaptively balancing convection and diffusion, GNSN is able to efficiently handle datasets with varying levels of homophily. Extensive evaluations across twelve real-world datasets demonstrate that GNSN consistently outperforms state-of-the-art baselines in classification accuracy. Moreover, experimental results further emphasize its effectiveness in alleviating the oversmoothing problem.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。