对等图神经网络在物理动力学学习中松弛连续约束
💡
原文中文,约400字,阅读约需1分钟。
📝
内容提要
该论文介绍了一种名为ESTAG的新动力学模拟方法,利用过去时期的轨迹预测物理系统的动态。ESTAG通过等变离散傅里叶变换和等变空间模块提取周期模式和传递空间信息,再通过等变时间模块聚合时间信息。实验证明ESTAG在不同级别的数据集上比传统方法更有效。
🎯
关键要点
- 学习表示和模拟物理系统的动力学是一个关键且具有挑战性的任务。
- 该论文将动力学模拟重新定义为时空预测任务,利用过去时期的轨迹恢复非马尔科夫相互作用。
- 提出了 ESTAG(等变时空注意力图网络),作为时空 GNN 的等变版本。
- 设计了等变离散傅里叶变换(EDFT)以提取历史帧中的周期模式。
- 构建了等变空间模块(ESM)以完成空间信息传递。
- 使用前向注意力和等变池化机制构建等变时间模块(ETM)以聚合时间信息。
- 在三个实际数据集上评估模型,涵盖分子级、蛋白质级和宏观级。
- 实验结果验证了 ESTAG 相对于典型的时空 GNN 和等变 GNN 的有效性。
➡️