用扩散桥模型生成能折叠成目标结构的蛋白序列。
Bridge-IF: Learning Inverse Protein Folding with Markov Bridges
- 基于马尔可夫桥建模结构与序列的生成关系。
- 在多个基准上优于现有方法,设计出高可折叠性蛋白。
- 结合结构条件调优语言模型,高效生成合理序列。
逆向蛋白折叠是计算蛋白设计中的基础任务,旨在设计能折叠成特定骨架结构的蛋白序列。尽管机器学习方法已取得显著进展,但主流判别式方法常出现误差累积问题,且难以捕捉多样化的合理序列。为此,我们提出Bridge-IF,一种用于逆向折叠的生成式扩散桥模型,旨在学习骨架结构与蛋白序列分布间的概率依赖关系。具体地,我们使用一个表达能力强的结构编码器,从结构中推导出离散且信息丰富的先验,并建立马尔可夫桥将该先验连接到天然序列。推理阶段,Bridge-IF逐步优化先验序列,生成更合理的蛋白设计。此外,我们从重参数化视角重新审视马尔可夫桥模型,推导出简化损失函数,提升训练效率。同时,通过结构条件调制蛋白语言模型(PLMs),精确逼近马尔可夫桥过程,在保持参数高效训练的同时显著提升生成性能。大量实验表明,Bridge-IF在序列恢复和可折叠蛋白设计方面均优于现有基线。代码已公开于 https://github.com/violet-sto/Bridge-IF。
原文摘要 · Abstract (English)
Inverse protein folding is a fundamental task in computational protein design, which aims to design protein sequences that fold into the desired backbone structures. While the development of machine learning algorithms for this task has seen significant success, the prevailing approaches, which predominantly employ a discriminative formulation, frequently encounter the error accumulation issue and often fail to capture the extensive variety of plausible sequences. To fill these gaps, we propose Bridge-IF, a generative diffusion bridge model for inverse folding, which is designed to learn the probabilistic dependency between the distributions of backbone structures and protein sequences. Specifically, we harness an expressive structure encoder to propose a discrete, informative prior derived from structures, and establish a Markov bridge to connect this prior with native sequences. During the inference stage, Bridge-IF progressively refines the prior sequence, culminating in a more plausible design. Moreover, we introduce a reparameterization perspective on Markov bridge models, from which we derive a simplified loss function that facilitates more effective training. We also modulate protein language models (PLMs) with structural conditions to precisely approximate the Markov bridge process, thereby significantly enhancing generation performance while maintaining parameter-efficient training. Extensive experiments on well-established benchmarks demonstrate that Bridge-IF predominantly surpasses existing baselines in sequence recovery and excels in the design of plausible proteins with high foldability. The code is available at https://github.com/violet-sto/Bridge-IF.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。