DeepMind开源AlphaFold，蛋白质预测模型登上《Nature》

AI科技评论 · 公众号 · AI · 2020-01-19 13:28

正文

作者 | 十、年

编辑 | Camel

还记得AlphaFold成名的那一战么？

2018年的11月2日，在第13届全球蛋白质结构预测竞赛（CASP）上，AlphaFold获得了预测43种蛋白中的25种蛋白结构的最高分，在98名参赛者中排名第一。

对于DeepMind的预测方法，由于当时没有具体论文发布，众多学者认为是计算能力突出使得AlphaFold获得冠军。

1月15日，DeepMind关于AlphaFold模型与代码通过了同行评审发布在了杂志《nature》上面。并且模型和代码已经开源。

代码： https://github.com/deepmind/deepmind-research/tree/master/alphafold_casp13

模型：https://www.biorxiv.org/content/10.1101/846279v1.full.pdf

根据DeepMind的介绍，在预测蛋白质结构的物理性质方面使用了两种不同的方法来构建预测模型。并且两种方法都是基于深度神经网络来设计的，另外，预测模型主要完成对基因序列中蛋白质的特性的预测，这些特性包括：a：成对的氨基酸之间的距离；b：连接这些氨基酸的化学键之间的角度。

两种图片展示方式都表达了AlphaFold预测的准确度，像素亮度代表氨基酸之间的距离，像素颜色越亮，两个残基（residues）对就越近。第一行图片是真实距离，中间一行图片展示的是平均距离。

具体操作步骤为用神经网络预测蛋白质中的每一对残基的概率分布，然后将这些概率合并为一个分数，从而能够估计预测蛋白质结构的准确性。另外，还训练了一个单独的神经网络，基于所有距离的总和来评估预测的蛋白质结构与实际的结构的接近程度。然后使用评分系统（也是用神经网络构建）找到最优的预测。

整个蛋白质结构预测的过程

第二种方法是采用梯度下降优化分数，从而达到更准确的精度。更具体的是将梯度下降应用在整个蛋白质结构链的预测过程，而不仅仅是蛋白质展开的“碎片”。

模型简介

https://www.biorxiv.org/content/10.1101/846279v1.full.pdf

根据DeepMind公开的论文《Improved protein structure prediction using predicted inter-residue orientations》。AlphaFold由深度学习和传统算法混合而成：CNN+Rosetta。

具体来说构建神经网络预测残基间的方向和距离，然后用Rosetta对蛋白质结构同源建模、结构修复。同时设计一种Rosetta的优化方法来补充Rosetta能量函数的预测约束，进而生成更精确的模型。尽管训练数据集全部为天然的蛋白质，但该模型始终将更高的概率分配给新设计的蛋白质，并找到决定残基的关键折叠和建立蛋白质结构“理想性（ideality）”的独立定量衡量标准。

更为具体的，模型的由两个关键部分组成：一个是深度残差卷积神经网络，即将多序列比对（ multiple sequence alignments）作为输入；输出的信息为蛋白质中的残基对中的相对距离和方向。另一个是在网络输出的基础上针对残基对的距离和方向最小化的约束建立快速Rosetta模型（fast Rosetta model）

在训练数据集方面，DeepMind使用的蛋白质数据库（PDB）里面有15051个蛋白质链条信息，其中有30%是被标注的数据。

关于对模型的测试，DeepMind使用的是两个独立的测试集：第一个来自CASP13，第二个来自CAMEO实验。另外，在CASP13数据集上，DeepMind使用完整的蛋白质序列而不是结构域序列来模拟。

图注：A：用角度和距离表示从一个残基到另一残基的转换。B：神经网络结构根据MSA预测残基间的几何形状；C：预测过程概览

在残基间的几何预测方面，DeepMind使用深度残差神经网络。与大多数预测方法不同，DeepMind除了预测

之间的距离之外，残基对之间的方位也属于其预测的范围。如上图所示，残基1与残基2之间的方位由3个二面角

以及2个平面角

表示。其中ω表示沿虚轴（连接两个残基的

原子）旋转角度。

角度定义了从残基1看到残基2的Cβ原子的方向，

同理。另外，与d和ω不同，θ和φ坐标是不对称的，其取决于残基的顺序。综上6个参数d，ω，

定义了两个残基的主干原子的相对位置，这6个参数也是神经网络所要预测的。

神经网络的输入是从MSA提取的特征，并且执行动态运算。输入的一维特征包括：蛋白质氨基酸序列的独热编码、位置特定频率矩阵（position-specific frequency matrix）、配位熵（positional entropy）。随后这些一维特征被水平和垂直平铺，然后堆叠在一起以产生2×42=84个2维特征映射。

此外，DeepMind从MSA中提取配对统计信息，这些信息从输入MSA构造的缩小协方差矩阵的逆导出的耦合来表示。首先计算一个站点（one-site）和两个站点（two-site）的频率计数，计算公式如下：