SciANN: A Keras/TensorFlow wrapper for scientific computations and physics-informed deep learning using artificial neural networks

被引:250
作者
Haghighat, Ehsan [1 ]
Juanes, Ruben [1 ]
机构
[1] MIT, 77 Massachusetts Ave, Cambridge, MA 02139 USA
关键词
SciANN; Deep neural networks; Scientific computations; PINN; vPINN;
D O I
10.1016/j.cma.2020.113552
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
In this paper, we introduce SciANN, a Python package for scientific computing and physics-informed deep learning using artificial neural networks. SciANN uses the widely used deep-learning packages TensorFlow and Keras to build deep neural networks and optimization models, thus inheriting many of Keras's functionalities, such as batch optimization and model reuse for transfer learning. SciANN is designed to abstract neural network construction for scientific computations and solution and discovery of partial differential equations (PDE) using the physics-informed neural networks (PINN) architecture, therefore providing the flexibility to set up complex functional forms. We illustrate, in a series of examples, how the framework can be used for curve fitting on discrete data, and for solution and discovery of PDEs in strong and weak forms. We summarize the features currently available in SciANN, and also outline ongoing and future developments. (C) 2020 Elsevier B.V. All rights reserved.
引用
收藏
页数:17
相关论文
共 38 条
[1]  
Abadi M, 2016, PROCEEDINGS OF OSDI'16: 12TH USENIX SYMPOSIUM ON OPERATING SYSTEMS DESIGN AND IMPLEMENTATION, P265
[2]  
[Anonymous], 2016, P NIPS
[3]  
[Anonymous], ARXIV151201274
[4]  
Baydin AG, 2018, J MACH LEARN RES, V18
[5]   A unified deep artificial neural network approach to partial differential equations in complex geometries [J].
Berg, Jens ;
Nystrom, Kaj .
NEUROCOMPUTING, 2018, 317 :28-41
[6]   Machine learning for data-driven discovery in solid Earth geoscience [J].
Bergen, Karianne J. ;
Johnson, Paul A. ;
de Hoop, Maarten V. ;
Beroza, Gregory C. .
SCIENCE, 2019, 363 (6433) :1299-+
[7]  
Bergstra J., 2010, P PYTH SCI COMP C SC, V4
[8]  
Bishop C. M., 2006, Pattern recognition and machine learning
[9]   Perspective on machine learning for advancing fluid mechanics [J].
Brenner, M. P. ;
Eldredge, J. D. ;
Freund, J. B. .
PHYSICAL REVIEW FLUIDS, 2019, 4 (10)
[10]   Machine Learning for Fluid Mechanics [J].
Brunton, Steven L. ;
Noack, Bernd R. ;
Koumoutsakos, Petros .
ANNUAL REVIEW OF FLUID MECHANICS, VOL 52, 2020, 52 :477-508