arXiv:2407.02025v3 通知类型: 替换-交叉 摘要：受到化学和其他科学应用的启发，我们研究了消息传递神经网络在几何图上的表达能力，其中节点特征对应于三维位置。近期的工作表明，这类模型可以区分任意的非同构几何图，尽管它们可能无法区分一些罕见和复杂的实例。然而，这些结果假设是一个完全连接的图，其中每个节点都拥有所有其他节点的完全信息。相反，在实际应用中，通常每个节点只拥有其有限数量的最近邻节点的信息。 本文表明，在基础图连接的情况下，具有旋转不变特征的消息传递网络可以区分任意的非同构几何图。当仅允许不变中间特征时，对于代数全局刚性图，可以保证任意区分。我们引入了一个简单的架构 EGENNET，它实现了我们的理论保证，并在合成和化学基准测试中优于其他架构。我们的代码可在 https://github.com/yonatansverdlov/E-GenNet 获取。