MindSpore Science 文档
======================
简介
----
将 AI 与科学计算相结合(即 AI+科学计算),是指利用人工智能技术(如机器学习、深度学习等)对科学难题进行计算和分析。这种结合突破了传统数学模型和算法的局限,能够借助 AI 强大的计算能力,探索未知领域,并提升计算效率与准确性。AI+科学计算的能力主要来源于大数据的支持和算法的优化,通过不断学习和迭代,AI 可以更好地应对复杂的科学计算问题。
MindSpore Science 是基于 MindSpore 融合架构打造的高性能科学计算行业套件,提供业界领先的数据集、SOTA 模型、常用模型接口和前后处理工具。该套件面向 Ascend 深度优化,加速科学行业应用开发。
支持特性
--------
MindScience-Core 是 MindScience 的核心组件,为开发者提供易用性强、昇腾亲和的常用模型接口,加速 AI4Science 模型开发,主要提供以下几个能力:
常用模型接口:
- 图计算:由于图的拓扑结构天然适配网格、分子结构等数据形态,AI4Science 领域的模型大量采用以 GNN(Graph Neural Network,图神经网络)为核心的架构。MindScience 为此提供了一套功能完备、与 Ascend 平台深度亲和的 GNN 接口,同时保持与业界主流框架接口的兼容性,有效提升了相关模型的开发效率。
- 等变计算:等变计算库 e3nn 专为处理具有对称性的数据场景设计,尤其适配分子、晶体等包含旋转、平移等对称属性的结构。它提供了一套完整的等变神经网络构建工具,能精准捕捉数据的对称性特征,研究者可高效开发符合对称性原理的模型,显著提升分子模拟、材料科学等领域的任务表现。
- 物理信息神经网络:物理信息神经网络库 PINNs(Physics-Informed Neural Networks)专为融合物理规律与数据驱动建模设计,尤其适用于流体力学、热传导等偏微分方程(PDE)求解场景。MindScience 的 pde 模块提供了将物理守恒律、边界条件等约束嵌入神经网络的完整工具链,研究者无需依赖大量标注数据,即可高效构建符合物理规律的模型,大幅提升工程仿真、科学计算等领域的求解效率与泛化能力。
- 可微分求解器:可微分求解器是一类融合数值计算与自动微分能力的专用工具,核心用于高效求解常微分方程(ODE)、偏微分方程(PDE)等科学与工程问题。它能在输出数值解的同时,反向传播计算梯度信息,完美兼容 MindSpore 深度学习框架。借助这一特性,研究者可将物理方程的求解过程嵌入神经网络训练流程,为科学计算与深度学习的结合提供关键支撑,广泛应用于动力学模拟、控制策略优化等 AI4Science 场景。
- 科学计算算子:科学计算算子是支撑 AI4Science 模型构建的核心基础组件,算子经过与 Ascend 结合的深度优化,实现复杂科学计算任务的高效执行。
- 分布式并行:分布式并行是应对 AI4Science 大规模任务的核心高效计算方案,通过将海量数据、复杂模型或计算任务拆分到多节点、多设备集群中协同处理,突破单设备的算力与内存瓶颈,为 AI4Science 向更复杂、更精细的研究方向拓展提供了关键算力保障。
典型模型支持:
--------------
.. raw:: html
MindChem
.. list-table::
:header-rows: 1
* - 应用名称
- 简介
- 学习类型
* - `DiffCSP `__
- 晶体结构生成
- 扩散模型/生成式学习
* - `NequIP `__
- 分子力场
- 监督学习
* - `Orb `__
- 分子力场
- 监督学习
* - `MatFormer `__
- 材料性质预测
- 监督学习
* - `DeepHE3nn `__
- 哈密顿量预测
- 监督学习
* - `CrystalFlow `__
- 晶体结构生成
- 扩散模型/生成式学习
.. raw:: html
.. raw:: html
MindEarth
.. list-table::
:header-rows: 1
* - 应用名称
- 简介
- 学习类型
* - `LeadFormer `__
- 北极海冰冰间水道预测
- 监督学习
.. raw:: html
.. raw:: html
MindEnergy
.. list-table::
:header-rows: 1
* - 应用名称
- 简介
- 学习类型
* - `DAE-PINN `__
- 物理信息神经网络
- 监督+物理约束
* - `DeepONet-Grig-UQ `__
- 电网故障后轨迹预测
- 监督学习
* - `PowerFlowNet `__
- 潮流计算
- 监督学习
.. raw:: html
.. raw:: html
MindFlow
.. list-table::
:header-rows: 1
* - 应用名称
- 简介
- 学习类型
* - `Acoustic `__
- 声学模拟
- 自动迭代学习
* - `P2C2Net `__
- 物理建模求解二维Burgers方程组
- 监督学习
* - `FNO2D `__
- 傅里叶神经算子求解二维Navier-Stokes方程组
- 监督学习
.. raw:: html
.. raw:: html
MindSPONGE
.. list-table::
:header-rows: 1
* - 应用名称
- 简介
- 学习类型
* - `AlphaFold3 `__
- 蛋白质结构预测
- 监督学习
* - `ProteinMPNN `__
- 蛋白质序列设计
- 监督学习
* - `RFdiffusion `__
- 蛋白质扩散生成
- 扩散模型/生成式学习
* - `Protenix `__
- 蛋白质功能预测
- 监督学习
.. raw:: html
.. toctree::
:maxdepth: 1
:caption: 组件介绍
:hidden:
MindChem
MindEarth
MindEnergy
MindFlow
MindSPONGE
.. toctree::
:maxdepth: 1
:caption: 安装指南
:hidden:
install.md
.. toctree::
:maxdepth: 1
:caption: 快速开始
:hidden:
quick_start.md
.. toctree::
:maxdepth: 1
:caption: API参考
:hidden:
api/mindscience.common
api/mindscience.data
api/mindscience.diffuser
api/mindscience.distributed
api/mindscience.e3nn
api/mindscience.gnn
api/mindscience.models
api/mindscience.pde
api/mindscience.sciops
api/mindscience.solvers
api/mindscience.utils
.. toctree::
:maxdepth: 1
:caption: 贡献指南
:hidden:
CONTRIBUTING
.. toctree::
:maxdepth: 1
:caption: RELEASE NOTES
:hidden:
RELEASE