PINNacle

This repository is our codebase for PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs.

Our NIPS paper is currently under review. We will provide more detailed guide soon.

Installation

# install torch manually
# install other dependencies
pip install -r requirements.txt

Usage

Run all 22 cases with default settings:

python benchmark.py [--name EXP_NAME] [--seed SEED] [--device DEVICE]

Please read the comments in benchmark.py to modify the experiment configuration and apply different methods.

Multi-GPU Training

One can use --device to specify the GPU devices to use. If more than one GPU is specified, the program will run multiple experiments in parallel.

Citation

If you find out work useful, please cite our paper at:

@misc{hao2023pinnacle,
      title={PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEs}, 
      author={Zhongkai Hao and Jiachen Yao and Chang Su and Hang Su and Ziao Wang and Fanzhi Lu and Zeyu Xia and Yichi Zhang and Songming Liu and Lu Lu and Jun Zhu},
      year={2023},
      eprint={2306.08827},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}