arXiv:2502.09571cs.LGq-bio.QM2025-02ICML被引 47

用质谱生成分子结构,精度和效率均领先现有方法。

DiffMS: Diffusion Generation of Molecules Conditioned on Mass Spectra

  • 基于Transformer编码器与图扩散解码器,结合化学公式约束生成分子。
  • 在基准数据集上生成准确率超越现有模型,且预训练数据越大效果越优。
  • 适合需要从质谱逆向推导分子结构的化学家与药物研发人员。

质谱在解析未知分子结构及推动科学发现中起着基础性作用。一种结构解析的表述方式是根据质谱条件生成分子结构。为实现更精确高效的分子科学研究流程,我们提出DiffMS,一种受分子式约束的编码器-解码器生成网络,在该任务上达到当前最优性能。编码器采用Transformer架构,建模质谱领域的知识如峰公式和中性丢失;解码器是受限于已知化学式重原子组成的离散图扩散模型。为构建鲁棒解码器以连接潜在表征与分子结构,我们使用指纹-结构对进行预训练,其数量远超结构-质谱对(后者仅数万)。大量实验表明,DiffMS在去新分子生成任务上优于现有模型。多组消融实验证明了扩散机制与预训练策略的有效性,并显示性能随预训练数据量增加而持续提升。代码已公开于https://github.com/coleygroup/DiffMS。

原文摘要 · Abstract (English)

Mass spectrometry plays a fundamental role in elucidating the structures of unknown molecules and subsequent scientific discoveries. One formulation of the structure elucidation task is the conditional de novo generation of molecular structure given a mass spectrum. Toward a more accurate and efficient scientific discovery pipeline for small molecules, we present DiffMS, a formula-restricted encoder-decoder generative network that achieves state-of-the-art performance on this task. The encoder utilizes a transformer architecture and models mass spectra domain knowledge such as peak formulae and neutral losses, and the decoder is a discrete graph diffusion model restricted by the heavy-atom composition of a known chemical formula. To develop a robust decoder that bridges latent embeddings and molecular structures, we pretrain the diffusion decoder with fingerprint-structure pairs, which are available in virtually infinite quantities, compared to structure-spectrum pairs that number in the tens of thousands. Extensive experiments on established benchmarks show that DiffMS outperforms existing models on de novo molecule generation. We provide several ablations to demonstrate the effectiveness of our diffusion and pretraining approaches and show consistent performance scaling with increasing pretraining dataset size. DiffMS code is publicly available at https://github.com/coleygroup/DiffMS.

分子生成扩散模型质谱分析

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