SVDinsTN: A Tensor Network Paradigm for Efficient Structure Search from Regularized Modeling Perspective
About
Tensor network (TN) representation is a powerful technique for computer vision and machine learning. TN structure search (TN-SS) aims to search for a customized structure to achieve a compact representation, which is a challenging NP-hard problem. Recent "sampling-evaluation"-based methods require sampling an extensive collection of structures and evaluating them one by one, resulting in prohibitively high computational costs. To address this issue, we propose a novel TN paradigm, named SVD-inspired TN decomposition (SVDinsTN), which allows us to efficiently solve the TN-SS problem from a regularized modeling perspective, eliminating the repeated structure evaluations. To be specific, by inserting a diagonal factor for each edge of the fully-connected TN, SVDinsTN allows us to calculate TN cores and diagonal factors simultaneously, with the factor sparsity revealing a compact TN structure. In theory, we prove a convergence guarantee for the proposed method. Experimental results demonstrate that the proposed method achieves approximately 100 to 1000 times acceleration compared to the state-of-the-art TN-SS methods while maintaining a comparable level of representation ability.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Realistic color video completion | Claire 144×176×3×30 | PSNR41.65 | 70 | |
| Realistic color video completion | Akiyo 144×176×3×30 | PSNR42.02 | 70 | |
| Light field data completion | Greek 128x128x3x30 | PSNR39.42 | 70 | |
| Light field data completion | Medieval2 128x128x3x30 | PSNR42.4 | 70 | |
| Realistic color video completion | News 144×176×3×30 | PSNR37.16 | 70 | |
| Realistic color video completion | Grandma 144×176×3×30 | PSNR41.4 | 70 | |
| Light field data completion | Museum 128x128x3x30 | PSNR42.19 | 70 | |
| Light field data completion | Vinyl 128x128x3x30 | PSNR45.37 | 70 | |
| MSI Completion | Flowers 256x256x31 (test) | PSNR41.63 | 35 | |
| MSI Completion | Feathers 256x256x31 (test) | PSNR41.18 | 35 |