CN114678061A - 基于预训练语言模型的蛋白质构象感知表示学习方法 - Google Patents
基于预训练语言模型的蛋白质构象感知表示学习方法 Download PDFInfo
- Publication number
- CN114678061A CN114678061A CN202210122014.5A CN202210122014A CN114678061A CN 114678061 A CN114678061 A CN 114678061A CN 202210122014 A CN202210122014 A CN 202210122014A CN 114678061 A CN114678061 A CN 114678061A
- Authority
- CN
- China
- Prior art keywords
- protein
- representation
- task
- amino acid
- embedded
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Pending
Links
- 108090000623 proteins and genes Proteins 0.000 title claims abstract description 206
- 102000004169 proteins and genes Human genes 0.000 title claims abstract description 203
- 238000000034 method Methods 0.000 title claims abstract description 40
- 238000012549 training Methods 0.000 title claims abstract description 39
- 230000008447 perception Effects 0.000 title claims abstract description 9
- 125000003275 alpha amino acid group Chemical group 0.000 claims abstract description 30
- 230000006870 function Effects 0.000 claims abstract description 25
- 238000010276 construction Methods 0.000 claims abstract description 4
- 239000010410 layer Substances 0.000 claims description 56
- 150000001413 amino acids Chemical class 0.000 claims description 47
- 238000013507 mapping Methods 0.000 claims description 28
- 230000004927 fusion Effects 0.000 claims description 22
- 230000009471 action Effects 0.000 claims description 19
- 230000004853 protein function Effects 0.000 claims description 8
- 238000006757 chemical reactions by type Methods 0.000 claims description 4
- GOJUJUVQIVIZAV-UHFFFAOYSA-N 2-amino-4,6-dichloropyrimidine-5-carbaldehyde Chemical group NC1=NC(Cl)=C(C=O)C(Cl)=N1 GOJUJUVQIVIZAV-UHFFFAOYSA-N 0.000 claims description 3
- 239000002355 dual-layer Substances 0.000 claims description 2
- 230000014509 gene expression Effects 0.000 abstract description 27
- 230000006916 protein interaction Effects 0.000 description 13
- 238000003780 insertion Methods 0.000 description 11
- 230000037431 insertion Effects 0.000 description 11
- 230000003993 interaction Effects 0.000 description 8
- 238000010586 diagram Methods 0.000 description 6
- 230000008569 process Effects 0.000 description 6
- 101710177166 Phosphoprotein Proteins 0.000 description 4
- 239000013598 vector Substances 0.000 description 4
- 239000011159 matrix material Substances 0.000 description 3
- 101710141454 Nucleoprotein Proteins 0.000 description 2
- 102100034574 P protein Human genes 0.000 description 2
- 101710181008 P protein Proteins 0.000 description 2
- 238000007792 addition Methods 0.000 description 2
- 239000000470 constituent Substances 0.000 description 2
- 230000000694 effects Effects 0.000 description 2
- 230000004044 response Effects 0.000 description 2
- 230000002269 spontaneous effect Effects 0.000 description 2
- 239000002253 acid Substances 0.000 description 1
- 230000004913 activation Effects 0.000 description 1
- 230000009286 beneficial effect Effects 0.000 description 1
- 238000006555 catalytic reaction Methods 0.000 description 1
- 238000006243 chemical reaction Methods 0.000 description 1
- 238000002474 experimental method Methods 0.000 description 1
- 238000002372 labelling Methods 0.000 description 1
- 229920002521 macromolecule Polymers 0.000 description 1
- 230000000873 masking effect Effects 0.000 description 1
- 230000004048 modification Effects 0.000 description 1
- 238000012986 modification Methods 0.000 description 1
- 230000004660 morphological change Effects 0.000 description 1
- 238000005457 optimization Methods 0.000 description 1
- 238000007500 overflow downdraw method Methods 0.000 description 1
- 230000004481 post-translational protein modification Effects 0.000 description 1
- 125000002924 primary amino group Chemical group [H]N([H])* 0.000 description 1
- 238000012545 processing Methods 0.000 description 1
- 238000005070 sampling Methods 0.000 description 1
- 238000000547 structure data Methods 0.000 description 1
- 239000000126 substance Substances 0.000 description 1
- 238000012706 support-vector machine Methods 0.000 description 1
Images
Classifications
-
- G—PHYSICS
- G16—INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR SPECIFIC APPLICATION FIELDS
- G16B—BIOINFORMATICS, i.e. INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR GENETIC OR PROTEIN-RELATED DATA PROCESSING IN COMPUTATIONAL MOLECULAR BIOLOGY
- G16B15/00—ICT specially adapted for analysing two-dimensional or three-dimensional molecular structures, e.g. structural or functional relations or structure alignment
- G16B15/20—Protein or domain folding
-
- G—PHYSICS
- G16—INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR SPECIFIC APPLICATION FIELDS
- G16B—BIOINFORMATICS, i.e. INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR GENETIC OR PROTEIN-RELATED DATA PROCESSING IN COMPUTATIONAL MOLECULAR BIOLOGY
- G16B35/00—ICT specially adapted for in silico combinatorial libraries of nucleic acids, proteins or peptides
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F18/00—Pattern recognition
- G06F18/20—Analysing
- G06F18/21—Design or setup of recognition systems or techniques; Extraction of features in feature space; Blind source separation
- G06F18/214—Generating training patterns; Bootstrap methods, e.g. bagging or boosting
-
- G—PHYSICS
- G16—INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR SPECIFIC APPLICATION FIELDS
- G16B—BIOINFORMATICS, i.e. INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR GENETIC OR PROTEIN-RELATED DATA PROCESSING IN COMPUTATIONAL MOLECULAR BIOLOGY
- G16B15/00—ICT specially adapted for analysing two-dimensional or three-dimensional molecular structures, e.g. structural or functional relations or structure alignment
-
- G—PHYSICS
- G16—INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR SPECIFIC APPLICATION FIELDS
- G16B—BIOINFORMATICS, i.e. INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR GENETIC OR PROTEIN-RELATED DATA PROCESSING IN COMPUTATIONAL MOLECULAR BIOLOGY
- G16B40/00—ICT specially adapted for biostatistics; ICT specially adapted for bioinformatics-related machine learning or data mining, e.g. knowledge discovery or pattern finding
-
- G—PHYSICS
- G16—INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR SPECIFIC APPLICATION FIELDS
- G16B—BIOINFORMATICS, i.e. INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR GENETIC OR PROTEIN-RELATED DATA PROCESSING IN COMPUTATIONAL MOLECULAR BIOLOGY
- G16B40/00—ICT specially adapted for biostatistics; ICT specially adapted for bioinformatics-related machine learning or data mining, e.g. knowledge discovery or pattern finding
- G16B40/20—Supervised data analysis
-
- G—PHYSICS
- G16—INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR SPECIFIC APPLICATION FIELDS
- G16B—BIOINFORMATICS, i.e. INFORMATION AND COMMUNICATION TECHNOLOGY [ICT] SPECIALLY ADAPTED FOR GENETIC OR PROTEIN-RELATED DATA PROCESSING IN COMPUTATIONAL MOLECULAR BIOLOGY
- G16B5/00—ICT specially adapted for modelling or simulations in systems biology, e.g. gene-regulatory networks, protein interaction networks or metabolic networks
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Health & Medical Sciences (AREA)
- Medical Informatics (AREA)
- Bioinformatics & Cheminformatics (AREA)
- Spectroscopy & Molecular Physics (AREA)
- Theoretical Computer Science (AREA)
- Bioinformatics & Computational Biology (AREA)
- Evolutionary Biology (AREA)
- Biophysics (AREA)
- Biotechnology (AREA)
- General Health & Medical Sciences (AREA)
- Data Mining & Analysis (AREA)
- Evolutionary Computation (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Artificial Intelligence (AREA)
- Databases & Information Systems (AREA)
- Public Health (AREA)
- Bioethics (AREA)
- Software Systems (AREA)
- Epidemiology (AREA)
- Molecular Biology (AREA)
- Chemical & Material Sciences (AREA)
- Physiology (AREA)
- Crystallography & Structural Chemistry (AREA)
- General Engineering & Computer Science (AREA)
- General Physics & Mathematics (AREA)
- Biochemistry (AREA)
- Library & Information Science (AREA)
- Management, Administration, Business Operations System, And Electronic Commerce (AREA)
Abstract
本发明公开了一种基于预训练语言模型的蛋白质构象感知表示学习方法,包括:获取由氨基酸序列组成的蛋白质,根据蛋白质构象构建不同数据集,为每类蛋白质构象定义提示符;基于预训练语言模型构建表示学习模块,用于将每类提示符的嵌入表示融合到蛋白质的嵌入表示,以得到提示符标识下的蛋白质嵌入表示;构建任务模块,用于针对每类蛋白质构象对应的任务,基于提示符标识下的蛋白质嵌入表示进行任务预测;基于任务预测结果和标签构建每类任务的损失函数,结合所有类任务的损失函数和不同数据集,更新表示学习模块和任务模块的模型参数;模型参数更新结束后,提取表示学习模块作为蛋白质表示模块。该方法能够得到不同构象下的蛋白质表示。
Description
技术领域
本发明属于表示学习领域,具体涉及一种基于预训练语言模型的蛋白质构象感知表示学习方法。
背景技术
蛋白质是由数个氨基酸组成的序列数据,目前有众多研究利用自然语言中的语言模型学习蛋白质的嵌入表示,以预测蛋白质的结构和性质。
其中,无监督学习方法通过使用掩码语言模型目标为模型提供初始的参数,并在有标签数据集上与映射函数共同进行微调,学习蛋白质在任务空间上的表示,并应用于下游的预测任务。但是蛋白质作为一类大分子,在不同环境中会存在不同的构象,如可以通过氨基酸的化学性质进行自发的折叠,又可以通过与其它蛋白质的接触发生形态上的变化。而目前的蛋白质语言模型对于特定的蛋白质,只能给出唯一一种嵌入矩阵,很难同时处理好在结构预测和功能预测任务上的效果,限制了模型的表达能力和实际场景中的应用。
自然语言中提示符思想的成功表示说明语言模型具有很强的表达能力,可以通过改变序列局部符号对整体的输出产生人们期望的影响。传统的提示符思想在于使用已有的预训练模型,在语义空间内找到最优的提示符。但在蛋白质预训练模型中,语义单元是氨基酸,没有办法像语言提示符那样,可以由人工手动构建。其次,连续空间上的提示符缺乏可解释性。
专利文献CN 106845149 A公开了一种新的基于基因本体信息的蛋白质序列表示方法,首先使用BLAST程序搜索Swiss-Prot数据库找到蛋白质序列P所有的相似蛋白质序列,将训练数据集中所有蛋白质输入到GO数据库中,搜寻每个蛋白质所具有的GO本体信息;然后在基因本体库中搜寻P蛋白质所具有的标注基因本体信息;根据预测问题具有的M个标签,将P蛋白质定义为M个元素的离散向量,即通过将序列集中的蛋白质GO信息,融合成新的蛋白质P的向量描述,来降低蛋白质的序列表示维度。
专利文献CN104134017A公开了一种基于紧凑特征表示的蛋白质作用关系对抽取方法,包括以下步骤:1)选取所需的语料,语料是以句子为单位,已经有了蛋白质实体的标注及实体关系的标注;2)舍弃步骤1)中不包含蛋白质实体或只包含一个蛋白质实体的句子,得到句子集合;3)用占位符替换句子中相应的蛋白质实体并进行占位符融合,再进行词性标注和句法分析;4)以每个实体对为单位,获取词、词性、句法和模板的特征;5)对步骤4)中获得的特征进行紧凑化表达的操作;6)利用支持向量机对从步骤4)得到的特征进行训练或者利用已训练的模型进行预测。该方法目的是尽最大努力利用句子中的可用信息,提升特征向量的信息量。
上述两件专利申请虽然有各自的优点,但是仍然存在没有考虑蛋白质构象导致得到的表示向量不准确的问题,限制蛋白质在结构预测和功能预测任务上的效果。
发明内容
鉴于上述,本发明的目的是提供一种基于预训练语言模型的蛋白质构象感知表示学习方法,通过在多个数据集上联合预训练的方法,得到不同构象下的蛋白质表示,提升蛋白质在结构预测和功能预测任务的准确性。
为实现上述发明目的,本发明实施例提供了一种基于预训练语言模型的蛋白质构象感知表示学习方法,包括以下步骤:
获取由氨基酸序列组成的蛋白质,根据蛋白质构象构建不同数据集,为每类蛋白质构象定义提示符;
基于预训练语言模型构建表示学习模块,用于将每类提示符的嵌入表示融合到蛋白质的嵌入表示,以得到提示符标识下的蛋白质嵌入表示;
构建任务模块,用于针对每类蛋白质构象对应的任务,基于提示符标识下的蛋白质嵌入表示进行任务预测;
基于任务预测结果和标签构建每类任务的损失函数,结合所有类任务的损失函数和不同数据集,更新表示学习模块和任务模块的模型参数;
模型参数更新结束后,提取表示学习模块作为蛋白质表示模块。
在一个实施例中,所述表示学习模块包括提示符嵌入层、氨基酸嵌入层、融合层以及预训练语言模型,其中,提示符嵌入层用于学习每类提示符的嵌入表示,氨基酸嵌入层用于学习蛋白质的嵌入表示,融合层用于融合每类提示符的嵌入表示和蛋白质的嵌入表示得到融合表示,预训练语言模型用于对融合表示进行表示学习,得到每类提示符标识下的蛋白质嵌入表示;
所述任务模块包括每类蛋白质构象对应的任务映射层,每个任务映射层用于基于每类提示符标识下的蛋白质嵌入表示进行任务预测。
在一个实施例中,所述氨基酸嵌入层包括氨基酸信息嵌入层和氨基酸位置嵌入层,分别用于根据氨基酸序列提取氨基酸信息表示和氨基酸位置表示,氨基酸信息表示和氨基酸位置表示经叠加得到蛋白质的嵌入表示。
在一个实施例中,所述预训练语言模型为可插拔的掩码预训练语言模型,其中,掩码预训练语言模型包括BERT、RoBERTa、ALBERT、XLNet。
在一个实施例中,所述任务映射层包括双层MLP,双层MLP用于基于每类提示符标识下的蛋白质嵌入表示进行任务预测。
在一个实施例中,针对每类预测任务,以最小化任务预测结果和标签的误差为损失函数。
在一个实施例中,所述蛋白质表示模块应用于蛋白质结构和/或蛋白质功能的预测任务;应用时,在蛋白质表示模块中,将所有类提示符的嵌入表示同时融合到蛋白质的嵌入表示,以得到全部提示符标识下的蛋白质嵌入表示;该全部提示符标识下的蛋白质嵌入表示用于蛋白质结构和/或蛋白质功能的预测。
在一个实施例中,所述将所有类提示符的嵌入表示同时融合到蛋白质的嵌入表示,包括:
将所有类提示符的嵌入表示先进行拼接,得到的拼接表示再与蛋白质的嵌入表示进行融合,该融合包括拼接方式、全连接映射和卷积映射。
在一个实施例中,所述蛋白质构象包括自然折叠状态、作用状态;
针对自然折叠状态,以具有掩码的氨基酸序列为样本,以正常氨基酸序列作为样本标签,组成数据集,该自然折叠状态对应的提示符为序列提示符,对应的任务为掩码氨基酸的预测任务;
针对作用状态,以至少2条氨基酸序列为样本,以蛋白质反应类型为样本标签,组成数据集,该作用状态对应的提示符为作用提示符,对应的任务为蛋白质作用的预测任务。
与现有技术相比,本发明具有的有益效果至少包括:
基于蛋白质数据构建以蛋白质构象为中心的不同数据集,通过添加提示符来反应蛋白质构象,来学习提示符标识下的蛋白质嵌入表示,在此基础上,采用不同提示符标识下的蛋白质嵌入表示进行不同预测任务,以让预训练语言模型计算多个不同预测任务的损失之和,以学习到更具区分度的构象提示符表示,融合该构象提示符表示的蛋白质嵌入表示能够提升蛋白质在结构预测和功能预测任务的准确性。
附图说明
为了更清楚地说明本发明实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图做简单地介绍,显而易见地,下面描述中的附图仅仅是本发明的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动前提下,还可以根据这些附图获得其他附图。
图1是基于预训练语言模型的蛋白质构象感知表示学习方法的流程图;
图2是实施例提供的蛋白质表示模块的构建过程示意图;
图3是实施例提供的蛋白质表示模块的应用过程示意图。
具体实施方式
为使本发明的目的、技术方案及优点更加清楚明白,以下结合附图及实施例对本发明进行进一步的详细说明。应当理解,此处所描述的具体实施方式仅仅用以解释本发明,并不限定本发明的保护范围。
图1是基于预训练语言模型的蛋白质构象感知表示学习方法的流程图。如图1所示,实施例提供的基于预训练语言模型的蛋白质构象感知表示学习方法,包括以下步骤:
步骤1,获取由氨基酸序列组成的蛋白质,根据蛋白质构象构建不同数据集,为每类蛋白质构象定义提示符。
实施例中,基于生物领域对蛋白质的实验结果,获得蛋白质结构数据,每条蛋白质结构数据由氨基酸序列组成,在获得蛋白质结构数据的同时,还获得蛋白质的功能数据,即蛋白质之间的相互作用数据,该相互作用数据为蛋白质的反应类型。
蛋白质构象是指蛋白质表现出来的不同状态,包括自然折叠状态、作用状态。在获得蛋白质数据后,根据不同蛋白质构象构建表现不同性质的数据集,为以蛋白质构象为中心的蛋白质数据处理方式提供基础,以进行针对不同预测任务的蛋白质嵌入表示学习。
实施例中,针对自然折叠状态,以具有掩码[Mask]的氨基酸序列为样本,以正常氨基酸序列(不具有掩码的原始氨基酸序列)作为样本标签,组成氨基酸序列数据集。基于该氨基酸序列数据集,构建掩码氨基酸的预测任务,即利用氨基酸序列数据集进行掩码氨基酸的预测任务,以学习蛋白质在自然折叠状态下的嵌入表示。
实施例中,针对作用状态,以相互作用的至少2条氨基酸序列为样本,以蛋白质反应类型为样本标签,组成蛋白质相互作用数据集。基于该蛋白质相互作用数据集,构建蛋白质相互作用的预测任务,即利用蛋白质相互作用数据集进行蛋白质相互作用的预测任务,以学习学习蛋白质在作用状态下的嵌入表示。
实施例中,蛋白质相互作用数据集中相互作用蛋白质以网络图的方式表示,在得到网络图后根据物种将网络图进行切割,并根据预训练语言模型剔除无法作为输入的蛋白质,对于训练中的每一步,选择一个物种的蛋白质作用网络,随机采样出总长度不超过2048个氨基酸的蛋白质,并得到子图的相互作用矩阵,用于输入表示学习模型。
其中,蛋白质反应类型包括反应、表达、激活、翻译后修饰、结合、催化。在构建蛋白质相互作用数据集时,至少以一种蛋白质反应类型作为标签。
实施例中,为了标记蛋白质构象,为每类蛋白质构象定义提示符,为自然折叠状态定义提示符为xSeq,为作用状态定义提示符为xIC。提示符在学习过程中融合到蛋白质嵌入表示中,以标记蛋白质构象。
步骤2,基于预训练语言模型构建表示学习模块,用于将每类提示符的嵌入表示融合到蛋白质的嵌入表示,以得到提示符标识下的蛋白质嵌入表示。
实施例中,构建蛋白质嵌入表示的过程,将表示蛋白质构象的提示符添加到预训练语言模型的词表中,当蛋白质嵌入表示输入到预训练语言模型中,并带有蛋白质构象信息,预训练语言模型会将构象提示符的嵌入表示添加到蛋白质嵌入表示中。
在一个实施方式中,如图2所示,表示学习模块包括提示符嵌入层、氨基酸嵌入层、融合层以及预训练语言模型,其中,提示符嵌入层用于将每类提示符映射到高位空间,以学习每类提示符的嵌入表示;氨基酸嵌入层用于将蛋白质映射到高纬空间,以学习蛋白质的嵌入表示;融合层用于融合每类提示符的嵌入表示和蛋白质的嵌入表示得到融合表示,融合层可以采用拼接、全连接映射以及卷积映射的方式进行嵌入表示的融合;预训练语言模型用于对融合表示进行表示学习,得到每类提示符标识下的蛋白质嵌入表示。
在一个实施方式中,如图2所示,氨基酸嵌入层包括氨基酸信息嵌入层和氨基酸位置嵌入层,分别用于根据氨基酸序列提取氨基酸信息表示和氨基酸位置表示,氨基酸信息表示和氨基酸位置表示经叠加(图2中符号+)得到蛋白质的嵌入表示。其中,氨基酸信息表示表征了组成蛋白质的氨基酸信息,氨基酸位置表示表征了组成蛋白质的每个氨基酸的位置信息。
实施例中,预训练语言模型为可插拔的掩码预训练语言模型,其中,掩码预训练语言模型包括BERT、RoBERTa、ALBERT、XLNet。优选地,可以采用RoBERTa。这些预训练语言模型对输入的融合表示进行表示学习,以得到每类提示符标识下的蛋白质嵌入表示。
步骤3,构建任务模块,用于针对每类蛋白质构象对应的任务,基于提示符标识下的蛋白质嵌入表示进行任务预测。
实施例中,任务模块包括每类蛋白质构象对应的任务映射层,每个任务映射层采用不同映射函数将每类提示符标识下的蛋白质嵌入映射到不同的任务空间,即进行不同任务预测。优选地,任务映射层可以采用双层MLP作为映射函数,进行不同任务预测。
针对自然折叠状态对应的掩码氨基酸的预测任务,和针对作用状态对应的任务为蛋白质作用的预测任务,任务模块包含2个任务映射层,分别为作用构象任务映射头和序列任务映射头,每个任务映射层均采用双层MLP作为映射函数。
步骤4,基于任务预测结果和标签构建每类任务的损失函数,结合所有类任务的损失函数和不同数据集,更新表示学习模块和任务模块的模型参数。
实施例中,针对每类预测任务,以最小化任务预测结果和标签的误差为损失函数。然后结合所有类预测任务的损失函数,在每类预测任务对应的数据集上优化表示学习模块和任务模块的模型参数,即优化提示符嵌入层、氨基酸嵌入层、融合层、预训练语言模型以及任务映射层的参数。
实施例中,当蛋白质构象包括自然折叠状态、作用状态时,表示学习过程中,构建的总损失函数包括掩码氨基酸预测任务损失和蛋白质相互作用预测任务损失两部分,具体包括:
在掩码氨基酸预测任务中,一个训练批次由N条蛋白质组成,对于其中的每一条氨基酸链,随机选择其中15%的氨基酸进行预测。用于预测的每个氨基酸,其有80%概率变为[Mask]符号,10%概率变为其它氨基酸符号,10%概率保持不变,以这样的方式构建样本数据,得到蛋白质对(P,P′),其中,P′为被改变的蛋白质,P为原始蛋白质,作为蛋白质P′在掩码氨基酸预测任务中的标签。基于此,一条蛋白质中第i个氨基酸预测任务损失函数li(x,y)为:
在蛋白质相互作用预测任务中,一个训练批次由N条蛋白质组成,将蛋白质中所有氨基酸嵌入的均值作为蛋白质的嵌入表示,预测任意两个蛋白质之间是否存在相互作用,该任务的损失函数和以上掩码氨基酸预测任务类似,令X={x1,x2,…,xN}作为采样出来的蛋白质序列,y∈{0,1}n×n是相互作用矩阵,损失函数LPPI表示如下所示:
其中,i≠j,均为蛋白质索引,BCE表示二值交叉熵,p(yij|xi,xj)表示正确预测蛋白质xi和蛋白质xj之间相互作用为标签yij的概率。
则总损失为L=LS+λLPPI,其中λ是超参数。
实施例中,将掩码氨基酸预测任务对应的氨基酸序列数据集输入至表示学习模块,将蛋白质相互作用预测任务对应的蛋白质相互作用数据集输入至表示学习模块,并结合包括掩码氨基酸预测任务损失和蛋白质相互作用预测任务损失的总损失函数最小化为目标,来更新表示学习模块和任务模块的模型参数。
步骤5,模型参数更新结束后,提取表示学习模块作为蛋白质表示模块。
实施例中,在模型参数更新后,提取参数确定的表示学习模块包括的提示符嵌入层、氨基酸嵌入层、融合层以及预训练语言模型作为蛋白质表示模块,其中,参数确定的预训练语言模型作为蛋白质表示模型。
实施例中,得到的蛋白质表示模块应用于蛋白质结构和/或蛋白质功能的预测任务。应用时,在蛋白质表示模块中,将所有类提示符的嵌入表示同时融合到蛋白质的嵌入表示,以得到全部提示符标识下的蛋白质嵌入表示;该全部提示符标识下的蛋白质嵌入表示用于蛋白质结构和/或蛋白质功能的预测。
在一个实施方式中,将所有类提示符的嵌入表示同时融合到蛋白质的嵌入表示,包括:将所有类提示符的嵌入表示先进行拼接,得到的拼接表示再与蛋白质的嵌入表示进行融合,该融合包括拼接方式、全连接映射和卷积映射。
在一个实施方式中,如图3所示,蛋白质表示模块包括提示符嵌入层、氨基酸嵌入层、融合层、蛋白质表示模型。该蛋白质表示模块应用于蛋白质结构和/或蛋白质功能的预测任务时,在蛋白质表示模型后增加与预测任务相关的分类器(下游任务映射头),该蛋白质表示模型与分类器组成端到端的任务预测模型。
应用时,根据预测任务选择合适的提示符,并构建预测任务的小规模数据集,并将小规模数据集包括的蛋白质和提示符作为任务预测模型的输入,利用提示符嵌入层提取的每类提示符的嵌入表示经过拼接后形成拼接表示,并输入至融合层,融合层对蛋白质的嵌入表示和拼接表示进行融合后,输入至蛋白质表示模型经过表示学习,得到全部提示符标识下的蛋白质嵌入表示,基于该蛋白质嵌入表示以对任务预测模型包含的分类器进行参数微调,参数微调后的任务预测模型可以预测蛋白质功能、蛋白质结构。
具体地,当针对自然折叠状态、作用状态两类蛋白质构象时,这两类蛋白质构象对应的提示符为xSeq和xIC分别输入至提示符嵌入层,以提取对应的嵌入表示,两个嵌入表示经过拼接融合后输入至融合层,在融合层中与蛋白质的嵌入表示融合后输入至蛋白质表示模型,得到在两个提示符标识下的蛋白质嵌入表示。
上述实施例提供的基于预训练语言模型的蛋白质构象感知表示学习方法,首次基于公开蛋白质数据构建以蛋白质构象为中心的预训练数据集。
上述实施例提供的基于预训练语言模型的蛋白质构象感知表示学习方法,不同于现有的预训练语言模型,其生成的蛋白质表示是唯一,不能反应蛋白质在不同环境中的相关信息。提出的蛋白质表示方法不仅能通过添加构象符号反映蛋白质相互作用和蛋白质自发折叠的构象信息,还支持增量添加特定条件下新的构象信息。
上述实施例提供的基于预训练语言模型的蛋白质构象感知表示学习方法中,采用的构象嵌入和融合方法,为蛋白质中所有的氨基酸提供了无差异的构象信息,以便更好地得到氨基酸嵌入;
上述实施例提供的基于预训练语言模型的蛋白质构象感知表示学习方法中,不同于现有的单任务预训练模型,提出在不同构象符号下使用不同预训练任务的多任务预训练模型,让模型计算多个损失之和,以学习到更具区分度的构象符号表示。
以上所述的具体实施方式对本发明的技术方案和有益效果进行了详细说明,应理解的是以上所述仅为本发明的最优选实施例,并不用于限制本发明,凡在本发明的原则范围内所做的任何修改、补充和等同替换等,均应包含在本发明的保护范围之内。
Claims (9)
1.一种基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,包括以下步骤:
获取由氨基酸序列组成的蛋白质,根据蛋白质构象构建不同数据集,为每类蛋白质构象定义提示符;
基于预训练语言模型构建表示学习模块,用于将每类提示符的嵌入表示融合到蛋白质的嵌入表示,以得到提示符标识下的蛋白质嵌入表示;
构建任务模块,用于针对每类蛋白质构象对应的任务,基于提示符标识下的蛋白质嵌入表示进行任务预测;
基于任务预测结果和标签构建每类任务的损失函数,结合所有类任务的损失函数和不同数据集,更新表示学习模块和任务模块的模型参数;
模型参数更新结束后,提取表示学习模块作为蛋白质表示模块。
2.根据权利要求1所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,所述表示学习模块包括提示符嵌入层、氨基酸嵌入层、融合层以及预训练语言模型,其中,提示符嵌入层用于学习每类提示符的嵌入表示,氨基酸嵌入层用于学习蛋白质的嵌入表示,融合层用于融合每类提示符的嵌入表示和蛋白质的嵌入表示得到融合表示,预训练语言模型用于对融合表示进行表示学习,得到每类提示符标识下的蛋白质嵌入表示;
所述任务模块包括每类蛋白质构象对应的任务映射层,每个任务映射层用于基于每类提示符标识下的蛋白质嵌入表示进行任务预测。
3.根据权利要求2所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,所述氨基酸嵌入层包括氨基酸信息嵌入层和氨基酸位置嵌入层,分别用于根据氨基酸序列提取氨基酸信息表示和氨基酸位置表示,氨基酸信息表示和氨基酸位置表示经叠加得到蛋白质的嵌入表示。
4.根据权利要求2所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,所述预训练语言模型为可插拔的掩码预训练语言模型,其中,掩码预训练语言模型包括BERT、RoBERTa、ALBERT、XLNet。
5.根据权利要求2所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,所述任务映射层包括双层MLP,双层MLP用于基于每类提示符标识下的蛋白质嵌入表示进行任务预测。
6.根据权利要求1所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,针对每类预测任务,以最小化任务预测结果和标签的误差为损失函数。
7.根据权利要求1-6任一项所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,所述蛋白质表示模块应用于蛋白质结构和/或蛋白质功能的预测任务;
应用时,在蛋白质表示模块中,将所有类提示符的嵌入表示同时融合到蛋白质的嵌入表示,以得到全部提示符标识下的蛋白质嵌入表示;该全部提示符标识下的蛋白质嵌入表示用于蛋白质结构和/或蛋白质功能的预测。
8.根据权利要求7所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,所述将所有类提示符的嵌入表示同时融合到蛋白质的嵌入表示,包括:
将所有类提示符的嵌入表示先进行拼接,得到的拼接表示再与蛋白质的嵌入表示进行融合,该融合包括拼接方式、全连接映射和卷积映射。
9.根据权利要求1所述的基于预训练语言模型的蛋白质构象感知表示学习方法,其特征在于,所述蛋白质构象包括自然折叠状态、作用状态;
针对自然折叠状态,以具有掩码的氨基酸序列为样本,以正常氨基酸序列作为样本标签,组成数据集,该自然折叠状态对应的提示符为序列提示符,对应的任务为掩码氨基酸的预测任务;
针对作用状态,以至少2条氨基酸序列为样本,以蛋白质反应类型为样本标签,组成数据集,该作用状态对应的提示符为作用提示符,对应的任务为蛋白质作用的预测任务。
Priority Applications (3)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210122014.5A CN114678061A (zh) | 2022-02-09 | 2022-02-09 | 基于预训练语言模型的蛋白质构象感知表示学习方法 |
PCT/CN2022/126696 WO2023151314A1 (zh) | 2022-02-09 | 2022-10-21 | 基于预训练语言模型的蛋白质构象感知表示学习方法 |
US18/278,298 US20240233875A9 (en) | 2022-02-09 | 2022-10-21 | Perceptual representation learning method for protein conformations based on pre-trained language model |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202210122014.5A CN114678061A (zh) | 2022-02-09 | 2022-02-09 | 基于预训练语言模型的蛋白质构象感知表示学习方法 |
Publications (1)
Publication Number | Publication Date |
---|---|
CN114678061A true CN114678061A (zh) | 2022-06-28 |
Family
ID=82071767
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202210122014.5A Pending CN114678061A (zh) | 2022-02-09 | 2022-02-09 | 基于预训练语言模型的蛋白质构象感知表示学习方法 |
Country Status (3)
Country | Link |
---|---|
US (1) | US20240233875A9 (zh) |
CN (1) | CN114678061A (zh) |
WO (1) | WO2023151314A1 (zh) |
Cited By (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN115565607A (zh) * | 2022-10-20 | 2023-01-03 | 抖音视界有限公司 | 确定蛋白质信息的方法、装置、可读介质及电子设备 |
WO2023151314A1 (zh) * | 2022-02-09 | 2023-08-17 | 浙江大学杭州国际科创中心 | 基于预训练语言模型的蛋白质构象感知表示学习方法 |
CN116913393A (zh) * | 2023-09-12 | 2023-10-20 | 浙江大学杭州国际科创中心 | 一种基于强化学习的蛋白质进化方法及装置 |
CN116935952A (zh) * | 2023-09-18 | 2023-10-24 | 浙江大学杭州国际科创中心 | 基于图神经网络训练蛋白质预测模型的方法及装置 |
Families Citing this family (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN116992979B (zh) * | 2023-09-28 | 2024-01-23 | 北京智精灵科技有限公司 | 基于深度学习的认知训练任务推送方法及*** |
CN117711532B (zh) * | 2024-02-05 | 2024-05-10 | 北京悦康科创医药科技股份有限公司 | 多肽氨基酸序列生成模型训练方法以及多肽氨基酸序列生成方法 |
Family Cites Families (4)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
US20220375539A1 (en) * | 2019-08-23 | 2022-11-24 | Geaenzymes Co. | Systems and methods for predicting proteins |
CN113593631B (zh) * | 2021-08-09 | 2022-11-29 | 山东大学 | 一种预测蛋白质-多肽结合位点的方法及*** |
CN113887627A (zh) * | 2021-09-30 | 2022-01-04 | 北京百度网讯科技有限公司 | 噪音样本的识别方法、装置、电子设备以及存储介质 |
CN114678061A (zh) * | 2022-02-09 | 2022-06-28 | 浙江大学杭州国际科创中心 | 基于预训练语言模型的蛋白质构象感知表示学习方法 |
-
2022
- 2022-02-09 CN CN202210122014.5A patent/CN114678061A/zh active Pending
- 2022-10-21 US US18/278,298 patent/US20240233875A9/en active Pending
- 2022-10-21 WO PCT/CN2022/126696 patent/WO2023151314A1/zh active Application Filing
Cited By (7)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
WO2023151314A1 (zh) * | 2022-02-09 | 2023-08-17 | 浙江大学杭州国际科创中心 | 基于预训练语言模型的蛋白质构象感知表示学习方法 |
CN115565607A (zh) * | 2022-10-20 | 2023-01-03 | 抖音视界有限公司 | 确定蛋白质信息的方法、装置、可读介质及电子设备 |
CN115565607B (zh) * | 2022-10-20 | 2024-02-23 | 抖音视界有限公司 | 确定蛋白质信息的方法、装置、可读介质及电子设备 |
CN116913393A (zh) * | 2023-09-12 | 2023-10-20 | 浙江大学杭州国际科创中心 | 一种基于强化学习的蛋白质进化方法及装置 |
CN116913393B (zh) * | 2023-09-12 | 2023-12-01 | 浙江大学杭州国际科创中心 | 一种基于强化学习的蛋白质进化方法及装置 |
CN116935952A (zh) * | 2023-09-18 | 2023-10-24 | 浙江大学杭州国际科创中心 | 基于图神经网络训练蛋白质预测模型的方法及装置 |
CN116935952B (zh) * | 2023-09-18 | 2023-12-01 | 浙江大学杭州国际科创中心 | 基于图神经网络训练蛋白质预测模型的方法及装置 |
Also Published As
Publication number | Publication date |
---|---|
US20240233875A9 (en) | 2024-07-11 |
WO2023151314A1 (zh) | 2023-08-17 |
US20240136021A1 (en) | 2024-04-25 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN114678061A (zh) | 基于预训练语言模型的蛋白质构象感知表示学习方法 | |
CN111090987B (zh) | 用于输出信息的方法和装置 | |
CN113011186B (zh) | 命名实体识别方法、装置、设备及计算机可读存储介质 | |
CN111814487B (zh) | 一种语义理解方法、装置、设备及存储介质 | |
CN111401058B (zh) | 一种基于命名实体识别工具的属性值抽取方法及装置 | |
CN116450796A (zh) | 一种智能问答模型构建方法及设备 | |
CN112667782A (zh) | 一种文本分类方法、装置、设备及存储介质 | |
CN111831902A (zh) | 推荐理由筛选方法、装置、电子设备 | |
CN114691525A (zh) | 测试用例的选择方法及装置 | |
CN115186675A (zh) | 语言模型训练及自然语言任务处理方法、装置及相关设备 | |
CN114742016B (zh) | 一种基于多粒度实体异构图的篇章级事件抽取方法及装置 | |
CN115390806A (zh) | 基于双模态联合建模的软件设计模式推荐方法 | |
CN116127013A (zh) | 一种个人敏感信息知识图谱查询方法和装置 | |
CN110377910B (zh) | 一种表格描述的处理方法、装置、设备及存储介质 | |
CN117648429A (zh) | 基于多模态自适应检索式增强大模型的问答方法及*** | |
CN117453851B (zh) | 基于知识图谱的文本索引增强问答方法及*** | |
CN117828024A (zh) | 一种插件检索方法、装置、存储介质及设备 | |
CN114626378A (zh) | 命名实体识别方法、装置、电子设备及计算机可读存储介质 | |
CN115238708B (zh) | 文本语义识别方法、装置、设备、存储介质及程序产品 | |
CN116384379A (zh) | 一种基于深度学习的中文临床术语标准化方法 | |
CN111626059B (zh) | 一种信息处理方法及装置 | |
CN114822726A (zh) | 构建方法、分析方法、装置、存储介质和计算机设备 | |
CN115203532A (zh) | 一种项目推荐方法、装置、电子设备及存储介质 | |
CN114358021A (zh) | 基于深度学习的任务型对话语句回复生成方法及存储介质 | |
CN112966501A (zh) | 一种新词发现方法、***、终端及介质 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination |