Automatic network structure discovery of physics informed neural networks via knowledge distillation
Partial differential equations (PDEs) are fundamental for modeling complex physical processes, often exhibiting structural features such as symmetries and conservation laws. While physics-informed neural networks (PINNs) can simulate and invert PDEs, they mainly rely on external loss functions for physical constraints, making it difficult to automatically discover and embed physically consistent network structures. We propose a physics structure-informed neural network discovery method based on physics-informed distillation, which decouples physical and parameter regularization via staged optimization in teacher and student networks. After distillation, clustering and parameter reconstruction are used to extract and embed physically meaningful structures. Numerical experiments on Laplace, Burgers, and Poisson equations, as well as fluid mechanics, show that the method can automatically extract relevant structures, improve accuracy and training efficiency, and enhance structural adaptability and transferability. This approach offers a new perspective for efficient modeling and automatic discovery of structured neural networks.
Authors
- Xunshi Yan (ORCID: https://orcid.org/0000-0002-0372-7120)
- Han Nie (ORCID: https://orcid.org/0000-0002-9967-3998)
- Yang Liu (ORCID: https://orcid.org/0000-0003-1098-1116)
- Ziti Liu (ORCID: https://orcid.org/0009-0001-8422-1606)
- Liu Wen (ORCID: https://orcid.org/0000-0002-2171-282X)
- Shuaiqi Guo
- Chen-An Zhang (ORCID: https://orcid.org/0000-0003-0119-7100)
Institutions
- Chinese Academy of Sciences (CN)
- Institute of Mechanics (CN)
- University of Chinese Academy of Sciences (CN)
- Tsinghua University (CN)
Publication Details
- Journal
- Nature Communications
- Published
- 2025-10-29
- DOI
- https://doi.org/10.1038/s41467-025-64624-3
- Citations
- 5
- Primary Topic
- Model Reduction and Neural Networks
- Type
- article
- Field-Weighted Citation Impact
- 6.62
Funders
- Chinese Academy of Sciences
- Youth Innovation Promotion Association