【百万倍加速】加速计算助力基因测序突破极限
2023-02-28 10:57
在 1 月 13 日刚刚结束的摩根大通医疗健康大会上,NVIDIA 发布了与初创公司 InstaDeep、慕尼黑工业大学(TUM)在基因组学大型语言模型方面的联合研究成果。研究团队使用 NVIDIA 超级计算机 Cambridge-1,在各种基因组数据集上,训练了参数规模从 500M 到 2.5B 不等的各种大型语言模型(LLM),经过基因组学训练的大型语言模型可将应用扩展到大量基因组学任务, 这些任务有助于了解 DNA 如何转录生成 RNA 和蛋白质,从而开启新的临床应用。
基因测序 梦想照进现实
作为人类医疗技术发展的重要一步,基因测序可通过提取人体血液或唾液,对人类的 DNA、RNA、蛋白质及代谢物进行分析,从而锁定个人病变基因,以诊断、预测或预防遗传性疾病的发生。基因蕴含了人类诸多病症的秘密,使人类有机会尽早预防疾病,并多元化治疗疾病。
2001 年,被誉为生命科学“登月计划”的“人类基因组计划”(HGP)历时 13 年,耗资近 30 亿美元,发布了首张人类基因组草图。此后二十年中,以数字生物学为基础的基因组测序技术取得巨大进步,将全基因组测序的成本逐步降低至不到 1000 美元。一时间,全球对基因测序与分析的需求量出现暴涨。
加速计算突破基因分析算力瓶颈
随着基因组测序和分析需求的激增,由此产生的数据也迎来爆发式增长。过去十年间,基因测序的数据量平均每七个月就会翻一倍,2025 年全球测序能力预计将达到 Zb 级别,其中,仅人类全基因组存储的数据量就将达到每年 2-40 Eb。
要实现精准医疗,就需要对大量基因数据做出精确的计算和分析,这意味着巨大的计算成本和时间。因此,必须有创新的技术,打破全基因组测序数据医疗应用的主要瓶颈,推动行业进一步向前发展。
基因组测序与分析属于标准的计算密集型任务,而 GPU 擅长并行计算,可以将复杂计算问题分解为很多小任务,在多个 CUDA 内核上同时运行,从而大幅减少处理计算任务的时间,让基因组测序分析的速度获得大幅提升。NVIDIA 以 GPU 为核心构建出“全栈加速计算平台”,与人工智能相结合,在全球包括基因测序在内的多个领域掀起一场加速革命。
除了具有全球领先的硬件加速计算平台,NVIDIA 还推出了基于 GPU 的基因测序分析加速软件— NVIDIA Clara Parabricks。Clara Parabricks 是一款 GPU 加速的计算基因组学工具包,可提供多种生物信息学工具和功能,如序列比对、预处理和质量指标、变异检测、UMI 以及用于胚系、体细胞和 RNA 分析的端到端工作流程。与仅使用 CPU 的解决方案相比,Parabricks 速度提升高达 80 倍,计算成本降低高达 50%。借助 Clara Parabricks 和 GPU,将深度学习的强大功能应用到基因组分析,可为测序中心、临床团队、基因组学研究人员以及新一代测序仪器开发者提供快速准确的分析,进而助力更快速、更准确的基因组学分析。
此外,为了更好地帮助科学家了解疾病,为患者找到治疗方法,NVIDIA 还推出了 BioNeMo 框架,用于训练和部署超算规模的大型生物分子语言模型。该大型语言模型框架可支持化学、蛋白质、DNA 和 RNA 数据格式,还可提供云 API 服务,用于未来支持越来越多的预训练 AI 模型。
基因测序领域降本增效创造新记录
在 NVIDIA 加速计算平台、人工智能与基因测序专业软件的赋能下,来自全球的学研机构、政府组织以及相关企业,正在不断突破基因测序与分析领域的速度与成本极限,创造出一项项新的记录:
斯坦福大学、NVIDIA、牛津纳米孔科技公司、谷歌、贝勒医学院和加州大学圣克鲁兹分校的研究人员共同合作推出 DNA 测序技术,实现了创纪录的测序速度。该方案使用 NVIDIA Clara Parabricks 计算基因组学应用框架,以及 GPU 加速版本的 PEPPER-Margin-DeepVariant 流程,在谷歌云上使用 NVIDIA GPU 加速碱基判定和变体识别,将基因诊断时间从数周缩短到 5.2 小时,使医生能够快速判断如何治疗危重病人。
麻省理工学院与哈佛大学旗下的博德研究所与 NVIDIA 合作,为拥有超过 2.5 万用户的 Terra 云平台提供快速分析海量医疗数据所需的 AI 能力和加速工具。通过 GPU 加速的 NVIDIA Clara Parabricks 工作流,可助力从事各种基因组数据分析的研究人员降本增效。在博德研究所的 GATK 最佳实践—生殖细胞突变检测分析流中,Parabricks 在 GPU 上进行分析的速度提高了 24 倍,而成本减半。
阿里云与 NVIDIA 达成合作,成为国内首家在基因分析平台集成和部署 NVIDIA Clara Parabricks 的公有云厂商。Clara Parabricks 为阿里云基因分析平台的行业标准基因组分析工具包以及基因调用器等工具带来了 GPU 加速,据测算,用户可以在 30 分钟内完成一个 30x 测序深度的全基因组分析,且与 GATK 的最佳实践结果 99.99% 一致,而在过去基于 CPU 环境中这项工作需要 20 多个小时才能完成,并且计算成本可降低 40%-80%。
普渡大学首席研究员 Carpi 博士和团队将 Clara Parabricks 的表现和疟疾学界使用的鉴定变体和跟踪疟疾传播的现有方法进行了对比,并使用 1000 个疟疾基因组监测抗疟药物的耐药性。与只使用 CPU 的传统方式相比,基于 GPU 加速的 Clara Parabricks 分析速度提高了 27 倍,成本降低了 5 倍,同时准确率达到 99.9%。
圣路易斯华盛顿大学的 Tychele Turner 博士与团队使用基于 GPU 加速的 Clara Parabricks 开发出一种快速基因组学工作流,用于在自闭症患者中发现 de novo 变异(DNVs)。通过将三重分析整合到 NVIDIA Clara Parabricks 中,Turner 博士将生成 DNV 初始分析的时间从原来的 800 小时(在 CPU 上进行)缩短至 8.5 小时(在 GPU 上使用只有 4 个 GPU 的服务器),加快了 100 倍。
为推广基因组医学计划,泰国国家生物库采用 NVIDIA DGX A100 系统和 NVIDIA Clara Parabricks 来打造 IT 基础设施,以加速基因组测序。加速方案将全基因组测序的数据处理时间缩短了 4 个月,每位用户的处理时间也从 30 多个小时缩短到 1 到 2 小时。
生物技术领军企业 Regeneron 高通量测序中心与大规模的生物医学数据库和研究资源平台英国生物样本库合作,在 DNAnexus 平台上使用 NVIDIA Clara Parabricks 运行分析对超过 50 万名生物库参与者的外显子组进行了测序和分析,5 分钟内完成了在 32-vCPU 机器上需要花费一小时计算的全外显子组分析,同时成本降低约 40%。
凭借加速计算领域的全栈能力以及深耕行业所积累的专业知识,NVIDIA 现在已经推出了智能计算平台 NVIDIA Clara,全面布局医疗健康行业。该平台涵盖了用于医学影像的 Clara Holoscan、基因组学的 Clara Parabricks、患者监控的 Clara Guardian 以及药物研发的 Clara Discovery,并可端到端部署至本地、嵌入式系统、边缘以及云端等任何地方,助力行业实现无缝创新,加快实现精准医疗的目标。
无论是尖端医学研究、大规模基因组分析还是日常应用检测,NVIDIA Million-X 百万倍计算性能飞跃的愿景已经在基因测序的各个领域结出累累硕果,并将持续为人类消除疾病、促进健康做出贡献。