消息传递神经网络(MPNN)

消息传递神经网络(MPNN) 来自Google Brain在2017年发布的Neural Message Passing for Quantum Chemistry,是一种专门用来处理图结构数据(graph data)的通用计算框架。它将各种 GNN 模型纳入同一框架,便于理解和比较。

阶段一:Message Passing(消息传递)

让每个节点通过与邻居"交流",逐步获得关于其局部子图结构的信息。

|----------|------------------------------|
| Xi(k)​ | 节点 i 在第 k 层的特征(状态) |
| Xi(k−1)​ | 节点 i 在上一层(第 k−1 层)的特征 |
| N(i) | 节点 i 的邻居集合(和它相连的节点) |
| Xj(k−1)​ | 邻居节点 j 在上一层的特征 |
| ej,i​ | 从节点 j 到节点 i 的边上的特征(比如化学键类型) |
| ϕ(k) | 消息生成函数(message function) |
| ⨁ | 聚合操作(如求和、平均等) |
| γ(k) | 更新函数(update function) |

1. 消息生成(Message Function):ϕ

  • 每个节点将生成自己的消息,然后向自己的邻居节点"传播"自己的消息:

2. 消息聚合(Aggregate):⨁

  • 每个节点会聚合来自邻居的消息:

3. 消息更新(Update):γ

  • 你用这些信息更新自己的状态:

阶段二:Readout(读出)

把所有节点的表示"压缩"成一个固定长度的向量 ,用来代表整张图

相关推荐
那个村的李富贵4 小时前
光影魔术师:CANN加速实时图像风格迁移,让每张照片秒变大师画作
人工智能·aigc·cann
腾讯云开发者6 小时前
“痛点”到“通点”!一份让 AI 真正落地产生真金白银的实战指南
人工智能
CareyWYR6 小时前
每周AI论文速递(260202-260206)
人工智能
hopsky7 小时前
大模型生成PPT的技术原理
人工智能
禁默7 小时前
打通 AI 与信号处理的“任督二脉”:Ascend SIP Boost 加速库深度实战
人工智能·信号处理·cann
心疼你的一切8 小时前
昇腾CANN实战落地:从智慧城市到AIGC,解锁五大行业AI应用的算力密码
数据仓库·人工智能·深度学习·aigc·智慧城市·cann
AI绘画哇哒哒8 小时前
【干货收藏】深度解析AI Agent框架:设计原理+主流选型+项目实操,一站式学习指南
人工智能·学习·ai·程序员·大模型·产品经理·转行
数据分析能量站8 小时前
Clawdbot(现名Moltbot)-现状分析
人工智能
那个村的李富贵8 小时前
CANN加速下的AIGC“即时翻译”:AI语音克隆与实时变声实战
人工智能·算法·aigc·cann
二十雨辰8 小时前
[python]-AI大模型
开发语言·人工智能·python