比AlphaFold2快10倍!国产开源项目加速蛋白质预测,超长序列22分钟就搞定(2)
扫一扫
分享文章到微信
扫一扫
关注99科技网微信公众号
此外,推理阶段昂贵的经济和时间成本使得AlphaFold模型相关的研究和开发十分困难,尤其是面对长序列推理时,计算复杂度和内存消耗都面临着巨大挑战。 xTrimo Multimer开源版模型针对AlphaFold-Multimer模型中的计算特性,进行了针对性的CUDA优化,并且进行了Kernel Fusion。 相较于AlphaFold2和哥伦比亚大学的OpenFold,xTrimo Multimer开源版模型在单卡推理上性能有明显提升 ,推理速度分别提高 1.58~2.14倍 和 1.14~2.23倍 。
xTrimo Multimer开源版模型还支持超长序列的分布式推理。 由于使用了动态轴并行(Dynamic Axial Parallelism)技术,xTrimo Multimer可以高效地将计算和部分显存分配到不同的设备上,从而解决超长序列面临的计算和内存挑战。 在2K到3K的序列长度下,xTrimo Multimer使用多卡进行推理,相对于OpenFold和AlphaFold 2推理速度最高提升 8.47倍 和 11.15倍 ,对比Uni-Fold2.0速度最高提升 4.45倍 。 xTrimo Multimer可以支持长达4K的序列推理,此时OpenFold和AlphaFold 2受限于显存无法完成推理,而xTrimo Multimer可以在20分钟左右完成。 百图生科首席AI科学家宋乐 表示:
从对单个蛋白质结构的模拟,到对不同蛋白质之间的相互作用的识别,再到对蛋白质复合物的绘制,百图生科致力于解码、建模复杂人体免疫系统,开发突破创新药物,编程免疫系统,治愈多种免疫相关疾病。
此次xTrimo Multimer开源版模型的发布,是我们携手潞晨科技Colossal-AI 团队,借助其在高性能计算上的优势以及百图生科生物计算领域前沿的技术积淀,朝着百图生科xTrimo多模态生物计算大模型体系迈出的又一步。
正如我们一直以来所努力的方向,精准高效地探寻癌症、衰老等疾病的免疫规律,逐一攻破AI技术在靶点挖掘、创新药物设计落地应用上的难关,让更多疾病可预警、可控制、可治愈! 潞晨科技创始人尤洋教授 表示:
此次 Colossal-AI 团队与百图生科合作的蛋白质单体与复合物结构预测最新方案,是面向大模型时代的通用深度学习系统 Colossal-AI 在生物医药领域应用落地的重要进展。
未来我们双方将会继续在生物计算大模型上有更深入的合作,助力深度学习在创新药物研发上的应用和落地。 随着人工智能对各个领域的高度渗透,各行业的智能化产业转型升级都离不开高性能AI的加持,而潞晨的AI大模型的高效训练和推理方案,刚好可为AI市场提供解决方案。
团队介绍 关于百图生科百图生科(BioMap)是生物计算引擎驱动的突破创新药物研发平台,由百度创始人李彦宏发起创立,致力于将先进AI等信息技术(IT)与前沿生物技术(BT)相结合,构建独特的靶点挖掘及药物设计能力,聚焦于解析免疫调控机理,开发创新的治疗性药物,造福人类健康。 目前公司拥有近万平米的实验室,50+个靶点及药物研发资产组合。
99科技网:http://www.99it.com.cn