arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.09896cs.LGphysics.chem-phq-bio.BM
收藏

基于统一文本-图扩散模型的指令式分子图生成

Instruction-Based Molecular Graph Generation with Unified Text-Graph Diffusion Model

State Key Laboratory of General Artificial Intelligence, School of Intelligence Science and Technology, Peking University(通用人工智能国家重点实验室,智能科学与技术学院,北京大学);State Key Laboratory of General Artificial Intelligence, School of Intelligence Science(通用人工智能国家重点实验室,智能科学学院);The University of Hong Kong(香港大学)

Yuran Xiang, Haiteng Zhao, Chang Ma, Zhi-Hong Deng

AI总结:

提出UTGDiff框架,利用语言模型进行离散图扩散,从文本指令生成分子图,以更少参数超越序列基线。

摘要:

近年来,计算化学领域的进展日益聚焦于根据文本指令合成分子。将图生成与这些指令相结合是复杂的,导致大多数现有方法采用分子序列并结合预训练的大型语言模型。针对这一挑战,我们提出了一种新颖的框架,名为UTGDiff(统一文本-图扩散模型),该框架利用语言模型进行离散图扩散,从而根据指令生成分子图。UTGDiff采用一个统一的文本-图变换器作为去噪网络,该网络源自预训练语言模型,并通过注意力偏置进行最小化修改以处理图数据。我们的实验结果表明,在涉及基于指令的分子生成和编辑的任务中,UTGDiff始终优于基于序列的基线方法,在预训练语料库规模相当的情况下,以更少的参数实现了更优的性能。我们的代码可在https://github.com/ran1812/UTGDiff获取。

英文摘要:

Recent advancements in computational chemistry have increasingly focused on synthesizing molecules based on textual instructions. Integrating graph generation with these instructions is complex, leading most current methods to use molecular sequences with pre-trained large language models. In response to this challenge, we propose a novel framework, named $\textbf{UTGDiff (Unified Text-Graph Diffusion Model)}$, which utilizes language models for discrete graph diffusion to generate molecular graphs from instructions. UTGDiff features a unified text-graph transformer as the denoising network, derived from pre-trained language models and minimally modified to process graph data through attention bias. Our experimental results demonstrate that UTGDiff consistently outperforms sequence-based baselines in tasks involving instruction-based molecule generation and editing, achieving superior performance with fewer parameters given an equivalent level of pretraining corpus. Our code is availble at https://github.com/ran1812/UTGDiff.

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

↑