纯Transformer无需图结构也能精准预测分子能量与力。
Transformers Discover Molecular Structure Without Graph Priors
- 直接用坐标训练Transformer,不依赖预设图结构。
- 在OMol25数据集上误差接近顶尖等变GNN模型。
- 注意力随距离衰减,可自适应不同分子环境。
图神经网络(GNN)是分子机器学习的主流架构,尤其用于分子性质预测和机器学习原子间势能(MLIPs)。GNN通过预定义图进行消息传递,通常基于固定半径截断或k近邻方案。尽管这种设计契合许多分子任务中的局部性,但硬编码图结构会因固定感受野限制表达能力,并因稀疏图操作导致推理缓慢。本文研究纯、未经修改的Transformer在不使用预定义图或物理先验的情况下,能否直接从笛卡尔坐标中逼近分子能量与力。以分析起点,我们展示了在相同训练算力下,该Transformer在OMol25数据集上的能量与力均方绝对误差达到与先进等变GNN相媲美的水平。我们发现,Transformer能学习到物理一致的模式,如注意力权重随原子间距离反比衰减,并因无硬编码偏差而灵活适应不同分子环境。标准Transformer的使用也带来了可预测的资源扩展性提升,符合其他领域观察到的经验缩放定律。结果表明,许多GNN的优良特性可在变压器中自适应涌现,挑战了硬编码图归纳偏置的必要性,为分子建模指明了标准化、可扩展的架构方向。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) are the dominant architecture for molecular machine learning, particularly for molecular property prediction and machine learning interatomic potentials (MLIPs). GNNs perform message passing on predefined graphs often induced by a fixed radius cutoff or k-nearest neighbor scheme. While this design aligns with the locality present in many molecular tasks, a hard-coded graph can limit expressivity due to the fixed receptive field and slows down inference with sparse graph operations. In this work, we investigate whether pure, unmodified Transformers trained directly on Cartesian coordinates$\unicode{x2013}$without predefined graphs or physical priors$\unicode{x2013}$can approximate molecular energies and forces. As a starting point for our analysis, we demonstrate how to train a Transformer to competitive energy and force mean absolute errors under a matched training compute budget, relative to a state-of-the-art equivariant GNN on the OMol25 dataset. We discover that the Transformer learns physically consistent patterns$\unicode{x2013}$such as attention weights that decay inversely with interatomic distance$\unicode{x2013}$and flexibly adapts them across different molecular environments due to the absence of hard-coded biases. The use of a standard Transformer also unlocks predictable improvements with respect to scaling training resources, consistent with empirical scaling laws observed in other domains. Our results demonstrate that many favorable properties of GNNs can emerge adaptively in Transformers, challenging the necessity of hard-coded graph inductive biases and pointing toward standardized, scalable architectures for molecular modeling.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。