超长序列，超快预测！深势科技联手阿里云，AI 蛋白质预测再下一城-低调大师

超长序列，超快预测！深势科技联手阿里云，AI 蛋白质预测再下一城

2022-09-06 374

摘要：强强联合，突破 AI 蛋白质预测模型推理性能瓶颈，支持最高 6.6k 长氨基酸序列蛋白质的预测计算，达到目前已知最优推理效果。

近日，深势科技与阿里云机器学习 PAI 团队联手，通过全新的蛋白质结构预测推理加速方案 FoldAcc，结合深势 Uni-Fold 最新模型代码和参数，将单次预测能支持的最大氨基酸序列长度提升至 6.6k，覆盖 99.992%已知的蛋白序列，同时推理速度显著提升，达到目前已知的最佳推理优化效果，将为 AI 预测蛋白质结构落地应用提供重要助力。

以新冠病毒研究重点之一——具有三聚体结构的刺突蛋白为例，其氨基酸序列典型总长度接近 4k，原版 AlphaFold 会因为超出显存限制（OOM）而无法进行预测计算，使用 Uni-Fold + FoldAcc 则能在 10 分钟左右完成 AI 推理计算。

自 2020 年 DeepMind 推出基于深度学习模型的 AlphaFold2 以来，AI 技术辅助蛋白质结构精确预测的相关研究备受关注，产学研界不断涌现创新成果。然而，在推动 AI 蛋白质结构预测规模化、产业化落地进程中，基础设施及工具完善性、AI 模型开发与部署效率等问题，仍然是横亘在研究者面前的难题。

2022 年 8 月，深势科技升级并开源 Uni-Fold 项目，成功复现了 AlphaFold2、AlphaFold-Multimer 模型的全尺寸从头训练，并通过多项效率优化，并将 AlphaFold 训练速度提升 220%，超越 OpenFold、FastFold 等方案，惠及更多研究者。

其中，针对困扰业界已久的 Evoformer 神经网络结构推理性能瓶颈问题，深势科技与阿里云 PAI 团队，基于在 AI 模型系统优化领域的长期积累，融合多卡并行、混合精度、编译优化等多项推理优化技术，使 Uni-Fold 训练的模型能进行多卡推理加速，并支持计算更长的氨基酸序列。

典型加速效果的测试结果如下（基于 A100-80G GPU，并启用 bf16）

Uni-Fold 升级开源，支持复合物训练作为生命科学领域重要的基础问题，蛋白质结构研究关系到癌症预警、靶向药物研究、衰老等临床医学和生命科学课题。传统的蛋白质结构研究手段，如：X 射线晶体学、冷冻电镜等，需要消耗大量时间和资源。如何快速高效并且能够规模化地预测蛋白质结构，一直是研究者们探寻求解的重要问题。

2021 年 12 月，深势科技推出 Uni-Fold v1.0.0，国际首次复现了 AlphaFold2 官方代码的全尺寸从头训练，并开源了训练与推理代码；2022 年 8 月，Uni-Fold 升级并开源了最新代码与模型参数，完整支持蛋白质单体、复合物结构预测模型的推理与训练。

此次开源的 Uni-Fold 基于 PyTorch 复现并改进的 AlphaFold (-Multimer) 模型，支持从头训练和推理部署，并且在 Protein Data Bank（PDB）最新公布的、模板相似度小于 40%的单体与复合物测试集上，都取得了与同类开源项目一致或更优的准确率。

此外 Uni-Fold 也进行了多项效率优化，将训练时间由 11 天缩短至约 4 天，显著优于其他同类开源项目。

机器学习平台 PAI 提供全链路 AI 工程支撑阿里云机器学习平台 PAI 为 Uni-Fold 项目提供了完善的 AI 工程能力。PAI 是国内唯一连续入选 Gartner 数据科学与机器学习平台报告的机器学习/深度学习平台，面向 AI 开发及应用全链路提供全面的工程化服务，并具备丰富的场景化落地实践。

针对模型推理优化场景，PAI 自研的通用推理加速器 PAI-Blade 能在不同业务场景下，通过模型系统联合优化，使模型达到最优推理性能，兼容主流机器学习框架，适配 GPU、CPU、端侧设备等多类型加速设备。

其中，PAI-Blade 核心组件 BladeDISC 具备业界领先的动态尺寸模型优化、大颗粒度算子融合等编译优化技术，在阿里巴巴集团内外客户的实际生产场景广泛应用，助力实现高性价比的 AI 应用部署。2022 年 2 月，BladeDISC 项目正式开源。

未来，为前沿科研提供更好用的 AI 基础设施以 AI 蛋白质结构预测为典型代表，AI for Science 的科学研究新范式正取得越来越多突破，人工智能与生命科学、物理、化学等领域的技术交织，将为科学研究和产业进步带来极大推动，也对 AI 基础技术与平台应用提出了新挑战。

深势科技是 AI for Science 科学研究范式的先行者，阿里云机器学习平台 PAI 是国内应用最广泛的机器学习平台之一，我们将持续为生物医药、能源、材料等领域的研究与产业应用提供更好用的 AI 基础设施，共同推动 AI for Science 领域的技术生态建设。

附录深势科技开源项目 Uni-Fold：https://github.com/dptech-corp/Uni-Fold

阿里云开源 AI 编译器 BladeDISC：https://github.com/alibaba/BladeDISC

微信关注我们

原文链接：https://my.oschina.net/u/5583868/blog/5573664

转载内容版权归作者及来源网站所有！

低调大师中文资讯倾力打造互联网数据资讯、行业资源、电子商务、移动互联网、网络营销平台。持续更新报道IT业界、互联网、市场资讯、驱动更新,是最及时权威的产业资讯及硬件资讯报道平台。

StratoVirt 中的 PCI 设备热插拔实现

热插拔即带电插拔，在虚拟化场景下，热插拔就是在虚拟机运行过程中对磁盘网卡等设备进行动态调整。常见的热插拔机制有 ACPI 机制的热插拔，PCIe-Native 机制的热插拔。ACPI 机制的热插拔依赖 ACPI 表，在 ACPI 表中会存放设备热插拔相关的信息。PCIe-Native 机制的热插拔是 PCI 规范中定义的，设备一般是热插到 Root Port 设备上，Root Port 设备可以认为是一个虚拟的桥设备，对应一个插槽。Root Port 设备本身不支持热插拔，因此需要在启动虚拟机前提前配置。目前，StratoVirt 标准机型中实现了基于 PCIe-Native 机制的热插拔。支持热插拔的设备包括磁盘、网卡、PCI 直通设备。热插拔的整体流程如下：对于热插主要分为两步：用户通过 QMP 下发 device_add 命令，StratoVirt 收到命令后会进行设备的实例化，然后插入到对应的 Root Port 设备上。 Root Port 设备更新相关的寄存器配置，然后发送中断通知虚拟机内驱动处理。对于热拔也可以分为两步：用户通过 QMP 下发 device_...

2022-09-05

448

文｜孙景文、吴迪（Dragonfly Contributor）本文 4138 字阅读10 分钟背景网络下载提起网络下载领域，你应该首先会想到基于 TCP/IP 协议簇的 C/S 模式。这种模式希望每一个客户机都与服务器建立 TCP 连接，服务器轮询监听 TCP 连接并依次响应，如下图：上世纪末期，基于 C/S 模式的思想，人们发展了 HTTP， FTP 等应用层协议。然而 C/S 模式的弊端很明显：服务器的负载过大，下载速率过慢。随着互联网规模的增大以及客户对于下载数据大小，下载速率等需求的上升，这些弊端被不断放大。 P2P 下载原理基于上述背景，有人结合 P2P 网络与负载均衡的思想，提出 P2P 下载模式。这种模式不再把所有的下载压力丢给服务器，服务器只负责传递文件元数据，真正的文件下载连接建立在客户机与客户机之间。同时一个文件可以被分片为多个块，同一个文件中不同的块可以在不同的客户机之上下载，使得下载文件在 P2P 网络中动态流通，大幅提升了下载效率，如下图：去中心化的 P2P 下载基于 DHT 技术，它采用分布式全网方式来进行信息的存储和检索。所有信息均以哈希...

2022-09-06

451

资源下载

更多资源

优质分享App

近一个月的开发和优化，本站点的第一个app全新上线。该app采用极致压缩，本体才4.36MB。系统里面做了大量数据访问、缓存优化。方便用户在手机上查看文章。后续会推出HarmonyOS的适配版本。

Mario

马里奥是站在游戏界顶峰的超人气多面角色。马里奥靠吃蘑菇成长，特征是大鼻子、头戴帽子、身穿背带裤，还留着胡子。与他的双胞胎兄弟路易基一起，长年担任任天堂的招牌角色。

Spring

Spring框架（Spring Framework）是由Rod Johnson于2002年提出的开源Java企业级应用框架，旨在通过使用JavaBean替代传统EJB实现方式降低企业级编程开发的复杂性。该框架基于简单性、可测试性和松耦合性设计理念，提供核心容器、应用上下文、数据访问集成等模块，支持整合Hibernate、Struts等第三方框架，其适用范围不仅限于服务器端开发，绝大多数Java应用均可从中受益。

Rocky Linux

Rocky Linux（中文名：洛基）是由Gregory Kurtzer于2020年12月发起的企业级Linux发行版，作为CentOS稳定版停止维护后与RHEL（Red Hat Enterprise Linux）完全兼容的开源替代方案，由社区拥有并管理，支持x86_64、aarch64等架构。其通过重新编译RHEL源代码提供长期稳定性，采用模块化包装和SELinux安全架构，默认包含GNOME桌面环境及XFS文件系统，支持十年生命周期更新。