首页   

NeurIPS 2024 | DePLM:用扩散去噪过程帮助大模型进行蛋白质优化

将门创投  · 科技创业  · 4 天前

正文

本文提出了一种名为DePLM的方法,通过特征编码器获取表达性蛋白质表示,并基于这些表示从嘈杂的进化似然中提取特定于属性的似然,进而用于突变效应预测。实验结果表明,DePLM不仅超越了现有技术的基线表现,还展现了出色的泛化能力。
进一步的分析表明,使用更大规模的数据集或结合其他相关属性的数据,能够显著提升模型性能。虽然实验中由于资源限制使用了野生型边缘概率进行预测,这种方法通过累加单个突变效应来估计多重突变的影响,但其局限性在于未能考虑突变之间的复杂相互作用。DePLM有潜力通过引入更高效的预测技术,例如掩蔽边缘概率,进一步提高多重突变预测的准确性。

论文题目: 

DePLM: Denoising Protein Language Models for Property Optimizations 

论文链接:

https://neurips.cc/virtual/2024/poster/95517

代码链接:

https://github.com/HICAI-ZJU/DePLM


一、动机

蛋白质优化是一项基础的生物学任务,旨在通过修改蛋白质序列来增强其性能。在自然界中,蛋白质经过数十亿年的进化,形成了多样化的结构和功能。这种进化的多样性为药物发现和材料科学等领域的进步提供了重要机遇。然而,现有蛋白质的固有属性,如热稳定性,往往不能满足各种实际场景的需求。因此,研究人员致力于优化蛋白质以增强其感兴趣的属性。传统的深度突变扫描(DMS)和定向进化(DE)依赖于昂贵的湿实验。近年来,计算方法通过准确模拟蛋白质与其属性适应度之间的关系,即所谓的“适应度景观”,在高效蛋白质优化中变得至关重要。

二、贡献

本文介绍了一种新颖的方法——去噪蛋白质语言模型(DePLM),用于改进蛋白质适应度预测。DePLM的核心思想是将蛋白质语言模型(PLMs)捕获的进化信息(EI)视为包含属性相关和不相关信息的混合体,其中不相关信息对于目标属性的优化任务类似于“噪声”,需要被消除。DePLM通过在属性值的排序空间中进行扩散过程来去噪EI,从而增强模型的泛化能力,并确保与数据集无关的学习。通过广泛的实验结果,证明了DePLM不仅在突变效应预测方面超越了最新技术,而且对新蛋白质展现出强大的泛化能力。

三、方法

DePLM的设计目标是从PLMs产生的嘈杂进化似然中过滤掉不相关信息。给定一个野生型蛋白质,PLM产生的进化似然可以表示为 ,其中 表示在位置 发生20种氨基酸的概率。这个似然可以分解为目标属性似然 和由不相关属性 引入的噪声似然,使得 。DePLM接受嘈杂的似然 作为输入,并通过基于排序的去噪扩散过程来提炼它,以隔离期望的似然
DePLM的框架包括两个主要过程:正向扩散过程和学习的逆向去噪过程。正向过程中,逐渐向真实情况添加少量噪声。逆向去噪过程然后学习通过逐步消除累积的噪声来恢复真实情况。DePLM提出了一种基于排序的去噪扩散过程,专注于最大化排序相关性。给定真实似然的目标属性排序 和噪声进化似然的排序 ,沿着这个过程序列生成中间排序变量
DePLM首先学习蛋白质表示(特征编码器),然后使用它来指导识别和消除噪声的过程(去噪块)。特征编码器从序列和结构中编码特征,因为它们互补地描述了突变的影响。结构信息通过训练有素的结构编码器(如ESM-IF)获得,处理蛋白质主链以产生结构表示 。然后将这两组表示使用多层感知机(MLPs)合并,得到统一的表示
去噪块给定 步的中间似然变量 ,使用去噪块来实现方程。核心前提是 应该只包含特定于属性的蛋白质信息。通过从噪声 的隐藏表示中减去 ,隔离噪声的隐藏表示。然后,这个隐藏表示被转换为似然空间。

四、实验

在实验部分,作者在多个数据集上广泛评估了DePLM,并证明了其卓越的性能和强大的泛化能力。实验旨在回答以下关键问题:

1.能比较(Q1) :DePLM能否在蛋白质适应度预测任务上击败最新技术?

通过在ProteinGym、β-Lactamase(β-乳酸酶)和Fluorescence(荧光)等四个基准测试中的实验,DePLM与九个基线模型进行了比较,包括CNN、ResNet、LSTM和Transformer等。结果显示,DePLM在性能上优于这些基线,证实了将进化信息与实验数据集成在蛋白质工程任务中的优势。

2. 泛化能力(Q2):DePLM在训练后是否保持其泛化能力?

使用ProteinGym测试了这一泛化能力。ProteinGym根据它们测量的蛋白质属性将DMS数据集分为五个粗略类别:稳定性、适应度、表达、结合和活性。在实验中,DePLM与四个自监督基线、两个基于结构的基线和三个监督基线进行了比较。结果表明,DePLM一致性地超越了所有基线模型,强调了仅依赖未经过滤的进化信息的基线的不足。

3. 消融研究(Q3):每个组件的改进程度是多少?

消融研究旨在验证DePLM设计的模块的有效性,包括使用结构信息、特征编码器、去噪模块和排序目标。结果表明,缺少这些模块中的任何一个都会导致性能下降,表明了它们的集体重要性。

4. 分析(Q4):进化信息中包含噪声的假设是否成立?

为了确定过滤掉与属性无关的信息的重要性,分析了使用针对不同优化目标的数据集进行训练的影响。结果表明,使用与测试数据集特性不同的训练数据集会导致稳定性、表达和活性属性的性能下降。这一发现强调了不同属性之间的不利干扰,强调了减轻这些不利影响的必要性。

五、总结

本文提出的DePLM方法,通过利用特征编码器获得表达性蛋白质表示,并使用它们从嘈杂的进化似然中提取特定于属性的似然,用于突变效应预测。实验表明,DePLM不仅超越了最新技术的基线,还显示出卓越的泛化能力。

此外,分析证实了使用足够大的数据集或结合其他相关属性的数据可以显著提高性能。尽管由于资源限制,实验是使用野生型边缘概率进行的,这种方法预测所有突变的影响,在单一前向传递中,通过累加个体突变的影响来估计多重突变的后果。然而,这种方法并不理想,因为它忽略了突变之间的复杂相互作用。DePLM有潜力通过利用更有效的预测技术,如掩蔽边缘概率,来实现更好的性能,以预测多重突变的影响。

作者:王泽元
来源:公众号【ZJUKG
llustration From IconScout By IconScout Store

-The End-

扫码观看

本周上新!


“AI技术流”原创投稿计划


TechBeat是由将门创投建立的AI学习社区(www.techbeat.net社区上线500+期talk视频,3000+篇技术干货文章,方向覆盖CV/NLP/ML/Robotis等;每月定期举办顶会及其他线上交流活动,不定期举办技术人线下聚会交流活动。我们正在努力成为AI人才喜爱的高质量、知识型交流平台,希望为AI人才打造更专业的服务和体验,加速并陪伴其成长。


投稿内容

// 最新技术解读/系统性知识分享 //

// 前沿资讯解说/心得经历讲述 //


投稿须知

稿件需要为原创文章,并标明作者信息。

我们会选择部分在深度技术解析及科研心得方向,对用户启发更大的文章,做原创性内容奖励


投稿方式

发送邮件到

melodybai@thejiangmen.com

或添加工作人员微信(yellowsubbj投稿,沟通投稿详情;还可以关注“将门创投”公众号,后台回复“投稿”二字,获得投稿说明。


关于我“

将门是一家以专注于数智核心科技领域新型创投机构,也是北京市标杆型孵化器公司致力于通过连接技术与商业,发掘和培育具有全球影响力的科技创新企业,推动企业创新发展与产业升级。

将门成立于2015年底,创始团队由微软创投在中国的创始团队原班人马构建而成,曾为微软优选和深度孵化了126家创新的技术型创业公司。

如果您是技术领域的初创企业,不仅想获得投资,还希望获得一系列持续性、有价值的投后服务,欢迎发送或者推荐项目给我“门”: 
bp@thejiangmen.com

    


点击右上角,把文章分享到朋友圈
点击“阅读原文”按钮,查看社区原文

推荐文章
© 2024 精读
删除内容请联系邮箱 2879853325@qq.com