Kronecker regression is a highly-structured least squares problem $\min_{\mathbf{x}} \lVert \mathbf{K}\mathbf{x} - \mathbf{b} \rVert_{2}^2$, where the design matrix $\mathbf{K} = \mathbf{A}^{(1)} \otimes \cdots \otimes \mathbf{A}^{(N)}$ is a Kronecker product of factor matrices. This regression problem arises in each step of the widely-used alternating least squares (ALS) algorithm for computing the Tucker decomposition of a tensor. We present the first subquadratic-time algorithm for solving Kronecker regression to a $(1+\varepsilon)$-approximation that avoids the exponential term $O(\varepsilon^{-N})$ in the running time. Our techniques combine leverage score sampling and iterative methods. By extending our approach to block-design matrices where one block is a Kronecker product, we also achieve subquadratic-time algorithms for (1) Kronecker ridge regression and (2) updating the factor matrices of a Tucker decomposition in ALS, which is not a pure Kronecker regression problem, thereby improving the running time of all steps of Tucker ALS. We demonstrate the speed and accuracy of this Kronecker regression algorithm on synthetic data and real-world image tensors.
翻译:克罗内克回归是一个高度结构化的最小二乘问题 $\min_{\mathbf{x}} \lVert \mathbf{K}\mathbf{x} - \mathbf{b} \rVert_{2}^2$,其中设计矩阵 $\mathbf{K} = \mathbf{A}^{(1)} \otimes \cdots \otimes \mathbf{A}^{(N)}$ 是因子矩阵的克罗内克积。该回归问题出现在广泛使用的交替最小二乘(ALS)算法中,用于计算张量的塔克分解的每一步。我们提出了首个次二次时间算法,用于求解克罗内克回归的 $(1+\varepsilon)$-近似解,从而避免了运行时间中的指数项 $O(\varepsilon^{-N})$。我们的技术结合了杠杆得分采样和迭代方法。通过将我们的方法扩展到其中一个块是克罗内克积的块设计矩阵,我们还实现了以下问题的次二次时间算法:(1) 克罗内克岭回归;(2) 在ALS中更新塔克分解的因子矩阵(这不是一个纯粹的克罗内克回归问题),从而改进了塔克ALS所有步骤的运行时间。我们在合成数据和真实世界图像张量上展示了该克罗内克回归算法的速度和准确性。