AAAIJul, 2022

基于能量激发的等变预训练对 3D 分子图的研究

TL;DR本文提出一种新的三维分子预训练方法,该方法采用等变能量基模型作为主干网络,以预测节点力信息为目标进行节点级预训练损失计算,并采用高斯分布以保证其 E(3)不变性,同时通过图层级噪声缩放预测任务来进一步提高模型性能,实验结果表明,该方法在两个挑战性的三维基准测试上均取得了比当前最先进的预训练方法更佳的表现。