References & Citations
Computer Science > Computer Vision and Pattern Recognition
Title: Heuristic Rank Selection with Progressively Searching Tensor Ring Network
(Submitted on 22 Sep 2020 (v1), last revised 30 May 2021 (this version, v2))
Abstract: Recently, Tensor Ring Networks (TRNs) have been applied in deep networks, achieving remarkable successes in compression ratio and accuracy. Although highly related to the performance of TRNs, rank selection is seldom studied in previous works and usually set to equal in experiments. Meanwhile, there is not any heuristic method to choose the rank, and an enumerating way to find appropriate rank is extremely time-consuming. Interestingly, we discover that part of the rank elements is sensitive and usually aggregate in a narrow region, namely an interest region. Therefore, based on the above phenomenon, we propose a novel progressive genetic algorithm named Progressively Searching Tensor Ring Network Search (PSTRN), which has the ability to find optimal rank precisely and efficiently. Through the evolutionary phase and progressive phase, PSTRN can converge to the interest region quickly and harvest good performance. Experimental results show that PSTRN can significantly reduce the complexity of seeking rank, compared with the enumerating method. Furthermore, our method is validated on public benchmarks like MNIST, CIFAR10/100, UCF11 and HMDB51, achieving the state-of-the-art performance.
Submission history
From: Nannan Li [view email][v1] Tue, 22 Sep 2020 14:44:27 GMT (2479kb,D)
[v2] Sun, 30 May 2021 08:44:25 GMT (2481kb,D)
Link back to: arXiv, form interface, contact.