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 与大语言模型融合成新前沿。
常见误解
日常交流中容易听到的简化说法,未必准确,但能帮助理解误解从何而来。
- 「在图结构上学习」
- 「节点之间传消息」
- 「分子和知识图谱常用」
相关术语
和本术语关联紧密的其他词条,便于串联理解。
🎯 考点练习
含该术语的高频面试题,含标准答案与追问。
- 高级概念查看详解 →
分子表示与分子生成有哪些方法?
分子可表示为 SMILES 串、分子图或 3D 构象;生成用 VAE/扩散/自回归,关键是有效性与合成可行性。
- 中级概念查看详解 →
AI 天气预报(如 GraphCast)是如何工作的?
GraphCast 等用 GNN/Transformer 在地球网格上做数据驱动预测,由再分析数据训练,比传统数值模式快几个量级且精度相当。
- 中级概念查看详解 →
图神经网络(GNN)的核心思想「消息传递」是什么?
GNN 通过「聚合邻居 + 更新自身」的消息传递逐层迭代,每层扩大一跳感受野,学到融合结构与特征的节点表示。
- 中级概念查看详解 →
图卷积与普通卷积有什么本质区别?
图无规则网格、邻居数可变且无固定顺序,需排列不变聚合;CNN 是图卷积在规则网格上的特例。
延伸阅读
从知识库精选 3 篇文章,帮助深入理解该术语。
- 1
AI+生物计算:微观世界模型在药物研发中的应用
深入理解微观世界模型如何在药物发现中发挥作用——从分子动力学模拟到蛋白质-配体结合预测,从虚拟筛选到药物分子优化,全面解读 AI 如何重构药物研发的底层计算范式
- 2
数值计算进阶:优化算法的数学基础
从条件数到收敛速率,深入理解优化算法的数学原理
- 3
Agent 记忆系统(四):向量数据库、知识图谱与记忆检索全景指南
AI Agent 的记忆系统是决定其智能水平的核心组件。本文系统讲解 Agent 记忆体系的完整架构:从短期工作记忆到长期语义记忆,从向量数据库的嵌入检索到知识图谱的关系推理,从记忆压缩策略到遗忘机制,帮助你在构建 Agent 时设计正确的记忆方案。
外部参考
维基百科:查看「GNN」词条本页内容为本站原创撰写;维基百科链接仅作延伸参考。
