Discrete Aware Tensor Completion via Convexized $\ell_0$-Norm Approximation
Niclas Führling, Getuar Rexhepi, Giuseppe Abreu
TL;DR
This work tackles the problem of completing low-rank tensors whose entries come from a discrete alphabet by introducing a discrete-aware tensor completion (DaLRTC) framework. The method augments the conventional nuclear-norm objective with a differentiable $l_0$-norm–based discrete regularizer, which is convexified via fractional programming and optimized with a proximal-gradient algorithm. The approach jointly leverages a data-fidelity term, a multi-linear nuclear-norm prior, and a discrete prior to improve recovery from partial observations, demonstrated to achieve superior NMSE and convergence over state-of-the-art NN-based and matrix-factorization-based methods on color images. This discrete-aware TC is applicable to image processing and broader multidimensional data problems with discrete-valued measurements, offering improved reconstruction quality in scenarios with partial and quantized data.
Abstract
We consider a novel algorithm, for the completion of partially observed low-rank tensors, where each entry of the tensor can be chosen from a discrete finite alphabet set, such as in common image processing problems, where the entries represent the RGB values. The proposed low-rank tensor completion (TC) method builds on the conventional nuclear norm (NN) minimization-based low-rank TC paradigm, through the addition of a discrete-aware regularizer, which enforces discreteness in the objective of the problem, by an $\ell_0$-norm regularizer that is approximated by a continuous and differentiable function normalized via fractional programming (FP) under a proximal gradient (PG) framework, in order to solve the proposed problem. Simulation results demonstrate the superior performance of the new method both in terms of normalized mean square error (NMSE) and convergence, compared to the conventional state of-the-art (SotA) techniques, including NN minimization approaches, as well as a mixture of the latter with a matrix factorization approach.
