DeepXDE is a deep learning library on top of TensorFlow. Use DeepXDE if you need a deep learning library that
- solves forward and inverse partial differential equations (PDEs) via physics-informed neural network (PINN),
- solves forward and inverse integro-differential equations (IDEs) via PINN,
- solves forward and inverse fractional partial differential equations (fPDEs) via fractional PINN (fPINN),
- approximates functions from multi-fidelity data via multi-fidelity NN (MFNN),
- approximates nonlinear operators via deep operator network (DeepONet),
- approximates functions from a dataset with/without constraints.
Documentation: ReadTheDocs, Extended abstract, Short paper, Full paper, Slides, Video
Papers
-
Algorithms & examples
- Solving PDEs and IDEs via PINN: Extended abstract, Short paper, Full paper
- Solving fPDEs via fPINN: SIAM J. Sci. Comput.
- Solving stochastic PDEs via NN-arbitrary polynomial chaos (NN-aPC): J. Comput. Phys.
- Learning from multi-fidelity data via MFNN: PNAS, J. Comput. Phys.
- Learning nonlinear operators via DeepONet: arXiv
-
Applications
- Inverse problems in nano-optics and metamaterials: Opt. Express
DeepXDE supports
- complex domain geometries without tyranny mesh generation. The primitive geometries are interval, triangle, rectangle, polygon, disk, cuboid, and sphere. Other geometries can be constructed as constructive solid geometry (CSG) using three boolean operations: union, difference, and intersection;
- multi-physics, i.e., coupled PDEs;
- 5 types of boundary conditions (BCs): Dirichlet, Neumann, Robin, periodic, and a general BC;
- time-dependent PDEs are solved as easily as time-independent ones by only adding initial conditions;
- residual-based adaptive refinement (RAR);
- uncertainty quantification using dropout;
- two types of neural networks: fully connected neural network, and residual neural network;
- many different losses, metrics, optimizers, learning rate schedules, initializations, regularizations, etc.;
- useful techniques, such as dropout and batch normalization;
- callbacks to monitor the internal states and statistics of the model during training;
- enables the user code to be compact, resembling closely the mathematical formulation.
All the components of DeepXDE are loosely coupled, and thus DeepXDE is well-structured and highly configurable. It is easy to customize DeepXDE to meet new demands.
DeepXDE requires TensorFlow to be installed. Both TensorFlow 1 and TensorFlow 2 can be used as the DeepXDE backend, but TensorFlow 1 is recommended:
- In my tests TensorFlow 2 is 2x~3x slower than TensorFlow 1;
- Currently L-BFGS optimizer is not supported in DeepXDE yet when using TensorFlow 2.
Then, you can install DeepXDE itself. If you use TensorFlow 2, you need to install DeepXDE by cloning the folder. If you use Python 2, you need to install DeepXDE using pip.
- Install the stable version with
pip:
$ pip install deepxde
- Install the stable version with
conda:
$ conda install -c conda-forge deepxde
- For developers, you should clone the folder to your local machine and put it along with your project scripts.
$ git clone https://github.com/lululxvi/deepxde.git
-
Dependencies
- Matplotlib
- NumPy
- SALib
- scikit-learn
- SciPy
- TensorFlow>=1.14.0
If you use DeepXDE for academic research, you are encouraged to cite the following paper:
@article{lu2019deepxde,
author = {Lu, Lu and Meng, Xuhui and Mao, Zhiping and Karniadakis, George E.},
title = {{DeepXDE}: A deep learning library for solving differential equations},
journal = {arXiv preprint arXiv:1907.04502},
year = {2019}
}
First off, thanks for taking the time to contribute!
- Reporting bugs. To report a bug, simply open an issue in the GitHub "Issues" section.
- Suggesting enhancements. To submit an enhancement suggestion for DeepXDE, including completely new features and minor improvements to existing functionality, let us know by opening an issue.
- Pull requests. If you made improvements to DeepXDE, fixed a bug, or had a new example, feel free to send us a pull-request.
- Questions. To get help on how to use DeepXDE or its functionalities, you can as well open an issue.