This repository contains the code for the paper
- Monte Carlo Neural PDE Solver for Learning PDEs via Probabilistic Representation (IEEE TPAMI 2025)
To generate the data-set, use,
python generate_data.pyTo train MCNP, use
python main.pyIf 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