doi.org/10.1186/s13321-023-00702-2
本文介绍 2023 年发布在 Journal of Cheminformatics 上的一篇文章,文章原标题为 Deep generative model for drug design from protein target sequence,文章设计了一种基于 GAN 的蛋白配体分子生成模型,该模型只需要获取氨基酸序列的信息就可以生成相应蛋白口袋的配体。
目前的分子生成方法可以分为两类,其中一种是基于配体的分子生成(ligand-based molecule generation, LBMG),另一种是基于口袋的分子生成(pocketbased molecule generation, PBMG)。LBMG 方法难以跳出目前的化空间,因而难以生成具有新颖结构的分子;PBMG 方法需要获取更多蛋白口袋的信息,但计算蛋白 3D 构象通常开销巨大。
文章提出了一种输入蛋白序列即可获得配体的分子生成模型,称为 DeepTarget。DeepTarget 既不需要考虑蛋白口袋的构象信息,也不需要在特定的分子库上微调,有效避免了上述两种方法的局限。
方法
数据
文章使用了来自于 ChEMBL 的分子-蛋白对数据,经数据清洗后,共得到 551223 个分子-蛋白对,涉及 1970 种蛋白质与 333399 种分子的 SMILES 序列。
模型
DeepTarget 由 3 个部分构成,分别是氨基酸序列嵌入(Amino Acid Sequence Embedding, AASE)、结构特征推理(Structural Feature Inference, SFI)和分子生成(Molecule Generation, MG)。
- 氨基酸序列嵌入:AASE 是模型的嵌入层,使用了 Transformer 的架构,主要用于将序列数据转化为模型计算并处理的特征向量。
- 结构特征推理:SFI 部分采用了 GAN 的结构,其主要任务是在 AASE 中得到蛋白特征表示上加上一定的噪声,再通过多层感知机得到潜变量 \(\boldsymbol{z}\)。
- 分子生成:MG 部分使用了 LSTM 结构,是一个预训练的解码器,它将潜变量 \(\boldsymbol{z}\) 解码为目标分子。该解码器是一个在 ChEMBL 大数据集上训练好的模型,能将分子的潜变量转换为相似的分子。
SFI 中的 GAN 模型是 DeepTarget 生成蛋白配体的关键,GAN 由生成器与分别器两个模块构成,生成器从潜变量 \(\boldsymbol{z}\) 生成分子,而分别器则识别生成分子与该蛋白口袋之间的关系,训练 GAN 就是让生成器不断生成分子,直至生成的分子可以「欺骗」分别器,也就是此时生成的分子满足该蛋白口袋的配体分布特征。在推理阶段,则将此生成分子的表示送入 MG 中得到分子的 SMILES 编码。
传统的生成模型关注于生成器与生成结果之间的关系,而在文章所设计的任务中,不同的蛋白口袋与配体分子在化学空间中有着不同的分布,这也是影响分子生成的因素。因此文章引入了对比学习(Contrastive Learning, CL)的手段,将不同的蛋白作为标签而使分子分簇,在相应的化学空间中生成分子。
结果与讨论
文章首先针对 DRD2 和 PARP1 两个蛋白的活性口袋生成分子。图 a 展示了对应的真实配体分子与生成分子的对接打分,生成分子相对于真实分子向打分更低方向偏移,说明具有更高的亲合力。
图 b 挑出了生成分子中的代表分子,与训练集分子计算相似性,两个分子与训练集的相似性都在 0.2-0.6 左右,说明这两个分子与训练集分子存在一定差异,DeepTarget 能生成新颖的分子。
图 c 展示了是否在模型中引入对比学习的了生成结果,使用对比学习策略模型生成的分子明显向打分更低处偏移,具有更好的效果。
图 d 测试了模型的泛化能力,先在训练集中删去 DRD2 和 PARP1 两个蛋白的数据,将生成分子与先前生成的分子对比,从测试结果中可以看出,删除相应训练数据后,生成分子的对接打分上升,但还是生成了相当数量打分低于 -6 的分子,作者认为这可以说明 DeepTarget 能针对训练集中不存在的活性口袋生成配体分子。
文章对 DeepTarget 生成的分子与其他模型针对这两个口袋生成的分子做了评估,结果如上表所示,DeepTarget 生成分子的 Valid 高于两个 GAN 模型,其他指标与其他模型相当。
作者认为这些指标只能做为模型的参考,因为模型并没有针对生成分子的 Valid 和 Unique 等指标进行优化,DeepTarget 的目标更着重于生成与指定口袋真正具有相互作用的配体分子。
结论
文章设计了一种基于 GAN 的蛋白配体分子生成模型,模型只需要获取氨基酸序列的信息就可以生成相应蛋白口袋的配体,文章验证了生成分子具有较好的对接打分,并且模型表现出了一定的泛化能力,可用于针对训练集以外的蛋白生成配体。但文章中设计的模型评估实验有限,只针对两个靶点生成了分子,从文章中的数据来看,与其他模型相比该模型没有特别明显的优势。