Jun, 2024

多维向量神经元:更好更快的 O (n) 等变 Clifford 图神经网络

TL;DR本研究通过测试几种基于 Clifford 多矢量的新颖消息传递图神经网络(GNNs),旨在解决当前大多数对 $O (n)$ 或 $SO (n)$ 等等变换具有等变性的深度学习模型在处理标量信息时效果较好,但计算复杂度较高的问题。我们的方法利用了高效的不变标量特征,并同时在多矢量表示上执行表达性学习,特别是通过使用等变的几何乘积算子。通过整合这些要素,我们的方法在 N-Body 模拟任务和蛋白质去噪任务上优于已建立的高效基准模型,同时保持高效性。特别是,在 N-body 数据集上,我们的方法的误差推动到 0.0035(平均 3 次运行);比最近的方法提升了 8%。我们的实现在 Github 上提供。