Monte Carlo Neural PDE Solver (MCNP Solver)

This repository contains the code for the paper

• Monte Carlo Neural PDE Solver for Learning PDEs via Probabilistic Representation

Quick Start

To generate the data-set, use,

python generate_data.py

To train MCNP, use

python main.py

Citation

If you find our work useful in your research, please consider citing:

@misc{zhang2023monte,
      title={Monte Carlo Neural PDE Solver for Learning PDEs via Probabilistic Representation}, 
      author={Rui Zhang and Qi Meng and Rongchan Zhu and Yue Wang and Wenlei Shi and Shihua Zhang and Zhi-Ming Ma and Tie-Yan Liu},
      year={2023},
      eprint={2302.05104},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

If you have any questions, please feel free to contact me via: rayzhang@amss.ac.cn