arXiv reaDer
Tensor Ring Network を段階的に検索するヒューリスティックなランク選択
Heuristic Rank Selection with Progressively Searching Tensor Ring Network
最近、Tensor Ring Networks (TRN) がディープ ネットワークに適用され、圧縮率と精度において目覚ましい成果を上げています。 TRN のパフォーマンスに大きく関係していますが、ランクの選択は以前の研究ではめったに研究されず、通常は実験で等しく設定されます。一方、ランクを決定するためのヒューリスティックな方法はなく、適切なランクを見つけるための列挙方法は非常に時間がかかります。興味深いことに、ランク要素の一部は敏感であり、通常は狭い領域、つまり関心領域に集約されることがわかりました。したがって、上記の現象に基づいて、最適なランクを正確かつ効率的に見つける能力を持つ、Progressive Searching Tensor Ring Network Search (PSTRN) と呼ばれる新しい進歩的な遺伝的アルゴリズムを提案します。進化段階と進行段階を通じて、PSTRN は関心領域に迅速に収束し、良好なパフォーマンスを獲得できます。実験結果は、PSTRN が列挙方法と比較して、ランクを求める複雑さを大幅に軽減できることを示しています。さらに、私たちの方法は、MNIST、CIFAR10/100、UCF11、HMDB51 などのパブリック ベンチマークで検証され、最先端のパフォーマンスを実現しています。
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.
updated: Sun May 30 2021 08:44:25 GMT+0000 (UTC)
published: Tue Sep 22 2020 14:44:27 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト