# 网络模型库

[![查看源文件](https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg)](https://gitee.com/mindspore/docs/blob/master/docs/sciai/docs/source_zh_cn/model_library.md)  

SciAI基础模型库提供了丰富的科学计算高频模型,下表中汇总了当前已实现的网络模型及其对应领域。

| 领域    | 模型                                                                                                                       |                                                                                                                                                                                                            MindSpore实现与网络参数                                                                                                                                                                                                             | Ascend | GPU |
|-------|--------------------------------------------------------------------------------------------------------------------------|:---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------:|:------:|:---:|
| 通用物理  | [auq_pinns](https://www.sciencedirect.com/science/article/pii/S0021999119303584)                                         |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/auq_pinns/README_CN.md#脚本参数)                                                                                                                                                                |   ✅    |  ✅  |
| 通用物理  | [cpinns](https://www.sciencedirect.com/science/article/abs/pii/S0045782520302127)                                        |                                                                                                                                                                 [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/cpinns/README_CN.md#脚本参数)                                                                                                                                                                  |   ✅    |  ✅  |
| 通用物理  | [deep_hpms](https://www.jmlr.org/papers/volume19/18-046/18-046.pdf)                                                      |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/deep_hpms/README_CN.md#脚本参数)                                                                                                                                                                |   ✅    |  ✅  |
| 通用物理  | [deep_ritz](https://arxiv.org/abs/1710.00211)                                                                            |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/deep_ritz/README_CN.md#脚本参数)                                                                                                                                                                |   ✅    |  ✅  |
| 通用物理  | [deepbsde](https://www.pnas.org/doi/10.1073/pnas.1718942115)                                                             |                                                                                                                                                           [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/deepbsde/README.md#script-parameters)                                                                                                                                                            |        |  ✅  |
| 通用物理  | [deeponet](https://www.nature.com/articles/s42256-021-00302-5)                                                           |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/deeponet/README_CN.md#脚本参数)                                                                                                                                                                 |        |  ✅  |
| 通用物理  | [dgm](https://arxiv.org/abs/1708.07469)                                                                                  |                                                                                                                                                                   [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/dgm/README_CN.md#脚本参数)                                                                                                                                                                   |   ✅    |  ✅  |
| 通用物理  | [fbsnns](https://arxiv.org/abs/1804.07010)                                                                               |                                                                                                                                                                 [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/fbsnns/README_CN.md#脚本参数)                                                                                                                                                                  |   ✅    |  ✅  |
| 通用物理  | [fpinns](https://arxiv.org/abs/1811.08967)                                                                               |                                                                                                                                                                 [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/fpinns/README_CN.md#脚本参数)                                                                                                                                                                  |   ✅    |  ✅  |
| 通用物理  | [gradient_pathologies_pinns](https://arxiv.org/abs/2001.04536)                                                           |                                                                                                                                                       [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/gradient_pathologies_pinns/README_CN.md#脚本参数)                                                                                                                                                        |   ✅    |  ✅  |
| 通用物理  | [hp_vpinns](https://arxiv.org/abs/2003.05385)                                                                            |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/hp_vpinns/README_CN.md#脚本参数)                                                                                                                                                                |   ✅    |  ✅  |
| 通用物理  | [laaf](https://doi.org/10.1016/j.jcp.2019.109136)                                                                        |                                                                                                                                                                  [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/laaf/README_CN.md#脚本参数)                                                                                                                                                                   |   ✅    |  ✅  |
| 通用物理  | [mgnet](https://link.springer.com/article/10.1007/s11425-019-9547-2)                                                     |                                                                                                                                                                  [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/mgnet/README_CN.md#脚本参数)                                                                                                                                                                  |   ✅    |  ✅  |
| 通用物理  | [multiscale_pinns](https://www.sciencedirect.com/science/article/abs/pii/S0045782521002759)                              |                                                                                                                                                            [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/multiscale_pinns/README_CN.md#脚本参数)                                                                                                                                                             |   ✅    |  ✅  |
| 通用物理  | [pfnn](https://www.sciencedirect.com/science/article/abs/pii/S0021999120308597)                                          |                                                                                                                                                                  [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pfnn/README_CN.md#脚本说明)                                                                                                                                                                   |        |  ✅  |
| 通用物理  | [phygeonet](https://www.sciencedirect.com/science/article/abs/pii/S0021999120308536)                                     |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/phygeonet/README_CN.md#脚本参数)                                                                                                                                                                |   ✅    |  ✅  |
| 通用物理  | [pi_deeponet](https://www.sciencedirect.com/science/article/abs/pii/S0021999122009184)                                   |                                                                                                                                                               [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pi_deeponet/README_CN.md#脚本参数)                                                                                                                                                               |        |  ✅  |
| 通用物理  | [pinns](https://www.sciencedirect.com/science/article/abs/pii/S0021999118307125)                                         |                                                                                                                                                                  [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pinns/README_CN.md#脚本参数)                                                                                                                                                                  |        |  ✅  |
| 通用物理  | [pinns_ntk](https://www.sciencedirect.com/science/article/pii/S002199912100663X)                                         |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pinns_ntk/README_CN.md#脚本参数)                                                                                                                                                                |   ✅    |  ✅  |
| 通用物理  | [ppinns](https://www.sciencedirect.com/science/article/abs/pii/S0045782520304357)                                        |                                                                                                                                                                 [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/ppinns/README_CN.md#脚本参数)                                                                                                                                                                  |   ✅    |  ✅  |
| 通用物理  | [xpinns](https://doi.org/10.4208/cicp.OA-2020-0164)                                                                      |                                                                                                                                                                 [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/xpinns/README_CN.md#脚本参数)                                                                                                                                                                  |   ✅    |  ✅  |
| 哈密顿系统 | [sympnets](https://www.sciencedirect.com/science/article/pii/S0893608020303063)                                          |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/sympnets/README_CN.md#脚本参数)                                                                                                                                                                 |   ✅    |  ✅  |
| 流体力学  | [hfm](https://www.science.org/doi/abs/10.1126/science.aaw4741)                                                           |                                                                                                                                                                   [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/hfm/README_CN.md#脚本参数)                                                                                                                                                                   |   ✅    |  ✅  |
| 流体力学  | [label_free_dnn_surrogate](https://www.sciencedirect.com/science/article/pii/S004578251930622X)                          |                                                                                                                                                        [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/label_free_dnn_surrogate/README_CN.md#脚本参数)                                                                                                                                                         |   ✅    |  ✅  |
| 流体力学  | [nsf_nets](https://www.sciencedirect.com/science/article/pii/S0021999120307257)                                          |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/nsf_nets/README_CN.md#脚本参数)                                                                                                                                                                 |   ✅    |  ✅  |
| 流体力学  | [*burgers_fno](https://arxiv.org/abs/2010.08895)                                                                         |                                                                                                                                                       [link](https://gitee.com/mindspore/mindscience/blob/master/MindFlow/applications/data_driven/burgers/fno1d/FNO1D_CN.ipynb)                                                                                                                                                        |   ✅    |  ✅  |
| 流体力学  | [*burgers_kno](https://arxiv.org/abs/2301.10022)                                                                         |                                                                                                                                                       [link](https://gitee.com/mindspore/mindscience/blob/master/MindFlow/applications/data_driven/burgers/kno1d/KNO1D_CN.ipynb)                                                                                                                                                        |   ✅    |  ✅  |
| 流体力学  | [*navier_stokes_fno](https://arxiv.org/abs/2010.08895)                                                                   |                                                                                                                                                    [link](https://gitee.com/mindspore/mindscience/blob/master/MindFlow/applications/data_driven/navier_stokes/fno2d/FNO2D_CN.ipynb)                                                                                                                                                     |   ✅    |  ✅  |
| 流体力学  | [*navier_stokes_kno](https://arxiv.org/abs/2301.10022)                                                                   |                                                                                                                                                    [link](https://gitee.com/mindspore/mindscience/blob/master/MindFlow/applications/data_driven/navier_stokes/kno2d/KNO2D_CN.ipynb)                                                                                                                                                     |   ✅    |  ✅  |
| 流体力学  | [*navier_stokes_3d_fno](https://arxiv.org/abs/2010.08895)                                                                |                                                                                                                                                    [link](https://gitee.com/mindspore/mindscience/blob/master/MindFlow/applications/data_driven/navier_stokes/fno3d/FNO3D_CN.ipynb)                                                                                                                                                     |   ✅    |  ✅  |
| 流体力学  | [*pde_net](https://arxiv.org/abs/1710.09668)                                                                             |                                                                                                                                                      [link](https://gitee.com/mindspore/mindscience/blob/master/MindFlow/applications/data_mechanism_fusion/pde_net/README_CN.md)                                                                                                                                                       |   ✅    |  ✅  |
| 流体力学  | [*percnn](https://www.nature.com/articles/s42256-023-00685-7)                                                            |                                                                                                                                                       [link](https://gitee.com/mindspore/mindscience/blob/master/MindFlow/applications/data_mechanism_fusion/percnn/README_CN.md)                                                                                                                                                       |   ✅    |  ✅  |
| 弹性动力学 | [pinn_elastodynamics](https://arxiv.org/abs/2006.08472)                                                                  |                                                                                                                                                           [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pinn_elastodynamics/README_CN.md#脚本参数)                                                                                                                                                           |   ✅    |  ✅  |
| 热力学   | [pinn_heattransfer](https://arxiv.org/abs/1711.10561)                                                                    |                                                                                                                                                            [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pinn_heattransfer/README_CN.md#脚本参数)                                                                                                                                                            |   ✅    |  ✅  |
| 气象学   | [enso](https://doi.org/10.1038/s41586-019-1559-7)                                                                        |                                                                                                                                                                  [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/enso/README_CN.md#脚本参数)                                                                                                                                                                   |   ✅    |  ✅  |
| 地质学   | [inversion_net](https://ieeexplore.ieee.org/abstract/document/8918045/)                                                  |                                                                                                                                                              [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/inversion_net/README_CN.md#脚本参数)                                                                                                                                                              |   ✅    |  ✅  |
| 地质学   | [pinn_helmholtz](https://academic.oup.com/gji/article-abstract/228/3/1750/6409132)                                       |                                                                                                                                                             [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pinn_helmholtz/README_CN.md#脚本参数)                                                                                                                                                              |   ✅    |  ✅  |
| 海洋物理  | [ocean_model](https://gmd.copernicus.org/articles/12/4729/2019/)                                                         |                                                                                                                                                               [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/ocean_model/README_CN.md#模型说明)                                                                                                                                                               |        |  ✅  |
| 海洋物理  | [pinns_swe](https://arxiv.org/abs/2104.00615)                                                                            |                                                                                                                                                                [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/pinns_swe/README_CN.md#脚本参数)                                                                                                                                                                |   ✅    |  ✅  |
| 电磁学   | [maxwell_net](https://arxiv.org/abs/2107.06164)                                                                          |                                                                                                                                                               [link](https://gitee.com/mindspore/mindscience/blob/master/SciAI/sciai/model/maxwell_net/README_CN.md#脚本参数)                                                                                                                                                               |   ✅    |  ✅  |
| 电磁学   | [*AD_FDTD_invert_f](https://www.mindspore.cn/mindelec/docs/zh-CN/r0.2/AD_FDTD.html)                                      |                                                                                                                                                          [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/AD_FDTD/fdtd_forward/README_CN.md#脚本参数)                                                                                                                                                           |        |  ✅  |
| 电磁学   | [*AD_FDTD_microstrip_filter](https://www.mindspore.cn/mindelec/docs/zh-CN/r0.2/AD_FDTD.html)                             |                                                                                                                                                          [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/AD_FDTD/fdtd_forward/README_CN.md#脚本参数)                                                                                                                                                           |        |  ✅  |
| 电磁学   | [*AD_FDTD_inverse](https://www.mindspore.cn/mindelec/docs/zh-CN/r0.2/AD_FDTD.html)                                       |                                                                                                                                                          [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/AD_FDTD/fdtd_inverse/README_CN.md#脚本参数)                                                                                                                                                           |        |  ✅  |
| 电磁学   | [*frequency_domain_maxwell](https://arxiv.org/abs/2107.06164)                                                            |                                                                                                                                                 [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/physics_driven/frequency_domain_maxwell/README_CN.md#脚本参数)                                                                                                                                                 |   ✅    |  ✅  |
| 电磁学   | [*frequency_domain_maxwell_3D_dielectric_slab](https://arxiv.org/abs/2107.06164)                                         |                                                                                                                                       [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/physics_driven/frequency_domain_maxwell_3D/dielectric_slab_3d/README.md#脚本参数)                                                                                                                                        |   ✅    |  ✅  |
| 电磁学   | [*frequency_domain_maxwell_3D_waveguide_cavity](https://arxiv.org/abs/2107.06164)                                        |                                                                                                                                       [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/physics_driven/frequency_domain_maxwell_3D/waveguide_cavity_3d/README.md#脚本参数)                                                                                                                                       |   ✅    |  ✅  |
| 电磁学   | [*meta_auto_decoder](https://arxiv.org/abs/2111.08823)                                                                   |                                                                                                                                                   [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/physics_driven/incremental_learning/README_CN.md#脚本参数)                                                                                                                                                   |   ✅    |  ✅  |
| 电磁学   | [*pinn_fwi](https://agupubs.onlinelibrary.wiley.com/doi/abs/10.1029/2021JB023120)                                        |                                                                                                                                                          [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/physics_driven/pinn_fwi/README.md#脚本参数)                                                                                                                                                           |   ✅    |  ✅  |
| 电磁学   | [*SED_ANN](https://gitee.com/mindspore/mindscience/tree/master/MindElec/examples/data_driven/sed_ann)                    |                                                                                                                                                           [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/data_driven/sed_ann/README_CN.md#脚本参数)                                                                                                                                                           |   ✅    |  ✅  |
| 电磁学   | [*time_domain_maxwell](https://www.ijcai.org/proceedings/2022/533)                                                       |                                                                                                                                                   [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/physics_driven/time_domain_maxwell/README_CN.md#脚本参数)                                                                                                                                                    |   ✅    |  ✅  |
| 电磁学   | [*metasurface_holograms](https://www.researching.cn/articles/OJ44d3746c3db8c1e1)                                         |                                                                                                                                                    [link](https://gitee.com/mindspore/mindscience/blob/master/MindElec/examples/metasurface/metasurface_holograms/README_CN.md#脚本参数)                                                                                                                                                    |   ✅    |  ✅  |
| 生物    | [*MEGA-Fold](https://arxiv.org/abs/2206.12240v1)                                                                         |        [link (推理)](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/MEGAProtein.md) [link (训练)](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/MEGAProtein.md)        |   ✅    |  ✅  |
| 生物    | [*MEGA-EvoGen](https://arxiv.org/abs/2208.09652)                                                                         |                                                                                                                           [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/MEGAProtein.md)                                                                                                                            |   ✅    |  ✅  |
| 生物    | [*MEGA-Assessment](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/MEGAProtein.md) | [link (推理)](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/MEGAProtein.md) [link (训练)](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/MEGAProtein.md) |   ✅    |  ✅  |
| 生物    | [*ColabDesign](https://www.biorxiv.org/content/10.1101/2021.11.10.468128.abstract)                                       |                                                                                                                                                            [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/ColabDesign.md)                                                                                                                                                             |   ✅    |  ✅  |
| 生物    | [*DeepFRI](https://www.nature.com/articles/s41467-021-23303-9)                                                           |                                                                                                                                                              [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/DeepFri.md)                                                                                                                                                               |   ✅    |  ✅  |
| 生物    | [*Multimer](https://www.biorxiv.org/content/10.1101/2021.10.04.463034v1)                                                 |                                                                                                                                                             [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/afmultimer.md)                                                                                                                                                             |   ✅    |  ✅  |
| 生物    | [*ProteinMPNN](https://www.science.org/doi/abs/10.1126/science.add2187)                                                  |                                                                                                                                                            [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/ProteinMPNN.MD)                                                                                                                                                             |   ✅    |  ✅  |
| 生物    | [*UFold](https://doi.org/10.1093/nar/gkab1074)                                                                           |                                                                                                                                                               [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/UFold.md)                                                                                                                                                                |   ✅    |  ✅  |
| 生物    | [*esm-if1](https://proceedings.mlr.press/v162/hsu22a.html)                                                               |                                                                                                                                                              [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/ESM-IF1.md)                                                                                                                                                               |   ✅    |  ✅  |
| 生物    | [*esm2](https://www.biorxiv.org/content/10.1101/2022.07.20.500902v1.full.pdf)                                            |                                                                                                                                                               [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/ESM-2.md)                                                                                                                                                                |   ✅    |  ✅  |
| 生物    | [*grover](https://proceedings.neurips.cc/paper/2020/file/94aef38441efa3380a3bed3faf1f9d5d-Paper.pdf)                     |                                                                                                                                                               [link](https://gitee.com/mindspore/mindscience/blob/master/MindSPONGE/applications/model_cards/GROVER.MD)                                                                                                                                                               |   ✅    |  ✅  |

注: 带有“*”的网络模型为MindSpore与MindScience先前已发布网络。