消息传递神经网络(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 小时前
RAG(Retrieval Augmented Generation)
人工智能·spring ai
冬奇Lab4 小时前
一天一个开源项目(第259篇):Modern Web Guidance —— Chrome 团队给 AI 编码 Agent 的「现代 Web 最佳实践」外挂
人工智能·开源
飞哥数智坊6 小时前
打造自己的 Personal Agent,从抄大肥鱼作业开始
人工智能·deepseek
对你无可奈何6 小时前
Magpie 实战:把 WorkBuddy、Qoder 的订阅模型接进一个本地池子
人工智能
Csvn6 小时前
每请求到底多少钱?把 Token 成本算成可观测指标(C01)
人工智能·aigc·agent
吴名氏.6 小时前
Java转AI Agent教程(第十三周)混合检索与重排
java·数据库·人工智能·ai·ai agent
老金带你玩AI8 小时前
惊人的一致性!Seedream 5.5实测!
人工智能
数据技术说9 小时前
在 KES AI 数据库上搭一条 NL2SQL 链路:语义、召回、权限各放哪
人工智能·sql·代码规范
jay神9 小时前
基于 YOLOv8 的手势识别系统(Web 版)
前端·人工智能·python·深度学习·yolo·计算机视觉·计算机毕业设计