Viability of Tensor Train Methods for Geophysical Fluid Dynamics
For researchers in geophysical fluid dynamics, this paper provides a critical evaluation showing that tensor train methods are not yet viable for realistic applications.
Tensor train methods show compression and speedup for simple flows but fail to efficiently represent complex states common in realistic geophysical fluid dynamics, as evaluated on shallow water equations with E3SM's discretization.
Tensor train (TT) methods have recently gained popularity for accelerating the solving of systems of PDEs. Here, we evaluate the performance of TT methods in the context of geophysical fluid dynamics (GFD) using the shallow water equations and a discretization scheme employed by the ocean component of the Energy Exascale Earth System Model (E3SM). Through a suite of four test cases of increasing complexity, we evaluate TT methods in terms of how much TT is able to compress the model state, the error incurred by the TT approximation, and the speedup obtained by TT versus an optimal standard non-TT implementation in a representative subproblem. We show that though TT is able to effectively compress and speed up simple flows, it struggles to efficiently represent more complex states that are common in realistic GFD applications.