arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.07077cs.LGphysics.chem-ph
收藏

D3MES:面向3D分子生成的带多头等变自注意力的扩散Transformer

D3MES: Diffusion Transformer with multihead equivariant self-attention for 3D molecule generation

School of Physics and Electronic Engineering, Jiangsu University(江苏大学物理与电子工程学院);Jiangsu Engineering Research Center on Quantum Perception and Intelligent Detection of Agricultural Information(江苏省农业信息量子感知与智能检测工程研究中心)

Zhejun Zhang, Yuanping Chen, Shibing Chu

AI总结:

本研究针对3D分子生成中氢原子连接难、跨类别生成受限的问题,提出融合多头等变自注意力的Diffusion Transformer扩散模型D3MES,在多项指标上达最优,适用于分子设计早期大规模生成。

摘要:

理解并预测分子的多样构象态对于推动化学、材料科学、药物研发等领域的发展至关重要。尽管生成模型已取得显著进展,但准确生成复杂的、与生物学或材料相关的分子结构仍是一项重大挑战。本研究提出一种用于三维(3D)分子生成的扩散模型,该模型将可分类扩散模型Diffusion Transformer与多头等变自注意力相结合。该方法解决了两大关键挑战:通过学习去除氢原子后的分子表示,在生成的分子中正确连接氢原子;同时克服现有模型无法同时生成多类别分子的局限性。实验结果表明,该模型不仅在多项关键指标上达到了state-of-the-art(当前最优)性能,还展现出鲁棒性与通用性,非常适用于分子设计早期的大规模生成流程,后续可通过验证与进一步筛选获得具备特定性质的分子。

英文摘要:

Understanding and predicting the diverse conformational states of molecules is crucial for advancing fields such as chemistry, material science, and drug development. Despite significant progress in generative models, accurately generating complex and biologically or material-relevant molecular structures remains a major challenge. In this work, we introduce a diffusion model for three-dimensional (3D) molecule generation that combines a classifiable diffusion model, Diffusion Transformer, with multihead equivariant self-attention. This method addresses two key challenges: correctly attaching hydrogen atoms in generated molecules through learning representations of molecules after hydrogen atoms are removed; and overcoming the limitations of existing models that cannot generate molecules across multiple classes simultaneously. The experimental results demonstrate that our model not only achieves state-of-the-art performance across several key metrics but also exhibits robustness and versatility, making it highly suitable for early-stage large-scale generation processes in molecular design, followed by validation and further screening to obtain molecules with specific properties.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑