用Mamba模型多角度理解分子结构,提升构象预测精度
Mamba-driven multi-perspective structural understanding for molecular ground-state conformation prediction
- 基于Mamba设计多视角结构理解框架,捕捉原子类型、位置与连接
- 在QM9和Molecule3D上超越现有方法,小样本下仍表现优异
- 适合需要精准构象预测的药物研发与分子设计场景
分子结构的全面理解对包含性质信息的分子基态构象预测至关重要。近年来,状态空间模型(如Mamba)在长序列建模中表现出色,已在语言和视觉任务中取得显著成果,但在分子基态构象预测中的应用尚不充分。为此,本文提出一种通用高效的Mamba驱动多视角结构理解框架(MPSU-Mamba),通过原子类型、原子位置及原子间连接三个要素构建分子结构感知。针对复杂多样的分子,探索三种专用扫描策略以实现全面结构感知,并引入亮通道引导机制识别关键构象相关原子信息。在QM9和Molecule3D数据集上的实验表明,MPSU-Mamba显著优于现有方法;尤其在少量训练样本情况下仍保持优异性能,证明其对分子结构理解具有实际价值。
原文摘要 · Abstract (English)
A comprehensive understanding of molecular structures is important for the prediction of molecular ground-state conformation involving property information. Meanwhile, state space model (e.g., Mamba) has recently emerged as a promising mechanism for long sequence modeling and has achieved remarkable results in various language and vision tasks. However, towards molecular ground-state conformation prediction, exploiting Mamba to understand molecular structure is underexplored. To this end, we strive to design a generic and efficient framework with Mamba to capture critical components. In general, molecular structure could be considered to consist of three elements, i.e., atom types, atom positions, and connections between atoms. Thus, considering the three elements, an approach of Mamba-driven multi-perspective structural understanding (MPSU-Mamba) is proposed to localize molecular ground-state conformation. Particularly, for complex and diverse molecules, three different kinds of dedicated scanning strategies are explored to construct a comprehensive perception of corresponding molecular structures. And a bright-channel guided mechanism is defined to discriminate the critical conformation-related atom information. Experimental results on QM9 and Molecule3D datasets indicate that MPSU-Mamba significantly outperforms existing methods. Furthermore, we observe that for the case of few training samples, MPSU-Mamba still achieves superior performance, demonstrating that our method is indeed beneficial for understanding molecular structures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。