用双流图扩散模型提升质谱生成分子的精度。
Unlocking High-Fidelity Molecular Generation from Mass Spectra via Dual-Stream Line Graph Diffusion

- 分原子与键两路并行推理,通过交叉注意力同步信息。
- 在两个数据集上准确率超旧方法三倍,达34.37%和23.89%。
- 适合做分子生成与质谱分析的研究者参考。
从串联质谱中进行从头分子生成是一个具有挑战性的逆问题,其核心难点在于原子级与键级推理之间的循环依赖:确定键的类型需要知道其端点原子的化学环境,而原子的环境又由其连接的键定义。现有图扩散方法在单一计算流中处理原子与键,信息同步仅能隐式跨层实现。我们提出DualLGD(双流线图扩散),将分子图去噪重构为两个耦合子问题的交替求解:原子级推理与键级推理,各自在专用表示空间中运行。线图为键空间提供自然数学构造,其中键角、二面角、共轭链与环均表现为键间的局部拓扑模式。约束型双向交叉注意力在每层同步两路信息,确保原子仅关注其连接的键,反之亦然,符合化学基本原理。在NPLIB1和MassSpecGym基准上,DualLGD分别达到34.37%和23.89%的顶1准确率,约为之前最先进水平的3倍。消融实验确认该架构是主要提升来源:即使无预训练,DualLGD仍超越此前全预训练最优模型。
原文摘要 · Abstract (English)
De novo molecular generation from tandem mass spectra is a challenging inverse problem whose core difficulty lies in the circular dependency between atom-level and bond-level reasoning: determining a bond's type requires knowing its endpoint atoms' chemical environment, yet an atom's environment is in turn defined by its incident bonds. Existing graph diffusion methods process atoms and bonds within a single computation stream, where atom-bond information synchronization can only occur implicitly across layers. We argue that this single-stream paradigm, rather than the choice of any particular aggregation kernel, is a key architectural bottleneck. We propose DualLGD (Dual-stream Line Graph Diffusion), which reformulates molecular graph denoising as the alternating solution of two coupled subproblems: atom-level reasoning and bond-level reasoning, each operating in its own dedicated representation space. The line graph provides a natural mathematical construction for the bond space, in which bond angles, dihedrals, conjugation chains, and rings correspond to local topological motifs between bonds. Incidence-constrained bidirectional cross-attention synchronizes the two streams at every layer, ensuring that each atom attends only to its incident bonds and vice versa, respecting the fundamental chemical principle that an atom's environment is determined by its bonding context. On the NPLIB1 and MassSpecGym benchmarks, DualLGD achieves top-1 accuracy of 34.37\% and 23.89\%, approximately $3\times$ the previous state of the art. Ablation studies confirm the architecture as the primary source of improvement: DualLGD without any pre-training already surpasses the previous best fully pretrained model.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。