Multi-Dimensional Visual Data Completion via Low-Rank Tensor Representation Under Coupled Transform

Jian-Li Wang,Ting-Zhu Huang,Xi-Le Zhao,Tai-Xiang Jiang,Michael K. Ng
DOI: https://doi.org/10.1109/tip.2021.3062995
IF: 10.6
2021-01-01
IEEE Transactions on Image Processing
Abstract:This paper addresses the tensor completion problem, which aims to recover missing information of multi-dimensional images. How to represent a low-rank structure embedded in the underlying data is the key issue in tensor completion. In this work, we suggest a novel low-rank tensor representation based on coupled transform, which fully exploits the spatial multi-scale nature and redundancy in spatial and spectral/temporal dimensions, leading to a better low tensor multi-rank approximation. More precisely, this representation is achieved by using two-dimensional framelet transform for the two spatial dimensions, one/two-dimensional Fourier transform for the temporal/spectral dimension, and then Karhunen–Loéve transform (via singular value decomposition) for the transformed tensor. Based on this low-rank tensor representation, we formulate a novel low-rank tensor completion model for recovering missing information in multi-dimensional visual data, which leads to a convex optimization problem. To tackle the proposed model, we develop the alternating directional method of multipliers (ADMM) algorithm tailored for the structured optimization problem. Numerical examples on color images, multispectral images, and videos illustrate that the proposed method outperforms many state-of-the-art methods in qualitative and quantitative aspects.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?
This paper aims to solve the tensor completion problem in multi - dimensional images. The specific goal is to recover the missing information in multi - dimensional visual data from a small number of known observations. The key to the paper lies in how to represent the low - rank structure embedded in the underlying data, which is the core problem of tensor completion. The author proposes a new low - rank tensor representation method based on coupled transformation. This method makes full use of the spatial multi - scale characteristics and the redundancy of spatial and spectral / temporal dimensions, thereby achieving a better multi - rank approximation of low - rank tensors. ### Main contributions: 1. **New low - rank tensor representation**: A low - rank tensor representation method based on coupled transformation is proposed, which can more effectively capture the internal structure in multi - dimensional visual data, especially achieving low - rank approximation in the transform domain. 2. **Low - rank tensor completion model**: Based on the proposed low - rank tensor representation method, a new low - rank tensor completion model is constructed, and an efficient alternating direction multiplier method (ADMM) algorithm is developed to solve this model. 3. **Experimental verification**: Through numerical experiments on color images, multispectral images and videos, it is proved that the proposed method is superior to many existing methods in both qualitative and quantitative aspects. ### Method overview: 1. **First - layer transformation (B1)**: The two - dimensional framelet transform is used to describe the local spatial correlation, which can express the rich spatial structures at different frequencies and spatial directions in detail, and the multi - level decomposition helps to capture the multi - scale features of the tensor. 2. **Second - layer transformation (B2)**: The Fourier transform is used to depict the global correlation in the time (or spectral) dimension, revealing the strong correlation between the multi - scale representation coefficients after the first - layer transformation. 3. **Third - layer transformation (B3)**: The Karhunen - Loève transform (KL transform) is used to characterize the global spatial correlation of the results of the first two - layer transformations through singular value decomposition (SVD), combining global correlation and local geometric details. ### Mathematical model: The proposed low - rank tensor completion model can be expressed as: \[ \min_{X} \| B(X) \|_1 \quad \text{s.t.} \quad P_\Omega(X) = P_\Omega(O) \] where \( B = B_3 \circ B_2 \circ B_1 \) is the coupled transformation, involving three - layer transformations, namely framelet transform, Fourier transform and KL transform. \( P_\Omega \) is the sampling operator, which retains the elements in the index set \( \Omega \) and sets other positions to zero. ### Experimental results: The experimental results show that the proposed method performs well on a variety of multi - dimensional visual data, especially at a low sampling rate, and can better recover specific textures rather than just the global outline. ### Conclusion: This paper effectively solves the tensor completion problem in multi - dimensional visual data by proposing a low - rank tensor representation method based on coupled transformation, which has high practical value and theoretical significance.