新闻中心

基于双任务的端到端无模板反应预测模型

2024-01-12
浏览次数:
返回列表

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

重新排版 | X

本文旨在介绍西南交通大学杨燕/江永全团队在《应用智能》杂志上发表的研究成果,其中的第一作者是胡昊哲,一位硕士生

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

论文链接:https://doi.org/10.1007/s10489-023-05048-8
相关代码:https://github.com/AILBC/BiG2S

作者以目前无模板逆合成领域兴起的图至序列模型框架为基础,进一步在同参数量规模下尝试构建一类在单个模型中同时解决逆合成预测与正向反应预测任务的模型 BiG2S(双向图至序列)

同时,作者对主流逆合成数据集 USPTO-50k 进行了初步分析,探讨了模型在训练过程中对不同 SMILES 片段的预测难度差异以及模型在验证集上 Top-k 匹配率的波动情况,并针对这些问题引入了不平衡损失函数以及改进了模型集成和束搜索策略

在对三个主要的反应预测数据集进行测试时,通过对逆合成和正向反应预测任务进行测试,以及对上述模块进行全面的消融实验证明,BiG2S能够在适当的参数规模下以单一模型处理逆合成和正向反应预测任务。与已有的基于预训练和数据增强的无模板方法相比,BiG2S的整体预测能力同样出色

研究的背景

逆合成与正向合成是有机化学、计算机辅助合成规划(CASP)以及计算机辅助药物设计(CADD)领域的基础性挑战

进行内容改写时,需要将原文改写为中文,同时保持原始意思的不变

早期的逆合成规划系统直接依赖于领域专家预先编码的反应规则,或者是基于物理化学的计算,而随着深度学习的快速发展。目前领域内的主流方法则是构建一个任务特异的神经网络框架以从数据驱动的角度完成反应预测任务。其中,不依赖于特定先验化学知识的无模板法通过其类似于端到端机器翻译的简洁思路以及灵活性逐渐成为了领域内的主流发展方向之一。

当前,大多数无模板逆合成模型的输入和输出都是分子的 SMILES 字符串,即采用了序列至序列(Seq2Seq)的流程。这种方法能够很好地利用在自然语言处理领域内已有的模型框架,以及针对于 SMILES 表示方法的成熟的数据处理流程

然而,由于SMILES作为一维字符串序列无法很好地表征和利用分子图所包含的二维/三维结构信息,因此在这个领域中逐渐出现了使用分子图代替SMILES作为模型输入的图至序列(Graph2Seq)方法,或者将分子图的附加结构信息嵌入到SMILES序列中的序列至序列方法。这两种方法都能很好地利用来自分子图的丰富结构特征

基于此,本文以新兴的图至序列方法为基础,在原先基于SMILES的模型对逆合成与正向反应预测任务同时训练的相关探索的基准上,进一步全面地探究对这类双任务模型的构建与实验,同时也初步地探索与分析了模型在训练过程中所展现的难度不平衡以及Top-k匹配率波动的问题;在此基础上构建的BiG2S模型能够较好地处理主流数据集中的逆合成与正向反应预测任务,并在不使用数据增强的情况下取得与其他无模板逆合成模型一致的反应预测能力

总体框架需要进行重写

BiG2S整体结构是一个端到端的编码器-解码器,如图1所示。编码器端采用局部定向消息传递图网络和融入图结构偏置信息的全局图Transformer来生成最终的分子图节点表征。解码器则使用标准的Transformer解码器以自回归的方式生成目标分子的SMILES序列

需要注意的是,为了同时学习逆合成和正向反应预测,解码器端的输入额外包含了不添加位置信息的双任务标签。同时,解码器端的归一化层和最终的线性层都有两套参数,分别用于学习逆合成任务和正向反应预测任务

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

图 1:BiG2S 整体框架图

需要双任务训练框架

逆合成与正向反应预测是两个相关的任务,其中逆合成任务以产物作为输入和反应物作为目标输出,而正向反应预测任务则相反。这两个任务之间存在着紧密的联系,因为它们可以通过互换逆合成任务的输入和目标输出来转化为正向反应预测任务

因此,一些基于SMILES的无模板模型已经尝试通过将反向合成与正向反应预测作为训练目标,来提高对化学反应的理解,并取得了一定的效果。基于这个思路,作者进一步尝试将双任务训练引入到图到序列模型中

具体而言,作者基于先前在其他方法上使用的参数共享策略,在解码器的归一化层和最终的线性层内构建了两套任务特定的参数。而在其他模块中,两类任务共享一套参数。同时,在输入的分子图节点和解码器的初始输入序列中额外添加了双任务标签。这样,即使在控制整体模型规模的情况下,模型也能够区分两类任务并学习它们不同的数据分布

需要训练和推理优化

芝麻乐开源众筹cms系统 芝麻乐开源众筹cms系统

芝麻乐开源众筹系统采用php+mysql开发,基于MVC开发,适用于各类互联网金融公司使用,程序具备模板分离技术,您可以根据您的需要进行应用扩展来达到更加强大功能。前端使用pintuer、jquery、layer等....系统易于使用和扩展简单的安装和升级向导多重业务逻辑判断,预防出现bug后台图表数据方式,一目了然后台包含但不限于以下功能:用户认证角色管理节点管理管理员管理上传配置支付配置短信平

芝麻乐开源众筹cms系统 1 查看详情 芝麻乐开源众筹cms系统

在训练过程中,作者进一步记录并分析了模型在训练过程中所反映出的两类问题

首先,作者记录了在USPTO-50k中不同SMILES字符的出现频次以及其在训练时对应的预测准确率,如图2所示。在训练过程中,对于在训练集中占比分别为0.4%和0.3%的S和Br,它们之间整体预测准确率的绝对差异达到了8%。这初步表明了不同的分子结构/片段间预测的难度存在明显的差异,由此,作者通过引入不平衡损失函数(如Focal Loss)来缓解此类问题,从而使模型能够更加关注训练时准确率更低的分子片段

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

图 2:在USPTO-50k训练集中,不同SMILES字符的出现频次以及其在训练时的整体预测准确率

此外,作者还记录了模型在训练期间对验证集的预测结果质量变化,如图3所示。作者发现,在USPTO-50k数据集的中后期训练阶段,模型在验证集上的Top-1准确率仍然在不断提高,但在Top-3、Top-5和Top-10的预测质量方面出现了明显下降

为了在提升模型 Top-1 预测质量的同时保持模型前十位反应物生成结果的整体质量,我们额外构建了一类基于自定义评价指标的模型集成策略。具体来说,我们构建了一类存储模型的队列,同时根据预定义的评价指标(如 Top-1 准确率,加权的 Top-k 准确率等)对存入的模型进行排序。在整个训练过程中,我们动态地存入待选模型并自动生成基于队列中前 3-5 位的集成模型,从而保留 Top-k 预测质量最高的模型。在推理阶段,我们也基于新的框架重新构建了更加注重于搜索广度的束搜索策略,以提升模型 Top-k 生成结果的整体质量

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

图3:在 USPTO-50k 数据集中训练期间模型对验证集的 Top-k 匹配率以及 Top-k 无效生成分子比例的变化曲线

需要进行双任务实验中的基准数据集

作者在逆合成任务与正向反应预测任务中进行了实验,使用了包含 5 万、50 万以及 100 万条化学反应数据的数据集 USPTO-50k、USPTO-MIT、USPTO-full。实验中比较了双任务模型和单任务模型的表现。根据图4的测试结果显示

在小规模数据集中,BiG2S基于双任务训练在逆合成任务中取得了领先的预测精度,同时也保持了较高的正向反应预测精度;然而在偏向于正向反应预测的USPTO-MIT数据集以及大规模数据集USPTO-full中,由于模型整体参数量的限制,双任务训练后的模型表现出现了降低。尽管如此,从双任务模型以几乎一致的参数量与小幅度的反应预测能力降低( Top-k 准确率的绝对差值位于 0.5% 左右)获得了同时处理逆合成任务与正向反应预测任务的能力这一角度来看,BiG2S 模型已经达到了预期目标

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

图 4:BiG2S 的双任务模型与单任务模型在三个基准数据集上的实验结果,其中上标 b 表示采用单任务模型分别完成两类任务

重新分析消融实验

作者通过消融实验进一步验证了新的束搜索算法以及采用不平衡损失后 BiG2S 在不同数据集中进行预测时的最佳温度超参数。这里的温度超参数是指 Softmax 中用于控制输出概率分布的温度参数 T。实验结果如图 5 和图 6 所示

在针对束搜索算法的实验中,可以观察到OpenNMT在搜索宽度扩大至3倍的同时搜索耗时仅扩大至1.74倍,而新的束搜索算法在Top-1精度与OpenNMT一致的情况下整体的搜索耗时扩大了1-2倍;但在Top-10预测结果的质量上,新的束搜索算法与OpenNMT相比具有至少3%的绝对精度优势以及2%的有效分子比例优势,可以说新的束搜索算法以搜索耗时为代价带来了明显提升了模型整体Top-k搜索结果的质量

在对温度超参数进行实验时,研究人员发现,在小规模数据集上使用较大的温度参数可以显著提高整体的 Top-k 预测精度。而在更大规模的数据集中,由于 BiG2S 模型规模不能完全适应所有反应数据,此时选择较小的温度参数往往有助于模型搜索

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

图 5:BiG2S 采用的束搜索算法与之前无模板逆合成模型常用的基于 OpenNMT 的束搜索算法在搜索结果质量与搜索耗时上的比较

西南交大杨燕/江永全团队:基于双任务的端到端图至序列无模板反应预测模型

图 6:使用不平衡损失后的 BiG2S 在不同数据集上采用不同温度超参数(T)进行搜索的 Top-k 预测准确率

研究的结论显示...

在本文中,作者提出了一种名为BiG2S的无模板反应预测模型,该模型可以同时处理逆合成任务和正向反应预测任务。通过采用适当的参数共享策略和额外的双任务标签,BiG2S能够以较小的参数量在不同规模的数据集上完成逆合成任务和反应预测任务,且其整体预测能力与主流模型相当

为了解决模型训练中不同 SMILES 字符预测难度不均衡和 Top-k 预测精度波动的问题,作者引入了不平衡损失、基于自定义评价指标的模型自动集成策略和基于新框架的束搜索算法来缓解这些问题

BiG2S 在三个不同规模的主流数据集上都表现出了很好的双任务预测能力,而进一步的消融实验也证明了额外引入的训练与推理策略的有效性

以上就是基于双任务的端到端无模板反应预测模型的详细内容,更多请关注其它相关文章!


# 理论  # 开源  # 端到  # 深度学习  # 喀什seo排名  # 中国联通推广营销面试  # 正规网站优化哪家好  # seo和sem是啥  # 重庆主题餐厅营销推广  # 赞皇技术网站推广哪个好  # 西安网站建设行者seo  # 谷歌seo推广公司南县  # 抚州运营seo什么价格  # 天津通用网站建设怎么样  # 南极  # 两类  # 所示  # 如图  # 过程中  # 很好  # 不平衡  # 交大 


相关栏目: 【 行业资讯67740 】 【 技术百科0 】 【 网络运营39195


相关推荐: 折叠屏手机信号哪个最强  win10锁屏壁纸怎么换360锁屏壁纸吗  5g手机4g卡怎么没有网络  市盈率是什么意思高好还是低好  如何查看硬盘是固态硬盘  j*a数组怎么放字符  ts什么意思  华为的type-c接口是什么接口  光刻机的作用及工作原理  j*a怎么用json数组  360n5锁屏壁纸怎么设置  路由器上面的power红灯是什么意思  选哪个折叠屏手机好用  如何用ftp连接命令行  如何用命令下载服务器网站  如何提高固态硬盘性能  春运抢票最多能抢几趟车  远程桌面如何发送命令  使用typescript对团队有什么要求  ospf中交换机命令如何设置  单片机学习视频怎么调色  征信不好如何短期恢复  哪些库是typescript  市盈率市净率是什么意思  ready是什么意思  苹果16送哪些配件  单片机怎么读取电流值  单片机怎么储存和显示  导航power在汽车上是什么意思  怎么在爱奇艺中投屏到电视最新方法  固态硬盘损坏如何修复  typescript的文件如何执行  台达plc只有power灯亮是什么意思  如何学习typescript  arp命令如何使用  performance是什么意思  vi命令如何退出  360f4怎么取消百变壁纸  征信信用不好如何恢复 征信信用不好如何恢复指南  电脑显示屏上power是什么意思  春运抢票可以抢几次票  为什么要用typescript6  j*a怎么用数组缓存  a股等权市盈率中位数是什么意思  单片机引脚怎么改成上拉  哪个品牌有折叠屏手机卖  怎么打印数组j*a  电动车power灯亮红灯是什么意思  哪里要用typescript  输入命令如何换行 

搜索