Multi-head PINNs for learning functional priors and UQ

This repository contains source code for the paper: Multi-head physics-informed neural networks for learning functional priors and uncertainty quantification. It requires the following dependencies to be installed:

• Python 3.8
• Tensorflow 2.9.0
• TensorFlow Probability 0.17.0
• Numpy 1.24.4
• Scipy 1.10.1
• Matplotlib 3.7.5

Cite us

@article{zou2025multi,
  title={Multi-head physics-informed neural networks for learning functional priors and uncertainty quantification},
  author={Zou, Zongren and Karniadakis, George Em},
  journal={Journal of Computational Physics},
  pages={113947},
  year={2025},
  publisher={Elsevier}
}