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 (IEEE TPAMI 2025)

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:

@ARTICLE{10916840,
  author={Zhang, Rui and Meng, Qi and Zhu, Rongchan and Wang, Yue and Shi, Wenlei and Zhang, Shihua and Ma, Zhi-Ming and Liu, Tie-Yan},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence}, 
  title={Monte Carlo Neural PDE Solver for Learning PDEs Via Probabilistic Representation}, 
  year={2025},
  volume={},
  number={},
  pages={1-18},
  doi={10.1109/TPAMI.2025.3548673}}

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