arXiv:2505.18181cs.LGcs.AI2025-05被引 3

首个面向2D NMR分子表征学习的标注数据集,支持机器学习模型训练与评估

2DNMRGym: An Annotated Experimental Dataset for Atom-Level Molecular Representation Learning in 2D NMR via Surrogate Supervision

  • 通过算法生成标注实现代理监督,模拟真实专家标注流程
  • 包含超过2.2万张HSQC谱图及对应分子结构信息
  • 适合从事分子表征、NMR分析与图神经网络研究者使用

二维核磁共振(2D NMR)谱学,尤其是异核单量子相干(HSQC)谱,在解析分子结构、相互作用和电子性质方面具有关键作用。然而,准确解读2D NMR数据仍高度依赖经验丰富的领域专家,尤其对复杂分子而言耗时且易出错。机器学习在2D NMR分析中展现出巨大潜力,可通过学习分子表示识别复杂模式。但进展受限于大规模高质量标注数据集的缺乏。本文提出2DNMRGym,首个专为2D NMR中基于机器学习的分子表征学习设计的标注实验数据集,包含超过22,000张HSQC谱图,以及对应的分子图和SMILES字符串。其独特之处在于采用代理监督机制:模型使用先前验证方法生成的算法标注进行训练,并在保留的人工标注黄金标准集上评估,从而严格检验模型从不完美监督到专家级解释的泛化能力。我们提供了多种2D/3D图神经网络与图注意力网络的基准结果,为未来研究奠定基础。2DNMRGym支持可扩展模型训练,引入了化学意义明确的原子级别分子表征评估基准。数据与代码已开源,发布于Huggingface与Github。

原文摘要 · Abstract (English)

Two-dimensional (2D) Nuclear Magnetic Resonance (NMR) spectroscopy, particularly Heteronuclear Single Quantum Coherence (HSQC) spectroscopy, plays a critical role in elucidating molecular structures, interactions, and electronic properties. However, accurately interpreting 2D NMR data remains labor-intensive and error-prone, requiring highly trained domain experts, especially for complex molecules. Machine Learning (ML) holds significant potential in 2D NMR analysis by learning molecular representations and recognizing complex patterns from data. However, progress has been limited by the lack of large-scale and high-quality annotated datasets. In this work, we introduce 2DNMRGym, the first annotated experimental dataset designed for ML-based molecular representation learning in 2D NMR. It includes over 22,000 HSQC spectra, along with the corresponding molecular graphs and SMILES strings. Uniquely, 2DNMRGym adopts a surrogate supervision setup: models are trained using algorithm-generated annotations derived from a previously validated method and evaluated on a held-out set of human-annotated gold-standard labels. This enables rigorous assessment of a model's ability to generalize from imperfect supervision to expert-level interpretation. We provide benchmark results using a series of 2D and 3D GNN and GNN transformer models, establishing a strong foundation for future work. 2DNMRGym supports scalable model training and introduces a chemically meaningful benchmark for evaluating atom-level molecular representations in NMR-guided structural tasks. Our data and code is open-source and available on Huggingface and Github.

分子表征NMR分析图神经网络数据集

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