PROSE: Predicting Multiple Operators and Symbolic Expressions

This repository contains code for the following papers:

• PROSE: Predicting Multiple Operators and Symbolic Expressions using Multimodal Transformers. More details can be found in prose_ode/README.md.

• Towards a Foundation Model for Partial Differential Equations: Multi-Operator Learning and Extrapolation. More details can be found in prose_pde/README.md.

Install dependencies

Using conda and the env.yml file:

conda env create --name prose --file=env.yml

Citation

If you find our paper and code useful, please consider citing:

@article{liu2023prose,
  title={PROSE: Predicting operators and symbolic expressions using multimodal transformers},
  author={Liu, Yuxuan and Zhang, Zecheng and Schaeffer, Hayden},
  journal={arXiv preprint arXiv:2309.16816},
  year={2023}
}

@article{sun2024foundation,
  title={Towards a Foundation Model for Partial Differential Equations: Multi-Operator Learning and Extrapolation}, 
  author={Sun, Jingmin and Liu, Yuxuan and Zhang, Zecheng and Schaeffer, Hayden},
  journal={arXiv preprint arXiv:2404.12355},
  year={2024}
}