arXiv:2502.07500cs.LG2025-02

统一图网络UGN让单一模型搞定多种图任务,性能超越现有方法。

Unified Graph Networks (UGN): A Deep Neural Framework for Solving Graph Problems

  • 结合GCN与2D卷积,构建通用图学习框架
  • 在12个数据集上10个超越顶尖基线,其余持平
  • 适合需要跨任务图分析的科研与工程人员

深度神经网络已催生如Transformer等通用框架,广泛应用于文本与图像领域,但图问题仍缺乏通用解决方案。图结构普遍存在于各类应用中,相关研究历史悠久,近年深度学习方法迅速发展,且多领域图数据集日益丰富。然而,现有方法多针对特定任务,难以泛化。本文提出一种名为统一图网络(UGN)的新框架,融合图卷积网络(GCN)与二维卷积网络(Conv2D)的特征提取能力,将链接预测、节点分类、社区发现、图到图翻译、知识图谱补全等多种图学习任务统一于同一框架内。通过极少的任务特定设计(如超节点构造以提升大规模网络可扩展性,使用均值目标连接矩阵(MTCM)表示以增强图翻译任务的可扩展性),显著提升图学习的泛化能力。我们在六个无关图任务上,使用十二个不同数据集验证该框架,实验结果表明,UGN在十个数据集上显著优于当前最优基线,在剩余两个数据集上表现相当。

原文摘要 · Abstract (English)

Deep neural networks have enabled researchers to create powerful generalized frameworks, such as transformers, that can be used to solve well-studied problems in various application domains, such as text and image. However, such generalized frameworks are not available for solving graph problems. Graph structures are ubiquitous in many applications around us and many graph problems have been widely studied over years. In recent times, there has been a surge in deep neural network based approaches to solve graph problems, with growing availability of graph structured datasets across diverse domains. Nevertheless, existing methods are mostly tailored to solve a specific task and lack the capability to create a generalized model leading to solutions for different downstream tasks. In this work, we propose a novel, resource-efficient framework named \emph{U}nified \emph{G}raph \emph{N}etwork (UGN) by leveraging the feature extraction capability of graph convolutional neural networks (GCN) and 2-dimensional convolutional neural networks (Conv2D). UGN unifies various graph learning tasks, such as link prediction, node classification, community detection, graph-to-graph translation, knowledge graph completion, and more, within a cohesive framework, while exercising minimal task-specific extensions (e.g., formation of supernodes for coarsening massive networks to increase scalability, use of \textit{mean target connectivity matrix} (MTCM) representation for achieving scalability in graph translation task, etc.) to enhance the generalization capability of graph learning and analysis. We test the novel UGN framework for six uncorrelated graph problems, using twelve different datasets. Experimental results show that UGN outperforms the state-of-the-art baselines by a significant margin on ten datasets, while producing comparable results on the remaining dataset.

图神经网络通用模型多任务学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。