arXiv:2511.09774physics.chem-phcs.AI2025-11被引 2

用几何感知图注意力模型预测分子溶解度,兼具高精度与可解释性。

Solvaformer: an SE(3)-equivariant graph transformer for small molecule solubility prediction

  • 基于SE(3)等变图注意力,同时建模分子内与分子间相互作用。
  • 在BigSolDB 2.0和CombiSolv-QM数据上实现最优性能,接近DFT基准。
  • 可生成化学合理的注意力热力图,揭示氢键对溶解度的影响机制。

利用数据节省型方法准确预测小分子溶解度对加速合成与工艺优化至关重要,但实验测量成本高,现有学习方法或依赖量子衍生描述符,或解释性有限。我们提出Solvaformer,一种几何感知的图变换器,将溶液建模为具有独立SE(3)对称性的多个分子。该架构结合分子内SE(3)等变注意力与分子间标量注意力,实现跨分子通信且不强加虚假相对几何关系。采用交替批次训练策略,在量子力学数据(CombiSolv-QM)与实验数据(BigSolDB 2.0)上进行多任务训练,以预测溶解度(log S)与溶剂化自由能。Solvaformer在所有学习模型中表现最佳,逼近基于DFT辅助的梯度提升基线,优于EquiformerV2消融实验与序列基线。此外,词元级注意力生成化学上一致的归因:案例研究复现了影响位置异构体溶解度差异的分子内与分子间氢键模式。综上,Solvaformer通过融合几何归纳偏置与混合数据训练策略,为溶液相性质预测提供了一种高精度、可扩展且可解释的新方法。

原文摘要 · Abstract (English)

Accurate prediction of small molecule solubility using material-sparing approaches is critical for accelerating synthesis and process optimization, yet experimental measurement is costly and many learning approaches either depend on quantumderived descriptors or offer limited interpretability. We introduce Solvaformer, a geometry-aware graph transformer that models solutions as multiple molecules with independent SE(3) symmetries. The architecture combines intramolecular SE(3)-equivariant attention with intermolecular scalar attention, enabling cross-molecular communication without imposing spurious relative geometry. We train Solvaformer in a multi-task setting to predict both solubility (log S) and solvation free energy, using an alternating-batch regimen that trains on quantum-mechanical data (CombiSolv-QM) and on experimental measurements (BigSolDB 2.0). Solvaformer attains the strongest overall performance among the learned models and approaches a DFT-assisted gradient-boosting baseline, while outperforming an EquiformerV2 ablation and sequence-based alternatives. In addition, token-level attention produces chemically coherent attributions: case studies recover known intra- vs. inter-molecular hydrogen-bonding patterns that govern solubility differences in positional isomers. Taken together, Solvaformer provides an accurate, scalable, and interpretable approach to solution-phase property prediction by uniting geometric inductive bias with a mixed dataset training strategy on complementary computational and experimental data.

分子性质预测图神经网络等变神经网络可解释性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。