Feb, 2023

基于分子几何的 Transformer 模型,用于准确建模三维原子系统

TL;DR我们提出了一种新颖的 Transformer 架构 Moleformer,它将节点(原子)和边(键和非键原子对)作为输入,并使用旋转不变性和平移不变性的几何感知空间编码来模拟它们之间的相互作用。我们在 OC20 和 QM9 数据集上进行了基准测试,Moleformer 在 OC20 的初始状态到松弛能量预测方面实现了最先进的水平,并且在预测量子化学性质方面与其他 Transformer 和图神经网络方法相比非常有竞争力,证明了所提出的几何感知空间编码在 Moleformer 中的有效性。