1: \begin{abstract}

2: This paper proposes a plane wave activation based neural network (PWNN) for solving Helmholtz equation, the basic partial differential equation to represent wave propagation, e.g. acoustic wave, electromagnetic wave, and seismic wave.  

3: Unlike using traditional activation based neural network (TANN) or $sin$ activation based neural network (SIREN) for solving general partial differential equations, 

4: we instead introduce a complex activation function $e^{\mathbf{i} {x}}$, the plane wave which is the basic component of the solution of Helmholtz equation. 

5: By a simple derivation, we further find that PWNN is actually a generalization of the plane wave partition of unity method (PWPUM) by additionally imposing a learned basis with both amplitude and direction to better characterize the potential solution. 

6: We firstly investigate our performance on a problem with the solution is an integral of the plane waves with all known directions. 

7: The experiments demonstrate that: PWNN works much better than TANN and SIREN on varying architectures or the number of training samples, 

8: that means the plane wave activation indeed helps to enhance the representation ability of neural network toward the solution of Helmholtz equation; 

9: PWNN has competitive performance than PWPUM, e.g. the same convergence order but less relative error. 

10: Furthermore, we focus a more practical problem, the solution of which only integrate the plane waves with some unknown directions. 

11: We find that PWNN works much better than PWPUM at this case. 

12: Unlike using the plane wave basis with fixed directions in PWPUM, PWNN can learn a group of optimized plane wave basis which can better predict the unknown directions of the solution. 

13: The proposed approach may provide some new insights in the aspect of applying deep learning in Helmholtz equation.

14: \end{abstract}

15: