Oct, 2023

在位置 - 方向空间中通过权重共享实现快速高效的 SE$(n)$ 等变网络

TL;DR基于均匀空间理论,我们得出了在灵活的消息传递框架中使用的几何最优边属性,将卷积网络中的权重共享定义为在应该等同处理的点对上共享消息函数的共享点对的等价类,并导出能够唯一标识这些等价类的属性。我们通过在处理 3D 点云时开发了一种高效的等变群卷积网络来应用这一理论。在三个不同的基准测试中,即原子间势能预测、N 体系统轨迹预测和通过等变扩散模型生成分子,我们以准确性和速度方面达到了最先进的结果。