本文介绍了一种新型等变图神经网络架构ESNN(Equivariant Sheaf Neural Network),旨在解决现有等变图神经网络在向量信息跨图传递时变换能力受限的问题。ESNN通过学习有向的矩阵值边传输机制,在保持一阶标量和向量特征及精确欧几里得等变性的前提下,将几何灵活性注入边传输过程,而非提高表示阶数。理论分析表明,当相对位移是唯一协变几何输入时,所有线性O(n)等变映射可分解为独立的径向和切向分量;引入学习到的协变特征后,可实现更丰富的特征条件化变换。该方法还支持受控对称性松弛,以适应具有偏好环境方向的系统,并能在方向通路未激活时恢复完全的E(n)等变性。实验涵盖粒子动力学、网格模拟、点云分类和分子性质预测等任务,结果显示ESNN在动力学预测、对称性破缺场景下的重力轴恢复、长时程网格任务滚动预测等方面表现优异,且对未见旋转具有鲁棒性。研究表明,学习几何信息在边上的传输方式为表达性等变消息传递提供了补充路径,无需依赖高阶表示。
| Equivariant Sheaf Neural Network (ESNN) | 一种等变层神经网络,通过边上的矩阵值传输增强图神经网络,同时保持欧几里得等变性。 |
| Equivariance | 等变性,指模型输出随输入变换(如旋转、平移)而相应变换的性质。 |
| Transport | 传输,指在图中相邻节点间传递向量信息的过程,此处为有向矩阵值传输。 |
| Symmetry relaxation | 对称性松弛,指在特定方向存在时允许对称性部分破缺,以更好地适应数据。 |
| Message passing | 消息传递,图神经网络中节点间信息交换的机制。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅