arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.18567cs.LGq-bio.BM
收藏

扩散语言模型是多功能的蛋白质学习器

Diffusion Language Models Are Versatile Protein Learners

ByteDance Research(字节跳动研究院);Nanjing University(南京大学)

Xinyou Wang, Zaixiang Zheng, Fei Ye, Dongyu Xue, Shujian Huang, Quanquan Gu

AI总结:

本文提出扩散蛋白质语言模型DPLM,通过生成式自监督离散扩散框架预训练,在蛋白质序列的无条件生成、预测任务及多种条件生成任务中均展现出卓越性能。

摘要:

本文介绍了扩散蛋白质语言模型(DPLM),这是一种多功能的蛋白质语言模型,对蛋白质序列展现出强大的生成和预测能力。我们首先在生成式自监督离散扩散概率框架内,从进化规模的蛋白质序列预训练可扩展的DPLM,该框架以有原则的方式推广了蛋白质的语言建模。预训练后,DPLM展现出为无条件生成生成结构合理、新颖且多样的蛋白质序列的能力。我们进一步证明,所提出的扩散生成式预训练使DPLM具备更好的蛋白质理解能力,使其成为优越的表示学习器,可针对各种预测任务进行微调,与ESM2(Lin等人,2022)相比表现更优。此外,DPLM可针对各种需求进行定制,在多个方面展示了其条件生成能力:(1)以部分肽序列为条件,例如,以高成功率为功能motif生成支架;(2)整合其他模态作为条件器,例如,用于反向折叠的结构条件生成;(3)通过即插即用的分类器引导,将序列生成导向所需属性,例如满足指定的二级结构。代码发布在\url{https://github.com/bytedance/dplm}。

英文摘要:

This paper introduces diffusion protein language model (DPLM), a versatile protein language model that demonstrates strong generative and predictive capabilities for protein sequences. We first pre-train scalable DPLMs from evolutionary-scale protein sequences within a generative self-supervised discrete diffusion probabilistic framework, which generalizes language modeling for proteins in a principled way. After pre-training, DPLM exhibits the ability to generate structurally plausible, novel, and diverse protein sequences for unconditional generation. We further demonstrate the proposed diffusion generative pre-training makes DPLM possess a better understanding of proteins, making it a superior representation learner, which can be fine-tuned for various predictive tasks, comparing favorably to ESM2 (Lin et al., 2022). Moreover, DPLM can be tailored for various needs, which showcases its prowess of conditional generation in several ways: (1) conditioning on partial peptide sequences, e.g., generating scaffolds for functional motifs with high success rate; (2) incorporating other modalities as conditioner, e.g., structure-conditioned generation for inverse folding; and (3) steering sequence generation towards desired properties, e.g., satisfying specified secondary structures, through a plug-and-play classifier guidance. Code is released at \url{https://github.com/bytedance/dplm}.

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

↑