Learning Specialized Activation Functions for Physics-Informed Neural Networks

Learning Specialized Activation Functions for Physics-Informed Neural Networks

Year:    2023

Author:    Honghui Wang, Lu Lu, Shiji Song, Gao Huang

Communications in Computational Physics, Vol. 34 (2023), Iss. 4 : pp. 869–906

Abstract

Physics-informed neural networks (PINNs) are known to suffer from optimization difficulty. In this work, we reveal the connection between the optimization difficulty of PINNs and activation functions. Specifically, we show that PINNs exhibit high sensitivity to activation functions when solving PDEs with distinct properties. Existing works usually choose activation functions by inefficient trial-and-error. To avoid the inefficient manual selection and to alleviate the optimization difficulty of PINNs, we introduce adaptive activation functions to search for the optimal function when solving different problems. We compare different adaptive activation functions and discuss their limitations in the context of PINNs. Furthermore, we propose to tailor the idea of learning combinations of candidate activation functions to the PINNs optimization, which has a higher requirement for the smoothness and diversity on learned functions. This is achieved by removing activation functions which cannot provide higher-order derivatives from the candidate set and incorporating elementary functions with different properties according to our prior knowledge about the PDE at hand. We further enhance the search space with adaptive slopes. The proposed adaptive activation function can be used to solve different PDE systems in an interpretable way. Its effectiveness is demonstrated on a series of benchmarks. Code is available at https://github.com/LeapLabTHU/AdaAFforPINNs.

You do not have full access to this article.

Already a Subscriber? Sign in as an individual or via your institution

Journal Article Details

Publisher Name:    Global Science Press

Language:    English

DOI:    https://doi.org/10.4208/cicp.OA-2023-0058

Communications in Computational Physics, Vol. 34 (2023), Iss. 4 : pp. 869–906

Published online:    2023-01

AMS Subject Headings:    Global Science Press

Copyright:    COPYRIGHT: © Global Science Press

Pages:    38

Keywords:    Partial differential equations deep learning adaptive activation functions physics-informed neural networks.

Author Details

Honghui Wang

Lu Lu

Shiji Song

Gao Huang