arXiv:2410.09795v4 宣告类型: 替换-交叉 摘要：预测分子基态构象（即能量最小化构象）对于许多化学应用，如分子对接和属性预测至关重要。经典的能量基模拟在解决这个问题时耗时较长，而现有的基于学习的方法在计算效率方面占有优势，但牺牲了准确性和可解释性。在本文中，我们提出了一个新颖且有效的方法，以弥合能量基模拟和基于学习策略之间的差距，设计并学习一个受Wasserstein梯度流驱动的SE(3)-Transformer，称为WGFormer，用于预测分子基态构象。具体而言，我们的方法在自编码框架中处理这项任务，利用所提出的WGFormer编码低质量构象，并由多层感知机（MLP）解码相应的基态构象。WGFormer的架构对应于Wasserstein梯度流——它通过最小化定义在原子的潜在混合模型上的能量函数来优化分子构象，从而显著提高性能和可解释性。广泛的实验表明，我们的方法在与最先进的竞争对手相比时始终表现出色，提供了一个新的且富有洞察力的范式来预测分子基态构象。