Structured covariance estimation via tensor-train decomposition
Artsiom Patarusau, Nikita Puchkin, Maxim Rakhuba, Fedor Noskov
TL;DR
The paper tackles high-dimensional covariance estimation by imposing a tensor-train structured Kronecker model $\Sigma = \sum_{j=1}^J \sum_{k=1}^K U_j \otimes V_{jk} \otimes W_k$, enabling dimension-free analysis via the rearrangement $\mathcal{R}(\Sigma) \in \mathbb{R}^{p^2 \times q^2 \times r^2}$. It introduces the HarTTh algorithm, a TT-SVD/HOOI-inspired iterative estimator that recovers the best $(J,K)$-TT approximation of $\mathcal{R}(\Sigma)$ from noisy observations, with estimator $\\tilde{\\Sigma} = \\mathcal{R}^{-1}(\\widehat{\\mathcal{T}})$. The authors establish non-asymptotic, high-probability bounds that decompose error into a bias term due to model misspecification and a variance term that scales as $\\|\\Sigma\\|$ times $\\sqrt{(J \\mathtt{r}_1^2(\\Sigma) + JK \\mathtt{r}_2^2(\\Sigma) + K \\mathtt{r}_3^2(\\Sigma) + \\log(1/\\delta))/n}$, with additional remainder terms. Numerical experiments demonstrate competitive performance and computational efficiency of HarTTh relative to TT-HOSVD, Tucker-based methods, and PRLS, illustrating practical viability for structured covariance estimation in multiway data.
Abstract
We consider a problem of covariance estimation from a sample of i.i.d. high-dimensional random vectors. To avoid the curse of dimensionality we impose an additional assumption on the structure of the covariance matrix $Σ$. To be more precise we study the case when $Σ$ can be approximated by a sum of double Kronecker products of smaller matrices in a tensor train (TT) format. Our setup naturally extends widely known Kronecker sum and CANDECOMP/PARAFAC models but admits richer interaction across modes. We suggest an iterative polynomial time algorithm based on TT-SVD and higher-order orthogonal iteration (HOOI) adapted to Tucker-2 hybrid structure. We derive non-asymptotic dimension-free bounds on the accuracy of covariance estimation taking into account hidden Kronecker product and tensor train structures. The efficiency of our approach is illustrated with numerical experiments.
