When Bayesian Tensor Completion Meets Multioutput Gaussian Processes: Functional Universality and Rank Learning
About
Functional tensor decomposition can analyze multi-dimensional data with real-valued indices, paving the path for applications in machine learning and signal processing. A limitation of existing approaches is the assumption that the tensor rank-a critical parameter governing model complexity-is known. However, determining the optimal rank is a non-deterministic polynomial-time hard (NP-hard) task and there is a limited understanding regarding the expressive power of functional low-rank tensor models for continuous signals. We propose a rank-revealing functional Bayesian tensor completion (RR-FBTC) method. Modeling the latent functions through carefully designed multioutput Gaussian processes, RR-FBTC handles tensors with real-valued indices while enabling automatic tensor rank determination during the inference process. We establish the universal approximation property of the model for continuous multi-dimensional signals, demonstrating its expressive power in a concise format. To learn this model, we employ the variational inference framework and derive an efficient algorithm with closed-form updates. Experiments on both synthetic and real-world datasets demonstrate the effectiveness and superiority of the RR-FBTC over state-of-the-art approaches. The code is available at https://github.com/OceanSTARLab/RR-FBTC.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Completion | Standard Images SR 20% (test) | PSNR26.83 | 48 | |
| Image Completion | Standard Images SR 30% (test) | PSNR27.52 | 48 | |
| 3D Sound Speed Field Reconstruction | SSF data | RMSE0.358 | 36 | |
| Tensor completion | Synthetic Discrete Data tensor size 30x30x30 (20% Sampling Rate) | RRSE0.134 | 12 | |
| Tensor completion | Synthetic Discrete Data tensor size 30x30x30 (30% Sampling Rate) | RRSE0.107 | 12 | |
| Continuous Tensor Completion | Synthetic continuous data SNR=10 dB (test) | RRSE0.024 | 10 | |
| Sound Speed Field Reconstruction | SSF on-grid Sea Surface Temperature data | RMSE0.287 | 8 | |
| Continuous Tensor Completion | US-Temperature (test) | RMSE0.342 | 7 |