Abstract
In this paper, we introduce a new deep learning framework for discovering the phase-field models from existing image data. The new framework embraces the approximation power of physics informed neural networks (PINNs) and the computational efficiency of the pseudo-spectral methods, which we named pseudo-spectral PINN or SPINN. Unlike the baseline PINN, the pseudo-spectral PINN has several advantages. First of all, it requires less training data. A minimum of two temporal snapshots with uniform spatial resolution would be adequate. Secondly, it is computationally efficient, as the pseudo-spectral method is used for spatial discretization. Thirdly, it requires less trainable parameters compared with the baseline PINN, which significantly simplifies the training process and potentially assures fewer local minima or saddle points. We illustrate the effectiveness of pseudo-spectral PINN through several numerical examples. The newly proposed pseudo-spectral PINN is rather general, and it can be readily applied to discover other PDE-based models from image data.
Similar content being viewed by others
References
Berg, J., Nystrom, K.: A unified deep artificial neural network approach to partial differential equations in complex geometries. Neurocomputing 317, 28–41 (2018)
Brunton, S.L., Proctor, J.L., Kutz, J.N.: Discovering governing equations from data by sparse identification of nonlinear dynamical systems. Proc. Natl. Acad. Sci. 113(15), 3932–3937 (2016)
Cahn, J.W., Hilliard, J.E.: Free energy of a nonuniform system. I. Interfacial free energy. J. Chem. Phys. 28, 258–267 (1958)
Chen, L., Zhao, J., Gong, Y.: A novel second-order scheme for the molecular beam epitaxy model with slope selection. Commun. Comput. Phys. 4(25), 1024–1044 (2019)
E, Weinan., Yu, B.: The deep Ritz method: a deep learning-based numerical algorithm for solving variational problems. Commun. Math. Stat. 6, 1–12 (2018)
Guillen-Gonzalez, F., Tierra, G.: Second order schemes and time-step adaptivity for Allen-Cahn and Cahn-Hilliard models. Comput. Math. Appl. 68(8), 821–846 (2014)
Han, D., Wang, X.: A second order in time uniquely solvable unconditionally stable numerical schemes for Cahn-Hilliard-Navier-Stokes equation. J. Comput. Phys. 290(1), 139–156 (2015)
Han, J., Jentzen, A., E, Weinan.: Solving high-dimensional partial differential equations using deep learning. Proc. Natl. Acad. Sci. 115(34), 8505–8510 (2018)
Higham, C., Higham, D.: Deep learning: an introduction for applied mathematicians. SIAM Rev. 61(4), 860–891 (2019)
Li, B., Tang, S., Yu, H.: Better approximations of high dimensional smooth functions by deep neural networks with rectified power units. Commun. Comput. Phys. 27, 379–411 (2020)
Long, Z., Lu, Y., Ma, X., Dong, B.: PDE-net: learning PDEs from data. Proceedings of the 35th International Conference on Machine Learning 80, 3208–3216 (2018)
Lu, L., Meng, X., Mao, Z., Karniadakis, G.E.: DeepXDE: a deep learning library for solving differential equations. arXiv: 1907.04502 (2019)
Qin, T., Wu, K., Xiu, D.: Data driven governing equations approximation using deep neural networks. J. Comput. Phys. 395, 620–635 (2019)
Raissi, M.: Deep hidden physics models: deep learning of nonlinear partial differential equations. J. Mach. Learn. Res. 19, 1–24 (2018)
Raissi, M., Perdikaris, P., Karniadakis, G.E.: Physics-informed neural networks: a deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations. J. Comput. Phys. 378, 686–707 (2019)
Rudy, S., Brunton, S.L., Proctor, J.L., Kutz, J.N.: Data-driven discovery of partial differential equations. Sci. Adv. 3(4), e1602614 (2017)
Shen, J., Yang, X.: Numerical approximations of Allen-Cahn and Cahn-Hilliard equations. Discrete and Continuous Dynamical Systems 28(4), 1669–1691 (2010)
Wang, C., Wang, X., Wise, S.: Unconditionally stable schemes for equations of thin film epitaxy. Discrete and Continuous Dynamical Systems 28(1), 405–423 (2010)
Wang, C., Wise, S.: An energy stable and convergent finite-difference scheme for the modified phase field crystal equation. SIAM J. Numer. Anal. 49(3), 945–969 (2011)
Wang, Y., Lin, C.: Runge-Kutta neural network for identification of dynamical systems in high accuracy. IEEE Trans. Neural Netw. 9(2), 294–307 (1998)
Wight, C.L., Zhao, J.: Solving Allen-Cahn and Cahn-Hilliard equations using the adaptive physics informed neural networks. Commun. Comput. Phys. (2021)
Xu, K., Xiu, D.: Data-driven deep learning of partial differential equations in modal space. J. Comput. Phys. 408, 109307 (2020)
Yang, X., Zhao, J., Wang, Q.: Numerical approximations for the molecular beam epitaxial growth model based on the invariant energy quadratization method. J. Comput. Phys. 333, 102–127 (2017)
Zhao, J., Mau, J.: Discovery of governing equations with recursive deep neural networks. arXiv: 2009.11500 (2020)
Acknowledgements
Jia Zhao would like to acknowledge the support from NSF DMS-1816783 and NVIDIA Corporation for their donation of a Quadro P6000 GPU for conducting some of the numerical simulations in this paper.
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Conflict of interest
On behalf of all authors, the corresponding author states that there is no conflict of interest.
Rights and permissions
About this article
Cite this article
Zhao, J. Discovering Phase Field Models from Image Data with the Pseudo-Spectral Physics Informed Neural Networks. Commun. Appl. Math. Comput. 3, 357–369 (2021). https://doi.org/10.1007/s42967-020-00105-2
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s42967-020-00105-2