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

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

相关推荐
碑 一7 分钟前
视频分割Video K-Net
人工智能·计算机视觉
renhongxia117 分钟前
ORACLE-SWE:量化Oracle 信息信号对SWE代理的贡献
人工智能·深度学习·学习·语言模型·分类
AI自动化工坊19 分钟前
Google LiteRT-LM生产级部署指南:如何在边缘设备实现高效LLM推理?
人工智能·ai·llama
互联网江湖1 小时前
携程当学胖东来
人工智能
陌殇殇1 小时前
001 Spring AI Alibaba框架整合百炼大模型平台 — 快速入门
人工智能·spring boot·ai
Proxy_ZZ01 小时前
用Matlab绘制BER曲线对比SPA与Min-Sum性能
人工智能·算法·机器学习
黎阳之光1 小时前
黎阳之光:以视频孪生领跑全球,赋能数字孪生水利智能监测新征程
大数据·人工智能·算法·安全·数字孪生
宇擎智脑科技2 小时前
基于 SAM3 + FastAPI 搭建智能图像标注工具实战
人工智能·计算机视觉
F_U_N_2 小时前
效率提升80%:AI全流程研发真实项目落地复盘
人工智能·ai编程
月诸清酒2 小时前
24-260409 AI 科技日报 (Gemma 4发布一周下载破千万,开源模型生态加速演进)
人工智能·开源