Spectral tensor-train decomposition

TL;DR

This paper introduces a spectral tensor-train decomposition method for high-dimensional function approximation, combining tensor-train efficiency with spectral convergence for improved accuracy in uncertainty quantification.

Contribution

It develops a functional spectral tensor-train framework, analyzes its convergence, and demonstrates its effectiveness on high-dimensional problems with regularity-based spectral approximations.

Findings

Achieves spectral convergence for functions with sufficient regularity.

Outperforms anisotropic adaptive Smolyak methods in numerical tests.

Successfully approximates solutions to PDEs with random inputs.

Abstract

The accurate approximation of high-dimensional functions is an essential task in uncertainty quantification and many other fields. We propose a new function approximation scheme based on a spectral extension of the tensor-train (TT) decomposition. We first define a functional version of the TT decomposition and analyze its properties. We obtain results on the convergence of the decomposition, revealing links between the regularity of the function, the dimension of the input space, and the TT ranks. We also show that the regularity of the target function is preserved by the univariate functions (i.e., the "cores") comprising the functional TT decomposition. This result motivates an approximation scheme employing polynomial approximations of the cores. For functions with appropriate regularity, the resulting \textit{spectral tensor-train decomposition} combines the favorable dimension-scaling of the TT decomposition with the spectral convergence rate of polynomial approximations, yielding efficient and accurate surrogates for high-dimensional functions. To construct these decompositions, we use the sampling algorithm \texttt{TT-DMRG-cross} to obtain the TT decomposition of tensors resulting from suitable discretizations of the target function. We assess the performance of the method on a range of numerical examples: a modifed set of Genz functions with dimension up to $100$, and functions with mixed Fourier modes or with local features. We observe significant improvements in performance over an anisotropic adaptive Smolyak approach. The method is also used to approximate the solution of an elliptic PDE with random input data. The open source software and examples presented in this work are available online.