NVIDIA 创建 AI 学习物理框架

NVIDIA Modulus框架训练创新物理学-机器学习模型,加速工业数字孪生、气候科学、蛋白质工程等领域的发展
作者 英伟达中国

美国加利福尼亚州圣克拉拉——GTC大会——2021119日——NVIDIA于今日发布的AI框架为工程师、科学家和研究者提供了一个可定制、易于采用的物理学工具包,使他们能够通过建立数字孪生神经网络模型加速解决当今一些最具挑战性的问题。

NVIDIA Modulus是一个用于开发物理学-机器学习模型的框架,它能够为缺乏AI专业知识,但对AI和物理驱动型数字孪生功能的需求快速增长的众多领域提供支持,例如蛋白质工程和气候科学领域。

数字孪生已成为解决从分子层面(如药物研发)到全球挑战(如气候变化)等各种问题的有力工具。NVIDIA Modulus为科学家所提供的框架能够为复杂、动态的系统构建高精度数字复制品,从而推动各行业的新一代技术突破。

基于物理学的神经网络

Modulus训练神经网络使用基本的物理学定律模拟各领域中复杂系统的行为。从工业用例到气候科学,该代理模型可用于各种数字孪生应用。

与大多数基于AI的方法一样,Modulus内置一个帮助管理观察或模拟数据的数据准备模块。它还能解释它所模拟的系统的几何图形以及输入几何图形所表示的空间的显式参数。

Modulus的关键工作流程和要素包括:

  • 采样计划器:使用户能够选择一种方法(如准随机采样或重要性采样)来提高被训练模型的收敛性和准确性。
  • 基于Python的API:采取象征性管理偏微分方程并构建基于物理学的神经网络。
  • 精选层和网络架构 :经证明能够有效解决物理学问题。
  • 物理学-机器学习引擎:使用这些输入来训练模型。所训练的模型能够使用PyTorch与TensorFlow、使用cuDNN实现GPU加速并且使用NVIDIA® Magnum IO实现多GPU和多节点扩展。

快速周转时间

GPU加速工具包实现快速周转,补充传统的分析并实现更快的洞察。Modulus使用户通过评估能够改变其参数的影响来探索系统的不同配置和场景。

基于高性能TensorFlow的Modulus实现使用XLA优化性能。XLA是一个用于加速TensorFlow模型的特定领域线性代数编译器。它使用Horovod分布式深度学习训练框架实现多GPU扩展。

在完成模型训练后,Modulus可以进行近乎实时的推理或互动式推理。相比之下,传统的分析每次运行时都要进行评估,而且每次评估的计算成本很高。

易于采用

Modulus可定制并且易于采用。它能提供用于实现新物理学和几何学的API。Modulus在设计上可以使那些刚开始使用AI数字孪生应用的人能够快速将其用于工作。

该框架包括计算流体力学、热传导等入门分步教程。它还为应用领域提供了一个不断增长的实现列表,如湍流建模、瞬态波方程、纳维-斯托克斯方程、电磁学领域的麦克斯韦方程、反问题和其他多物理场问题。

供应情况

现在可在NVIDIA开发者区域免费下载NVIDIA Modulus。