GNN(图神经网络)

GNN

在图结构上学习

亦作、亦称:图神经网络 · Graph Neural Network

图神经网络(Graph Neural Network,GNN)是一类专为图结构数据设计的深度学习模型,通过在节点与边之间反复传递并聚合信息来学习结构化表示。它将传统神经网络从欧几里得空间扩展到分子、社交网络、知识图谱等非欧几里得场景,是连接深度学习与关系推理的核心桥梁。

概述

GNN 的核心思想是「让节点通过反复与邻居交换信息来更新自身表示」。

  • 图结构数据:由节点(顶点)和边组成,天然表达实体间的关系,如原子键、好友关系、知识三元组。
  • 非欧几里得挑战:传统 CNN 依赖规则网格,无法直接处理节点数量和连接方式任意变化的图。
  • 消息传递范式:每一层聚合邻居的特征,多层叠加后节点「看到」更大范围的上下文。
  • 输出粒度:可做节点分类、边预测、整图分类三类任务,覆盖大多数图学习需求。
  • 表达能力上界:理论上等价于 Weisfeiler-Leman(WL)图同构测试,区分能力有限。

工作原理

消息传递可分为三个阶段,逐层迭代执行。

  • 消息生成(Message):每条边根据源节点、目标节点及边特征生成一条消息向量。
  • 聚合(Aggregate):目标节点收集所有入边消息,用求和、均值或最大池化等函数合并。
  • 更新(Update):将聚合结果与节点自身旧表示拼接或相加,送入可学习的变换(通常是 MLP)得到新表示。
  • 读出(Readout):做整图任务时,再对所有节点表示做全局池化,生成图级向量。
  • 多层叠加:第 k 层的节点表示包含 k 跳邻居的信息,但层数过深会导致过平滑问题。

主要变体

现代 GNN 家族以聚合方式和注意力机制为主要区分点。

  • GCN(图卷积网络):Kipf & Welling 于 2017 年提出,用归一化邻接矩阵做谱域卷积一阶近似,结构简单,是入门基准。
  • GraphSAGE:Hamilton 等于 2017 年提出,对邻居采样后聚合,支持归纳式推断(可泛化到新节点)。
  • GAT(图注意力网络):Veličković 等于 2018 年提出,对不同邻居赋予可学习的注意力权重,提升表达能力。
  • GIN(图同构网络):Xu 等于 2019 年提出,聚合函数改为求和加 MLP,理论上达到 WL 测试的表达上限。
  • 图 Transformer:借鉴 Transformer 全局注意力,突破局部邻居限制,适合长程依赖场景(如微软 Graphormer 2021)。

应用场景

GNN 在多个领域已有成熟落地,尤其擅长结构先验明确的任务。

  • 药物发现与分子设计:将分子建模为原子-键图,预测分子性质、药物毒性和蛋白质-配体结合亲和力。
  • 推荐系统:将用户-商品关系建模为二部图,PinSage(Pinterest,2018)是工业界早期大规模落地案例。
  • 知识图谱补全:在三元组图上预测缺失关系,捕获多跳关系推理能力。
  • 社交网络分析:节点分类(识别虚假账号)、社区发现、影响力传播预测。
  • 交通与时空预测:将路网建模为图,结合时序模型(如 STGCN)预测路况和流量。

局限与挑战

GNN 在工程落地中面临若干共性难题。

  • 过平滑(Over-Smoothing):层数增加时节点表示趋于同质化,无法区分远距节点;实践中通常控制在 2–4 层。
  • 过压缩(Over-Squashing):指数增长的邻居信息被压缩进固定维向量,导致远端信号丢失,长程依赖捕获困难。
  • 大图可扩展性:全批次训练内存开销随节点数线性增长;需要邻居采样(GraphSAGE)或集群采样(Cluster-GCN)缓解。
  • 异构图处理:现实图常有多种节点与边类型,标准 GNN 需扩展为异构图网络(如 HAN、HGT)。
  • 标注数据稀缺:图标注成本高,自监督预训练(如图对比学习)是活跃研究方向。

发展脉络

GNN 经历了从理论提出到工业爆发的约二十年演进。

  • 2005:Gori 等人首次提出图上神经网络的基本概念,尚未形成完整形式化。
  • 2009:Scarselli 等在 IEEE Transactions on Neural Networks 发表「The Graph Neural Network Model」,给出递归消息传递的严格数学框架,是公认的奠基论文。
  • 2013:Bruna 等提出谱图卷积,将谱图理论引入深度学习。
  • 2016:ChebNet(Defferrard 等)用切比雪夫多项式近似谱卷积,大幅降低计算复杂度。
  • 2017:GCN(Kipf & Welling)将 ChebNet 简化为一阶近似,成为最广泛引用的基线;GraphSAGE 同年提出归纳学习范式。
  • 2018:GAT 引入注意力机制;PinSage 证明 GNN 可在十亿级图上工业化落地。
  • 2019:GIN 从理论角度明确 GNN 表达能力上界。
  • 2021:Graphormer(微软)将 Transformer 系统性迁移到图,赢得 OGB 多项榜首。
  • 2022 至今:图基础模型(Graph Foundation Model)探索跨域预训练迁移,GNN 与大语言模型融合成新前沿。

常见误解

日常交流中容易听到的简化说法,未必准确,但能帮助理解误解从何而来。

  • 「在图结构上学习」
  • 「节点之间传消息」
  • 「分子和知识图谱常用」

相关术语

和本术语关联紧密的其他词条,便于串联理解。

🎯 考点练习

含该术语的高频面试题,含标准答案与追问。

延伸阅读

从知识库精选 3 篇文章,帮助深入理解该术语。

  1. 1

    AI+生物计算:微观世界模型在药物研发中的应用

    深入理解微观世界模型如何在药物发现中发挥作用——从分子动力学模拟到蛋白质-配体结合预测,从虚拟筛选到药物分子优化,全面解读 AI 如何重构药物研发的底层计算范式

  2. 2

    数值计算进阶:优化算法的数学基础

    从条件数到收敛速率,深入理解优化算法的数学原理

  3. 3

    Agent 记忆系统(四):向量数据库、知识图谱与记忆检索全景指南

    AI Agent 的记忆系统是决定其智能水平的核心组件。本文系统讲解 Agent 记忆体系的完整架构:从短期工作记忆到长期语义记忆,从向量数据库的嵌入检索到知识图谱的关系推理,从记忆压缩策略到遗忘机制,帮助你在构建 Agent 时设计正确的记忆方案。

外部参考

维基百科:查看「GNN」词条

本页内容为本站原创撰写;维基百科链接仅作延伸参考。