跳到主内容
快讯直播
AI智模界
AI 词典

图神经网络 GNN:让信息沿着关系流动

一句话定义:图神经网络(Graph Neural Network,GNN)是一类专门处理"图"这种数据结构的神经网络——数据点之间由边相连,模型通过反复让每个节点吸收邻居的信息来学习。

打个比方

你想了解公司里某位同事的口碑。只看他自己的简历不够,你还会去问他的下属、他的搭档、合作过的客户,把周围人的说法汇总起来,再形成判断。

GNN 做的就是这件事。每个节点(node)先带着自己的初始特征——比如用户的年龄和活跃度、商品的品类和价格。然后每一层做一次"消息传递"(message passing):把邻居节点的特征收集过来,按某种权重求和,与自己原有的特征拼在一起,过一个小型神经网络,得到这个节点的新表示。堆叠两三层,一个节点就能"看见"两跳、三跳之外的邻居。整个过程对所有节点同时、按同一套规则进行,所以这类操作常被称为图卷积(Graph Convolution)。

和相邻概念的区别

传统神经网络假设数据排列整齐:图片是规整的像素网格,表格是固定的几列。但图没有这种整齐结构——每个节点的邻居数量不一样,邻居之间也没有先后顺序。同一个社交网络里,有人有 3 个好友,有人有 3000 个,模型必须对这两种情况一视同仁。这就是 GNN 要专门设计的原因。

模型数据形态对"关系"的假设典型任务
全连接网络独立样本样本之间互不影响表格分类、回归
卷积网络 CNN规则网格邻居固定为上下左右像素图像识别
GNN任意图邻居数量和顺序都不固定节点分类、链路预测、推荐

对实际工作的意义

  • 推荐系统:用户和商品天然构成一张二分图。"和你口味相似的人买过什么"这句话,在 GNN 里就是消息沿着边传过来,对冷启动用户尤其有用。
  • 风控与反欺诈:账户、设备、IP 之间的转账和登录关系本身就是图。团伙作案在图上往往表现为联系异常密集的子图,比单看一条记录更容易发现。
  • 生物医药:分子就是原子构成的图,预测分子性质是 GNN 的经典战场。

给从业者一句提醒:GNN 的效果高度依赖图怎么建——边如何定义、权重怎么设、有没有噪声边,这些往往比换一个更花哨的模型结构更决定成败。图规模很大时还需要做邻居采样来控制计算量。具体框架的接口和用法,以官方页面为准。

AI 生成本文由 AI 基于公开信息自动生成,仅供参考。