Abstract
Introduction:
The moment quantities associated with the nonlinear Schrödinger equation offer important insights into the evolution dynamics of such dispersive wave partial differential equation (PDE) models. The effective dynamics of the moment quantities are amenable to both analytical and numerical treatments.
Methods:
In this paper, we present a data-driven approach associated with the “Sparse Identification of Nonlinear Dynamics” (SINDy) to capture the evolution behaviors of such moment quantities numerically.
Results and Discussion:
Our method is applied first to some well-known closed systems of ordinary differential equations (ODEs) which describe the evolution dynamics of relevant moment quantities. Our examples are, progressively, of increasing complexity and our findings explore different choices within the SINDy library. We also consider the potential discovery of coordinate transformations that lead to moment system closure. Finally, we extend considerations to settings where a closed analytical form of the moment dynamics is not available.
1 Introduction and motivation
The study of Nonlinear Schrödinger (NLS) type models (Sulem and Sulem, 1999; ) is of wide interest and significance in a diverse array of physical modeling settings (; ). Relevant areas of application extend from atomic physics (Pitaevskii and Stringari, 2003; Pethick and Smith, 2002; Kevrekidis et al., 2015) to fluid mechanical and hydrodynamic (notably ones stemming from deep water waves) problems (; Infeld and Rowlands, 2000) and from plasma physics (Kono and Skorić, 2010; Infeld and Rowlands, 2000) to nonlinear optics (Kivshar and Agrawal, 2003; Hasegawa and Kodama, 1995). Indeed, the relevant model is a prototypical envelope wave equation that describes the dynamics of dispersive waves. Specifically, in the context of nonlinear optics, it describes the envelope of the electric field of light in optical fibers (as well as waveguides), with the relevant measurable quantity being the light intensity proportional to the square modulus of the complex field . Generalizations of relevant optical applications, involving multiple polarizations or frequencies of light have also been widely considered, in both spatially homogeneous and spatially heterogeneous media (Kivshar and Agrawal, 2003; Hasegawa and Kodama, 1995).
It is worthwhile to note that even to this day, the subject of NLS-type models continues to fascinate researchers and to constitute a fertile platform for a wide range of physical, mathematical and computational works; see, e.g., Karjanto (2024) for a recent review. At the same time, over the past few years, there has been an explosion of interest in data-driven methods, whereby machine-learning techniques are brought to bear towards understanding, codifying and deducing the fundamental quantities of physical systems. Arguably, a turning point in this effort was the development of the so-called physics-informed neural networks (PINNs) by Raissi et al. (2019) and of similar methodologies such as the extension of PINNs via the so-called DeepXDE Lu et al. (2021), as well as the parallel track of sparse identification of nonlinear systems, so-called SINDy by , which is central to the considerations herein. Additional methods include, but are not limited to, the sparse optimization of Schaeffer (2017), meta-learning of Feliu-Faba et al. (2020), as well as the neural operators of Li et al. (2021). A review of relevant model identification techniques can be found, e.g., in Karniadakis et al. (2021). Notice that a parallel track to the above one seeks not to discover the models, but rather key features thereof such as conserved quantities (Liu and Tegmark, 2021; 2022; Liu et al., 2022; Zhu et al., 2023) and its potential integrability (Krippendorf et al., 2021; De Koster and Wahls, 2024).
In the present setting, we seek to combine this important class of dispersive wave models within optical (and other physical) applications with some of the above machine-learning toolboxes. Our aim is not to discover the full PDE, or its conservation laws/integrability as in some of the above works. Rather, our aim is to leverage the theoretical understanding that exists at the level of reduced-order ODEs in the form of moment methods (Pérez-García et al., 2007; García-Ripoll and Pérez-García, 1999). Indeed, it is well-known from these works that upon defining suitable moment quantities, one can obtain closed form ODE dynamical systems of a few degrees-of-freedom, often just two (lending themselves to dynamical systems analysis) or sometimes involving a few more but still offering valuable low-dimensional analytical insights on the evolution of the center of mass, variance, kurtosis etc. of the relevant distribution. It is those effective ODEs (that stem from the original PDE via the moments) that we aim to retrieve using the data-driven approach developed herein. It is also relevant to mention in passing here that the moment methods were also used successfully to other models such as Fisher-KPP equations, considering also applications to the dynamics, e.g., of brain tumors (). Our approach and presentation will be structured hereafter as follows. In section 2, we will present a “refresher” from a theoretical perspective of the method of moments, essentially revisiting some basic results from the work of Pérez-García et al. (2007); García-Ripoll and Pérez-García (1999). Then, in Section 3, we will give a brief overview of SINDy type methods and the types of choices (such as, e.g., of model libraries) that they necessitate. Additionally, we will introduce a data-driven approach for learning coordinate transformations to close moment systems when the initially chosen moments are not closed. Then, in Section 4, we present a palette of numerical results and their effective moment identification. Our narrative contains a gradation of examples from simpler ones (where, e.g., an analytical low-dimensional closure of moments may exist) to gradually more complex ones, where a closure may exist after a coordinate transformation and eventually to cases where a closed system does not exist at the moment level to the best of our knowledge. Our aim is to showcase not only the successes, but also the challenges that the method may encounter in cases where we do not know of a closure or when we may not rightfully choose the library of functions (even when a closure may exist). We hope that this will provide a more informed/balanced viewpoint to the reader about what these methods may (and what they may not) be expected to provide.
2 Moment equation theoretical background
To contextualize our perspective, we will focus on the following specific case of the (1 + 1)-dimensional nonlinear Schrödinger (NLS) equation with a harmonic potential, (Kevrekidis et al., 2015; Kivshar and Agrawal, 2003),where denotes the nonlinearity. This model is not only of relevance to optics (where the harmonic potential represents the heterogeneous profile of the refractive index) (Kivshar and Agrawal, 2003), but also to atomic Bose-Einstein condensates, where this parabolic confinement is a typical byproduct of magnetic traps (Pitaevskii and Stringari, 2003; Kevrekidis et al., 2015). We will consider the initial value problem of Equation 1 with localized and sufficiently regular initial conditions (ICs) .
2.1 The method of moments
Instead of fully characterizing the solution of the Cauchy problem of Equation 1, the method of moments (Pérez-García et al., 2007), seeks to provide qualitative description of the solution behavior by studying the evolution of several integral quantities, i.e., the moments, of the solution . This approach enables a reduced-order description of the NLS equation by transforming it into a system of (potentially) closed ordinary differential equations (ODEs). More specifically, according to Pérez-García et al. (2007), we define for the moment quantities of solution as follows,where is the complex conjugate of , , and is a function such that . Moments Equations 2–5 of the solution have intuitive physical meanings; for example, the first moment is associated with the center of mass as described by the (unnormalized) probability density . Higher moments are also associated with this density distribution (i.e., its variance etc.). The quantities are the respective ones associated with the momentum density (which is the quantity in the corresponding parenthesis in the right hand side of the definition. stems from (thinking quantum-mechanically) the kinetic part of the Schrödinger problem energy, while represents the nonlinear part of the corresponding energy. We assume that the IC is regular enough to ensure that all moments are well-defined for .
The method of moments aims to extract qualitative information about the solution of the PDE Equation 1 by deriving a closed set of evolution ODEs for the moments of . Depending on the nonlinearity , these ODEs can sometimes be determined analytically as is shown in the work of Pérez-García et al. (2007); García-Ripoll and Pérez-García (1999). Below, we provide a few examples.
Example 1The moments and satisfyThis indicates that the evolution of the center of mass behaves as a harmonic oscillator, independent of the nonlinearity . More generally, for a parabolic confinement of frequency , this would be reflected in the associated frequency of moment oscillations; this is the so-called dipolar motion (Pitaevskii and Stringari, 2003; Kevrekidis et al., 2015).
Example 2If , i.e., for the linear case of Equation 1, the set of moments are closed under
Example 3Assume the nonlinearity is time-independent and given by , where is a constant. Although the evolution of the moments , , and is not closed, it becomes closed under the coordinate transformation , i.e.,
While the examples and conditions under which the method of moments leads to closed equations are well-known, deriving such analytical closure systems requires knowledge of the underlying PDE system, Equation 1 and detailed calculations therewith. This work explores data-driven methods for obtaining analytical or approximate moment closure systems based on empirical observations or simulations of the NLS equation, rather than relying solely on such analytical understanding and derivations. Importantly, the reconstruction of these ODE models can, in principle, take place even for settings where the underlying PDE model is unavailable/has not been specified. Given “experimental” data for the field, one may aspire to utilize the toolboxes presented below in order to obtain these effective, reduced dynamical equations.
For systems with existing analytical closures of the moment equations, such as Examples 1–3, our method seeks to rediscover the moment evolution equations and potentially the necessary coordinate transformations, such as in Example 3, in a model-agnostic and data-driven manner. For systems lacking analytical closed moment equations, we seek to derive approximate moment closure equations, providing a principled and reduced-order description of the original PDE, capable of predicting the future evolution of the system. The relevant details will be explained in the following section.
3 Data-driven methods
We present two computational methodologies for finding analytical or approximate closures for moment equations.
3.1 Sparse Identification of Nonlinear Dynamics
Our first method leverages Sparse Identification of Nonlinear Dynamics (SINDy) by , a data-driven approach for discovering governing ODEs from simulated or observational data. Consider a nonlinear ODE system of the form:where represents the state evolution over time, governed by the dynamic constraint . SINDy aims to identify the unknown dynamics, from a time series of .
The key assumption is that has a “simple” form and can be expressed or approximated as a linear combination of only a few terms from a suitably chosen library, . For example, a monomial library of degree up to two, , is:with . In particular, e.g., the right-hand sides of Equations 6, 7 can all be written as sparse linear combinations of terms from . Given a dictionary of elements— is typically larger than —the sparsity assumption implies the existence of a sparse matrix such thatwhere each sparse column indicates which nonlinear functions among the library are used to parsimoniously represent .
To determine this sparse , SINDy employs sparse regressions on the data. Specifically, given a time series of the state at times — is generally much larger than , the library size—one can assemble the state matrix and the derivative matrix:where can be estimated by, e.g., finite differences on . Define the library matrix as
Evaluating Equations 8, 9 at all times yields
Sparse regression techniques, such as LASSO (Tibshirani, 1996) or sequential thresholded least-squares (), can then solve the overdetermined system (Equation 11) for the sparse . This provides an approximation to the governing equation as in Equation 10.
While SINDy can be directly applied to
Examples 1–3to discover closed moment systems based on simulated PDE data (
Equation 1), our specific interest lies in the following:
1. Given time series data of moments with a known closed moment system (e.g., Examples (1) and (2)), can we robustly discover the governing dynamics using a correct but potentially oversized dictionary , such as polynomials up to a high degree?
2. If a system is not closed for a chosen set of moments, but closure exists after a proper coordinate transformation (e.g., selecting in Ex. 3), what insights can we gain by directly applying SINDy to time series data from this “incorrect” set of moments?
3. For more general systems where analytical closure does not exist, can data-driven methods provide a “good enough” numerical approximation to predict the future evolution of the moment system, effectively serving as a principled reduced-order description of the underlying PDE?
These questions will be addressed in Section 4. Of particular interest is the second point, where we demonstrate that, in certain cases, we can gain insight into the appropriate transformation to close the system, even if the moment system is not closed under the originally selected variables. In the following section, we discuss a more principled strategy to discover such transformations in a data-driven fashion.
3.2 Data-driven discovery of coordinate transformations for moment system closure
To illustrate the idea, we focus on Example 3, where the initially selected moments are , and an analytical closure exists only after a coordinate transformation, ,
Remark 1Note that the transformation matrix is not unique. For any full-rank matrix , the moment system remains closed under the transformation , with .
Assume we aim to discover purely from simulated PDE data. We propose the following strategy: Let be the state matrix for the original coordinate , sampled at , as defined by Equation 8. Let be the new coordinate, and the associated new state matrix becomes
We can then solve for through the following optimization.where is the derivative matrix, Equation 8, associated with the new state matrix , and are the Frobenius norm and norm, respectively, defined for any matrix as follows:
In addition, is a non-negative weight, and is the library matrix, Equation 8, of a chosen library on the new coordinate .
The idea behind Equation 12 is very simple: we search for the transformation matrix such that the dynamics under the new coordinate can be parsimoniously represented from the library . The weight controls the sparsity-promoting -regularization, and the constraint prevents the trivial solution . The set of satisfying the constraint is called the Stiefel manifold (Edelman et al., 1998; ).
Remark 2When the weight , even with the Stiefel manifold constraint , the solution to Equation 12 is not unique. Just like Remark 1, if is a solution, then for any orthogonal matrix , the pair is also a solution, where
To solve Equation 12, we use alternating optimization, iteratively optimizing and while keeping the other fixed. Specifically, when is fixed, solving for reduces to a LASSO problem. Conversely, when is fixed, the problem becomes a Stiefel manifold optimization with a smooth objective function, which can be efficiently solved using methods such as those presented in (Oviedo and Dalmau, 2019; Xiao et al., 2020; Liu et al., 2021). To ensure the algorithm remains unbiased, we implement an annealing strategy for the hyperparameter . Initially, is kept constant for Iter_scheduled iterations. Following this period, is reduced by half every iterations. Refer to Algorithm 1 for the pseudocode to solve problem Equation 12.
Algorithm 1

4 Numerical results
In this section, we present numerical results on data-driven closure of moment systems (Examples 1–3) using the methodologies described in Section 3. To obtain the data, we first numerically solve the PDE Equation 1 with periodic boundary conditions and various ICs using an extended 4th-order Runge-Kutta method, the exponential integrator (ETDRK4) (see (Kassam and Trefethen, 2005) for a detailed explanation). The time series of the moments is subsequently extracted by numerical spatial integration according to Equations 2–5, with spatial derivatives computed using pseudo-spectral Fourier methods. We note once again that should the numerical data be replaced by “experimental” ones from a given physical process, the procedure can still be applied. Unless otherwise noted, the time series are evaluated at uniformly spaced times , where and .
4.1 Examples with analytical moment closure
We first examine Examples (1) and (2), where analytical (linear) closure exists for the chosen moments (Example 1) and (Example 2).
4.2 Example 1
Let the selected moments be . We construct the data matrices and from numerically solving the PDE Equation 1 with the ICs:
We apply SINDy to this system using monomial libraries of degree up to , , as defined in Equation 9. We find that as long as , SINDy applied to either data matrix or discovers the dynamics nearly perfectly. However, as we gradually expand the library by increasing , SINDy applied to either or individually typically produces erroneous dynamics, regardless of how carefully the sparsity-promoting parameter is tuned. A typical negative result from SINDy applied to with is presented in the Supplementary Material. This outcome is expected, as increasing the library size makes the problem more ill-posed, leading SINDy to overfit the data and produce incorrect dynamics.
To address the overfitting issue, we can enlarge the dataset by concatenating the data matrices and vertically, forming . When applying SINDy to this new data matrix (considering boundary issues when taking finite differences), SINDy can now discover the correct dynamics that match Equation 6, even with a much larger library for up to 16. For example, when , the output ODE from SINDy readswhere the coefficients are rounded to three decimal places. This suggests the potential usefulness of concatenating different time series, especially in cases where one may not be familiar with the order of the relevant closure.
4.3 Example 2
For Example 2 with the selected moments , our findings are similar to those in Section 4.2. When applying SINDy to a data matrix from a single IC, SINDy discovers erroneous dynamics for the quadratic dictionary . However, using larger data matrices from multiple ICs, SINDy can once again accurately identify the correct dynamics even with the larger dictionaries. Representative negative and positive results are presented in the Supplementary Material, similar to the previous example.
4.4 Examples where closure exists after coordinate transformations
We now turn to Example 3, where the selected moments are , and the moment system only closes after a coordinate transformation. Specifically, we consider the nonlinearity in Equation 1, which satisfies the condition in Example 3. We collect the moment time series data by numerically solving the PDE with the following four distinct ICs, Equations 13–16:
We consider the following questions:
What insights can we gain by directly applying SINDy to this system with the selected moments , where a closure does not exist?
Can the method described in Section 3.2 correctly identify the coordinate transformation that closes the moment system?
We note that normalized moment time series are used as input for SINDy, as a way of incorporating feature scaling. This is an important aspect that ensures that all the relevant quantities are considered on “equal footing,” when the sparse regression step takes place.
4.5 SINDy with linear library
We begin by applying SINDy with a linear library to the moment time series data from IC Equation 14. The resulting equations are:
The coefficients are rounded to three decimal places. The first four panels of Figure 1 compare the “ground-truth” time-evolution data of obtained from PDE integration (red curves) and the definition of these quantities through Equations 2–5 with those from integrating the SINDy-predicted ODEs Equation 17 (blue curves). While the SINDy-predicted time evolution of closely matches the ground truth, there is a significant discrepancy for the moments in Figure 1. Accordingly, the SINDy-predicted dynamics is not accurate for the original coordinates .
FIGURE 1
Nevertheless, interestingly, if we add the ODE for with that for in Equation 17, which corresponds to the correct coordinate transformation for closure, we obtain:
Equation 18 closely matches the ground-truth moment system Equation 7. The last panel of Figure 1 shows that the predicted time evolution of aligns perfectly with the ground truth, even though SINDy does not accurately recover those of and individually. This finding is intriguing because, even when SINDy is applied to the original coordinates without closure, it “strategically chooses” to sacrifice accuracy for and but indirectly captures the correct dynamics when considering the proper coordinate transformation . Moreover, a similar pattern is consistently observed when applying SINDy to the moments simulated from the other three ICs. The results are shown in the Supplementary Material. We have also provided an analogous example for 2D NLS equations. For details, please refer to Section 3 of the Supplementary Material. Indeed, this clearly illustrates that the methodology presented herein is not in any fundamental way limited to 1D NLS models, but can be extended to higher dimensional ones, as may be deemed desirable.
4.6 SINDy with quadratic library
Next, we investigate the performance of SINDy with an expanded quadratic library applied to the moment time series with the same IC Equation 14. The predicted ODEs now become:
As before, if we add the predicted dynamics of and in Equation 19, we obtain:
Unlike the previous case with , the SINDy-predicted governing Equation 20 using the expanded library after the (theoretically motivated) coordinate transformation still fails to match the ground truth ODE Equation 7 and remains unclosed (i.e., the third equation in Equation 20 cannot be written using only , , and ). Specifically, the coefficient of in Equation 20 is , deviating substantially from that of in the ground truth. Although the coefficients for the additional terms (, , and ) are relatively small, their presence hinders the accurate recovery of the correct coefficient for .
Figure 2 compares the ground-truth time evolutions of obtained from PDE integration with their SINDy-predicted dynamics after integrating Equations 19, 20. Similar to Figure 1, SINDy with the quadratic library applied to the unclosed moments sacrifices the exact recovery of the time evolution of and , but indirectly captures the correct time evolution of . However, unlike the linear library case in Section 4.5, SINDy with the quadratic library can yield the correct time evolution of but fails to uncover the correct governing equation for [Equation 20] due to the expanded library size, leading to overfitting issues discussed above.
FIGURE 2
Here, we see a key complication within SINDy that has been also present in works such as () and especially (), namely the methodology is likely to result in ODE models that are proximal to the theoretically expected ones, but not identical to them. This, in turn, may result in a nontrivial error outside of the training set. Especially when the library at hand is “richer” than the terms expected to be present, unfortunately, it does not generically seem that the reduced, theoretically expected model will be discovered. Rather, our results suggest that it is possible that the additional “wealth” of the libraries used can be leveraged to approximate the data via different (nonlinear) dependent variable combinations.
As an even more problematic example, applying SINDy with a quadratic library to the moment time series generated from another IC Equation 16 results in a predicted ODE system that not only fails to match the ground truth, even after the theoretically suggested coordinate transformation , but also causes the ODE system to blow up in finite time. For a detailed discussion, we refer the interested reader to the Supplementary Material.
4.7 Stiefel optimization for discovering coordinate transformations
Next, we test the methodology from Section 3.2 to discover the coordinate transformation needed to close the moment system.
Case 1: . We first consider the case where the hyperparameter in Equation 12 is set to , i.e., we do not promote sparsity in the matrix . We use the linear library , and set the maximum number of iterations to in Algorithm 1. The algorithm produces the following outputs
The coefficients are rounded to three decimal places. On the other hand, the “ground-truth” coordinate transformation (after normalization to satisfy the Stiefel manifold constraint) and the corresponding dynamics according to Equation 7 are:
At first glance, the predicted solution Equation 21 appears different from the ground truth in Equation 22. However, after a further change of coordinates usingwe have
Hence, according to Remark 2, and are equivalent solutions, and our method successfully predicts the correct coordinate transformation to close the moment system. However, as expected, due to , the linear combination matrix of the transformed dictionary is not sparse.
Case 2: . We next explore the case when in Equation 12. We use the same linear library , and is initially set to . In Algorithm 1, we set , , and . The algorithm returns
Similarly, after a further change of coordinates using:we again have
Note that the result in Equation 23 for is much closer to the ground truth in Equation 22, compared to in Equation 21 with . This demonstrates that a positive with annealed optimization not only discovers the correct coordinate transformation to close the system but also achieves a sparser solution for , reducing the number of terms on the right-hand side of the moment ODE system.
4.8 An example of unclosed moment system
Finally, we present a case without analytical closure. Our “data-driven closure” aims to provide an accurate, reduced-order description of the PDE by approximating the evolution of the moment systems. In principle, this is the type of problem that we are aiming for, namely the discovery of potential moment closures when these may not be analytically available; the examples presented previously are valuable benchmarks to raise the complications that may emerge when one seeks to use this type of methodology in systems where the answer may be unknown and what credibility one may wish to assign to the obtained results.
Specifically, we consider an NLS Equation 1 with a time-dependent nonlinearity:
Notice that such time-dependent nonlinearities are well-known for some time in atomic physics settings (Donley et al., 2001; Staliunas et al., 2002) (and continue to yield novel insights to this day (Shagalov and Friedland, 2024)) and similar dynamical scenarios have been considered in nonlinear optics (; Zhang et al., 2021).
Moment systems with such nonlinearity will not close to the best of our knowledge, so we aim to numerically approximate form of the dynamics of the moments , where . We consider the following three ICs:
Notably, the IC Equation 27 includes a quadratic phase, motivated by the quadratic phase approximation (QPA) ansatz for NLS equations discussed by (Pérez-García et al., 2007). It also includes regular, smooth localized initial conditions, as well as one involving Fourier mode oscillations, modulated by the Gaussian term. We expect the SINDy-predicted dynamics to vary with the different ICs of Equations 25–27, hence the relevant choices.
Due to the periodic nature of the nonlinearity in Equation 24, we expect the moment system to be non-autonomous and exhibit an oscillatory pattern. To capture this, we introduce the following two “auxiliary moments”:
Naturally, one can observe that these are inspired by the nature of . However, one can envision the use of Fourier modes even if the mathematical model was not known or if the data stemmed from experimental observations.
We then apply SINDy with a linear library to the expanded moment systems . The time series data of the moments are collected by integrating the PDE up to . The exact SINDy-predicted ODEs for all three ICs are presented in the Supplementary Material.
To test the accuracy of the predicted moment systems, we integrate the SINDy-predicted ODEs into the future, up to . Figure 3 compares the ground-truth and SINDy-predicted time evolutions for the three ICs Equations 25–27. In all cases, the model closely matches the ground truth up to . Indeed, this is well past the training time of and thus provides a satisfactory reduced-order description of the underlying PDE effective moment dynamics. Thus, despite the potential shortcomings of the method which we tried to present in an unbiased fashion in case examples where the analytical theory helps assess them, we still find it to be a worthwhile tool to consider. I.e., from a data-driven perspective, it can be seen to potentially provide an effective, low-dimensional dynamical representation of the associated high-dimensional PDE dynamics.
FIGURE 3
5 Conclusion and future directions
This study explores a data-driven approach to identifying moment equations in nonlinear Schrödinger models. This paves the way more generally towards the use of similar methods in nonlinear PDEs which may feature similar wave phenomena. We applied the relevant sparse regression/optimization methodology aiming to rediscover known analytical closures (progressively extending considerations to more complex settings), addressed overfitting by augmenting datasets with multiple initial conditions, and identified suitable coordinate transformations for systems requiring them so as to bring forth the reduced or analytically tractable form of the dynamics. Additionally, we demonstrated that our approach could provide a reduced-order description of systems without analytical closures by approximating the evolution of the moment systems. Our findings show that this data-driven method can capture complex dynamics in NLS models and offer insights for various physical applications, possibly well past the training time used for the data-driven methods.
Future work will focus on extending our method to more complex PDEs and exploring its applicability to other types of nonlinear dynamical high-dimensional models, such as, e.g., the ones we mentioned in the context Fisher-KPP models and their applications to brain tumor dynamics, as analyzed, e.g., in (). Another possible avenue is to, instead of recovering the moment systems through numerical differentiation of the moment time series (as done in SINDy), leverage numerical integration into future time of a suitably augmented system. This approach, similar to Neural ODEs (Chen et al., 2018) and shooting methods, can help avoid producing predicted ODEs that blow up in finite time which SINDy may produce (see details in Section 4.2.2 and the Supplementary Material). Additionally, we plan to develop techniques to identify nonlinear coordinate transformations that can close the moment system, further enhancing the applicability of our method. Lastly, one can envision such classes of techniques for obtaining additional reduced features of solitary waves, such as data-driven variants of the variational approximation (Malomed, 2002), or data-driven models of soliton interaction dynamics (Manton, 1979; Kevrekidis et al., 2004; Ma et al., 2016).
Statements
Data availability statement
The original contributions presented in the study are included in the article/Supplementary Material, further inquiries can be directed to the corresponding author.
Author contributions
SY: Formal Analysis, Investigation, Methodology, Software, Writing–original draft. SC: Formal Analysis, Investigation, Methodology, Software, Writing–original draft. WZ: Conceptualization, Funding acquisition, Methodology, Writing–review and editing. PK: Conceptualization, Funding acquisition, Methodology, Supervision, Writing–review and editing.
Funding
The author(s) declare that financial support was received for the research, authorship, and/or publication of this article. This material is based upon work supported by the U.S. National Science Foundation under the awards PHY-2110030 and DMS-2204702 (PGK), as well as DMS-2052525, DMS-2140982, and DMS2244976 (WZ).
Conflict of interest
The authors declare that the research was conducted in the absence of any commercial or financial relationships that could be construed as a potential conflict of interest.
Publisher’s note
All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article, or claim that may be made by its manufacturer, is not guaranteed or endorsed by the publisher.
Supplementary material
The Supplementary Material for this article can be found online at: https://www.frontiersin.org/articles/10.3389/fphot.2024.1444993/full#supplementary-material
References
1
AblowitzM. (2011). Nonlinear dispersive waves, asymptotic analysis and solitons. Cambridge: Cambridge University Press. 10.1017/cbo9780511998324
2
AblowitzM.ClarksonP. (1991). “Solitons, nonlinear evolution equations, and inverse scattering,” in 149 of London math. Soc. Lecture note series. Cambridge: Cambridge U. Press.
3
AblowitzM.PrinariB.TrubatchA. (2004). Discrete and continuous nonlinear schrödinger systems. Cambridge: Cambridge University Press.
4
AbsilP.-A.MahonyR.SepulchreR. (2008). Optimization algorithms on matrix manifolds. Princeton University Press.
5
BakarjiJ.ChampionK. P.KutzJ. N.BruntonS. L. (2022). Discovering governing equations from partial measurements with deep delay autoencoders. CoRR abs/2201, 05136. 10.1098/rspa.2023.0422
6
Belmonte-BeitiaJ.CalvoG. F.Pérez-GarcíaV. M. (2014). Effective particle methods for Fisher–Kolmogorov equations: theory and applications to brain tumor dynamics. Commun. Nonlinear Sci. Numer. Simul.19, 3267–3283. 10.1016/j.cnsns.2014.02.004
7
BruntonS. L.ProctorJ. L.KutzJ. N. (2016). Discovering governing equations from data by sparse identification of nonlinear dynamical systems, 3932–3937.
8
CenturionM.PorterM. A.KevrekidisP. G.PsaltisD. (2006). Nonlinearity management in optics: experiment, theory, and simulation. Phys. Rev. Lett.97, 033903. 10.1103/PhysRevLett.97.033903
9
ChampionK.LuschB.KutzJ. N.BruntonS. L. (2019). Data-driven discovery of coordinates and governing equations. Proc. Natl. Acad. Sci.116, 22445–22451. 10.1073/pnas.1906995116
10
ChenR. T.RubanovaY.BettencourtJ.DuvenaudD. K. (2018). Neural ordinary differential equations. Adv. neural Inf. Process. Syst.31. 10.48550/arXiv.1806.07366
11
De KosterP.WahlsS. (2024). Data-driven identification of the spectral operator in AKNS Lax pairs using conserved quantities. Wave Motion127, 103273. 10.1016/j.wavemoti.2024.103273
12
DonleyE. A.ClaussenN. R.CornishS. L.RobertsJ. L.CornellE. A.WiemanC. E. (2001). Dynamics of collapsing and exploding Bose–Einstein condensates. Nature412, 295–299. 10.1038/35085500
13
EdelmanA.AriasT. A.SmithS. T. (1998). The geometry of algorithms with orthogonality constraints. SIAM J. Matrix Analysis Appl.20, 303–353. 10.1137/s0895479895290954
14
Feliu-FabaJ.FanY.YingL. (2020). Meta-learning pseudo-differential operators with deep neural networks. J. Comput. Phys.408, 109309. 10.1016/j.jcp.2020.109309
15
García-RipollJ.Pérez-GarcíaV. (1999). The moment method in general nonlinear Schrödinger equations
16
HasegawaA.KodamaY. (1995). Solitons in optical communications. Oxford: Clarendon Press.
17
InfeldE.RowlandsG. (2000). Nonlinear waves, solitons and chaos. Cambridge: Cambridge University Press.
18
KarjantoN. (2024). Modeling wave packet dynamics and exploring applications: a comprehensive guide to the nonlinear Schrödinger equation. Mathematics12, 744. 10.3390/math12050744
19
KarniadakisG.KevrekidisI.LuL.PerdikarisP.WangS.YangL. (2021). Physics-informed machine learning. Nat. Rev. Phys.3, 422–440. 10.1038/s42254-021-00314-5
20
KassamA.-K.TrefethenL. N. (2005). Fourth-order time-stepping for stiff PDEs. SIAM J. Sci. Comput.26, 1214–1233. 10.1137/s1064827502410633
21
KevrekidisP. G.FrantzeskakisD. J.Carretero-GonzálezR. (2015). The defocusing nonlinear schrödinger equation. Philadelphia: SIAM.
22
KevrekidisP. G.KhareA.SaxenaA. (2004). Solitary wave interactions in dispersive equations using Manton’s approach. Phys. Rev. E70, 057603. 10.1103/PhysRevE.70.057603
23
KivsharY. S.AgrawalG. P. (2003). Optical solitons: from fibers to photonic crystals. Academic Press. 10.1016/B978-0-12-410590-4.X5000-1
24
KonoM.SkorićM. (2010). Nonlinear physics of plasmas. Heidelberg: Springer-Verlag.
25
KrippendorfS.LüstD.SyvaeriM. (2021). Integrability ex machina. Fortschritte Phys.69, 2100057. 10.1002/prop.202100057
26
LiZ.KovachkiN.AzizzadenesheliK.LiuB.BhattacharyaK.StuartA.et al (2021). “Fourier neural operator for parametric partial differential equations,” in International Conference on Learning Representations.
27
LiuX.XiaoN.Xiang YuanY. (2021). A penalty-free infeasible approach for a class of nonsmooth optimization problems over the Stiefel manifold. J. Sci. Comput.99, 30. 10.1007/s10915-024-02495-4
28
LiuZ.MadhavanV.TegmarkM. (2022). Machine learning conservation laws from differential equations. Phys. Rev. E106, 045307. 10.1103/physreve.106.045307
29
LiuZ.TegmarkM. (2021). Machine learning conservation laws from trajectories. Phys. Rev. Lett.126, 180604. 10.1103/PhysRevLett.126.180604
30
LiuZ.TegmarkM. (2022). Machine learning hidden symmetries. Phys. Rev. Lett.128, 180201. 10.1103/PhysRevLett.128.180201
31
LuL.MengX.MaoZ.KarniadakisG. (2021). DeepXDE: a deep learning library for solving differential equations. SIAM Rev.63, 208–228. 10.1137/19m1274067
32
MaM.NavarroR.Carretero-GonzálezR. (2016). Solitons riding on solitons and the quantum Newton’s cradle. Phys. Rev. E93, 022202. 10.1103/PhysRevE.93.022202
33
MalomedB. A. (2002). Progress in optics (Elsevier). Var. methods nonlinear fiber Opt. Relat. fields43, 69–191. 10.1016/S0079-6638(02)80026-9
34
MantonN. (1979). An effective Lagrangian for solitons. Nucl. Phys. B150, 397–412. 10.1016/0550-3213(79)90309-2
35
OviedoH.DalmauO. (2019). “A scaled gradient projection method for minimization over the Stiefel manifold,” in Advances in soft computing. Editors Martínez-VillaseñorL.BatyrshinI.Marín-HernándezA. (Cham: Springer International Publishing), 239–250.
36
Pérez-GarcíaV.TorresP.MontesinosG. (2007). The method of moments for nonlinear Schrödinger equations: theory and applications. SIAM J. Appl. Math.67, 990–1015. 10.1137/050643131
37
PethickC. J.SmithH. (2002). Bose–einstein condensation in Dilute Gases. Cambridge, United Kingdom: Cambridge University Press.
38
PitaevskiiL.StringariS. (2003). Bose-Einstein condensation. Oxford: Oxford University Press.
39
RaissiM.PerdikarisP.KarniadakisG. E. (2019). 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. 10.1016/j.jcp.2018.10.045
40
SchaefferH. (2017). Learning partial differential equations via data discovery and sparse optimization. Proc. R. Soc. A Math. Phys. Eng. Sci.473, 20160446. 10.1098/rspa.2016.0446
41
ShagalovA. G.FriedlandL. (2024). Autoresonant generation of solitons in Bose-Einstein condensates by modulation of the interaction strength. Phys. Rev. E109, 014201. 10.1103/PhysRevE.109.014201
42
StaliunasK.LonghiS.de ValcárcelG. J. (2002). Faraday patterns in Bose-Einstein condensates. Phys. Rev. Lett.89, 210406. 10.1103/PhysRevLett.89.210406
43
SulemC.SulemP. (1999). The nonlinear Schrödinger equation: self-focusing and wave collapse. New York: Springer.
44
TibshiraniR. (1996). Regression shrinkage and selection via the lasso. J. R. Stat. Soc. Ser. B Stat. Methodol.58, 267–288. 10.1111/j.2517-6161.1996.tb02080.x
45
XiaoN.LiuX.YuanY.-x. (2020). A class of smooth exact penalty function methods for optimization problems with orthogonality constraints. Optim. Methods Softw.37, 1205–1241. 10.1080/10556788.2020.1852236
46
ZhangS.FuZ.ZhuB.FanG.ChenY.WangS.et al (2021). Solitary beam propagation in periodic layered Kerr media enables high-efficiency pulse compression and mode self-cleaning. Light Sci. and Appl.10, 53. 10.1038/s41377-021-00495-9
47
ZhuW.ZhangH.-K.KevrekidisP. G. (2023). Machine learning of independent conservation laws through neural deflation. Phys. Rev. E108, L022301. 10.1103/PhysRevE.108.L022301
Summary
Keywords
NLS models, SINDy, data-driven methods, moment equations, reduced-order modeling
Citation
Yang S, Chen S, Zhu W and Kevrekidis PG (2024) Identification of moment equations via data-driven approaches in nonlinear Schrödinger models. Front. Photonics 5:1444993. doi: 10.3389/fphot.2024.1444993
Received
06 June 2024
Accepted
30 September 2024
Published
30 October 2024
Volume
5 - 2024
Edited by
Georgios D. Barmparis, Foundation for Research and Technology Hellas, Greece
Reviewed by
Lifu Zhang, Shenzhen University, China
Natanael Karjanto, Sungkyunkwan University, Republic of Korea
Updates
Copyright
© 2024 Yang, Chen, Zhu and Kevrekidis.
This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) and the copyright owner(s) are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.
*Correspondence: Su Yang, suyang@umass.edu
Disclaimer
All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.