2 HUIJUAN ZHOU, JUNCAI PU, AND YONG CHEN
physical applications [18]. Afterwards, global adaptive activation functions and locally adaptive activation functions
are proposed to approximate smooth and discontinuous functions as well as solutions of linear and nonlinear partial
differential equations by introducing a scalable parameters in the activation function and adding a slope recovery term
based on activation slope to the loss function of locally adaptive activation functions. It demonstrates the locally
adaptive activation functions further improve the training speed, performance and speed up the training process of
NNs. [19, 20].
Due to the good properties of integrable systems, it is possible to solve the data-drive forward and inverse problems
for abundant classical integrable nonlinear evolution equations via the PINN method. Since Chen’s team first proposed
the concept of integrable deep learning in 2020, a large number related literature has been published. In particular, Li
and Chen construct abundant numerical solutions of second-order and third-order nonlinear integrable equations with
different initial and boundary conditions by deep learning method based on the PINN model [21, 22]. Pu, Peng et al.
recover the solitons, breathers, rogue wave solutions and rogue wave on the periodic background of the nonlinear partial
differential equation with the aid of the PINN model [23,24]. Miao and Chen use the PINN method to high-dimensional
system to solve the (N+1)-dimensional initial boundary value problem with 2N+1 hyperplane boundaries [25]. Lin
and Chen devise a two-stage PINN method which is tailored to the nature of equations by introducing features of
physical systems into NNs based on conserved quantities [26]. In addition, many important work on data-driven
solutions and parameter discovery of different nonlinear systems have been doned by other scholars [27–32].
In the previous literature, many soliton solutions of constant coefficients integrable systems have been accurately
simulated numerically. The soliton solutions of variable coefficients integrable systems studied in this paper are quite
different from the soliton solutions of constant coefficients equations. From the perspective of the solution dynamics,
the center trajectory equation of the one-soliton solution of the constant coefficient equation is often a straight line,
while the one-soliton solution of the variable coefficient equation presents a richer and more complex shape. For
example, in this paper, we learned the case of the center trajectory equations of the one-solitons are parabola, ”S-
shape” and cosine wave type. For the two-soliton solution of variable coefficients equation, it can take on a more
complex shape. To our knowledge, there is rare study about the data-driven soliton solutions, parameter and function
discovery for the variable coefficient equation by the IPINN method. To fill this gap, the main purpose of this paper is
to recover various solitons, high-order soliton, as well as parameters and functions discovery for the variable coefficient
equation when only know the initial-boundary value conditions. It is note that we use PINN algorithm with sub-neural
networks to achieve data-driven function discovery of variable coefficient equations.
The rest of this paper is built up as follows. In Section 2, the IPINN method for the VC-Hirota equation is
introduced. To provide a direct and precise description of the IPINN method, the algorithm flow schematic and
algorithm steps are given. Section 3 provides the data-driven multi-soliton and high-order soliton solutions of the VC-
Hirota equation using the IPINN approach, and related plots and dynamic analyses are revealed in detail. Section 4
presents experimental results of learning data-driven parameter discovery of VC-Hirota system. Furthermore, we learn
the unknown function of the VC-Hirota via a PINN algorithm with sub-neural networks successfully. The conclusions
and discussions are given in Section 5.
2. The improved physics-informed neural networks method for the variable coefficient Hirota
equation
An improved PINN (IPINN) approach with neuron-wise locally adaptive activation function was presented to derive
data-driven localized waves and learn unknown parameters of integrable systems in complex space, and numerical
results demonstrated the improved approach has faster convergence and better simulation effect than the classical
PINN method [33,34]. In the following, we will introduce the IPINN method in detail and extend it to the application
of variable coefficient Hirota equations.
Considering the (1+1)-dimensional nonlinear time-dependent systems with unknown constant parameter δand
variable coefficients α1(z), α2(z) and α3(z) in complex space as below.
qz+N[q;δ, α1(z), α2(z), α3(z)] = 0,(2.1)
where N[·;δ, α1(z), α2(z), α3(z)] is nonlinear differential operators in space. In order to simplify the structure of the
complex-valued solutions q(t, z) in Eq. (2.1), we decompose q(t, z) into q(t, z) = u(t, z) + iv(t, z), where u(t, z) and
v(t, z) are real-valued functions. Then substituting q(t, z) = u(t, z) + iv(t, z) into Eq. (2.1), and letting the real and
imaginary parts be equal to 0, we have
uz+Nu[u, v;δ, α1(z), α2(z), α3(z)] = 0,
vz+Nv[u, v;δ, α1(z), α2(z), α3(z)] = 0,(2.2)
where the Nuand Nvare nonlinear differential operators in space. Defining the physics-informed part fu(x, t) and
fv(x, t) as
fu:= uz+Nu[u, v;δ, α1(z), α2(z), α3(z)],
fv:= vz+Nv[u, v;δ, α1(z), α2(z), α3(z)],(2.3)
which play a role of regularization.