arXiv:2407.02025v4 宣布类型: replace-cross 摘要：受化学和其他科学应用的启发，我们研究了消息传递神经网络在几何图中的表现力，其中节点特征对应于三维位置。近期的研究表明，这类模型可以分离通用的非同构几何图，尽管它们可能无法分离一些稀有的复杂实例。然而，这些结果假设了一个全连接的图，其中每个节点都拥有其他所有节点的全部知识。相比之下，通常情况下，每个节点只拥有其最近邻居的有限知识。 本文表明，在基础图连接的情况下，具有旋转不变特征的消息传递网络可以分离通用的非同构几何图。当仅允许不变的中间特征时，对于通用全局刚性的图，这种分离是得到保证的。我们介绍了一个简单的架构——EGENNET，该架构实现了我们的理论保证，并在合成和化学基准测试中与替代架构相比表现良好。我们的代码可在 https://github.com/yonatansverdlov/E-GenNet 获取。