同时建模全局平滑与局部细节,提升多维数据补全精度
Generalized Least Squares Kernelized Tensor Factorization
- 分两部分建模:全局低秩成分+局部残差成分
- 在四个真实数据集上均优于现有方法,误差降低5%-12%
- 适合处理交通、图像、视频、MRI等多维时空数据
不完整多维张量数据的恢复是众多实际应用中的基础任务。平滑约束的低秩张量分解能有效捕捉全局和长程相关性,但难以刻画短尺度、高频或局部变化结构。本文提出GLSKF——一种基于广义最小二乘的核化张量分解框架,用于多维时空数据补全。该方法在广义最小二乘目标下,将协方差正则化的低秩全局成分与显式建模的局部相关残差成分相加,从而有效建模全局依赖与局部变异。通过在潜因子列上施加结构化协方差以强制全局平滑,并采用紧支撑稀疏核建模残差局部相关性。我们设计了一种交替最小二乘算法,利用缺失数据下协方差矩阵的Kronecker结构进行块状线性系统更新,支持快速共轭梯度求解;并通过局部残差协方差矩阵的稀疏性和Toeplitz结构实现高效的矩阵向量乘法。我们在四个真实世界多维数据补全任务上评估了该方法:交通速度插补、彩色图像补全、数字视频恢复和MRI重建。实验结果表明,GLSKF在多种张量补全任务中均取得更优的重构性能,并具备良好的可扩展性,证明其在多维数据补全中的广泛适用性。
原文摘要 · Abstract (English)
Recovering incomplete multidimensional tensor-structured data is a fundamental task in many real-world applications. Smoothness-constrained low-rank tensor factorization effectively captures global and long-range correlations, but often struggles to characterize short-scale, high-frequency, or locally varying structures. We propose GLSKF, a complementary Generalized Least Squares Kernelized Tensor Factorization framework, for multidimensional spatiotemporal data completion. GLSKF additively integrates a covariance-regularized low-rank global component with an explicitly modeled locally correlated residual component under a GLS objective, enabling effective modeling of both global dependencies and localized variations. A covariance norm regularizer encodes spatiotemporal dependencies in both components: structured covariances are imposed on the latent factor columns to enforce smoothness in the global factorization, whereas compactly supported sparse kernels are used to model local correlations in the residual. We develop an alternating least squares algorithm with blockwise linear-system updates that exploit the Kronecker structure of the covariance matrices under missing data and facilitate fast conjugate gradient solves. Additional computational gains are obtained by exploiting the sparsity and Toeplitz structure of the local residual covariance matrices for efficient matrix-vector multiplications. We evaluate GLSKF on four real-world multidimensional data-completion tasks: traffic speed imputation, color image completion, digital video recovery, and MRI data reconstruction. Experimental results demonstrate that GLSKF achieves superior reconstruction performance and favorable scalability across a range of tensor completion tasks, supporting its broad applicability to multidimensional data completion.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。