消息传递神经网络(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(读出)

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

相关推荐
武子康3 分钟前
Embedding 替换怎么评测:把索引成本算进成功任务
人工智能·后端·agent
山东孙燕姿12 分钟前
维修工单的转派与协同:两个人一起上
大数据·运维·服务器·网络·人工智能
GPU实战笔记18 分钟前
L40S 云端任务验证:从推理、微调到生成工作流的边界
人工智能·模型微调·gpu租用·nvidia l40s·gpu 云服务器
让学习成为一种生活方式19 分钟前
厚朴(Magnolia officinalis)染色体水平基因组--iScience
大数据·人工智能
m4Rk_20 分钟前
【论文阅读】Agent记忆机制(109):MemComposer——用强化学习学会选哪些记忆、选多少以及怎样排序
论文阅读·人工智能·学习·开源·github
加宝的起点雷达22 分钟前
多个 AI 编程代理之间,上下文怎么共享:三类做法和三个坑
人工智能·架构
Data-Miner25 分钟前
AI 大模型安全深度分析:风险演化、治理现状与前瞻路径
人工智能·安全
具身AGI26 分钟前
三维动态只当训练信号,不进推理
人工智能·深度学习·计算机视觉
乐迪信息27 分钟前
AI算法盒子船体烟火检测:港口火灾早预警
人工智能·深度学习·算法
cc_瀚海知行33 分钟前
资料检索 1|直接放入全文、按需读文件和先检索,有什么区别?
java·人工智能·ai