Self-Attention Driven Tensor Representation for High-Order Data Recovery
Zhi-Wei Shi, Yu-Bang Zheng, Heng-Chao Li
Abstract
Low-rank tensor representation (LRTR) is an effective tool for compactly modeling high-order data. While nonlinear LRTR models can better capture real-world nonlinear dependencies, most existing methods rely on fixed local mappings of multilayer perceptrons (MLPs) or convolutional neural networks (CNNs), limiting their ability to model complex global dependencies. To overcome this limitation, we construct a novel paradigm called Self-Attention Driven Tensor Representation (SADTR), which is the first framework that models nonlinearity from the perspective of self-attention. Specifically, we design a factor self-representation mechanism to establish dynamic global mapping, thereby adaptively capturing both local and nonlocal nonlinear dependencies in the factor space. Moreover, we introduce an implicit sparse representation to impose sparsity constraint while avoiding additional optimization problems. As a result, the proposed SADTR can achieve a more accurate low-rank representation. In theory, we provide a detailed analysis to demonstrate the recoverability of SADTR. To validate the effectiveness of SADTR, we apply it to three representative high-order data recovery tasks. Experimental results demonstrate that SADTR consistently outperforms existing state-of-the-art LRTR methods.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on10
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- LoRA: Low-Rank Adaptation of Large Language ModelsEdward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu et al.ICLR 2022 · 18,833 citations
- Fully-Connected Tensor Network Decomposition and Its Application to Higher-Order Tensor CompletionYu-Bang Zheng, Ting-Zhu Huang, Xi-Le Zhao, Qibin Zhao et al.AAAI 2021 · 183 citations
- HLRTF: Hierarchical Low-Rank Tensor Factorization for Inverse Problems in Multi-Dimensional ImagingYi-Si Luo, Xile Zhao, Deyu Meng, Tai-Xiang JiangCVPR 2022 · 45 citations
- Imposing Consistency for Optical Flow EstimationJisoo Jeong, Jamie Menjay Lin, Fatih Porikli, Nojun KwakCVPR 2022 · 40 citations
Related papers
- HiNCoT: Hierarchical Nonlinear Continuous Transform-based Tensor Representation for Multi-Dimensional Data RecoveryTao Yang, Weihao Wu, Tingzhu HuangAAAI 2026
- Gaussian Splatting-based Low-Rank Tensor Representation for Multi-Dimensional Image RecoveryYiming Zeng, Xi-Le Zhao, Wei-Hao Wu, Teng-Yu Ji et al.CVPR 2026 · 2 citations
- Implicit Regularization in Tensor FactorizationNoam Razin, Asaf Maman, Nadav CohenICML 2021 · 60 citations
- High-Order Contrastive Learning with Fine-grained Comparative Levels for Sparse Ordinal Tensor CompletionYu Dai, Junchen Shen, Zijie Zhai, Danlin Liu et al.ICML 2024
- Reparameterized Tensor Ring Functional Decomposition for Multi-Dimensional Data RecoveryYangyang Xu, Junbo Ke, You-Wei Wen, Chao WangCVPR 2026 · 1 citation
