arXiv reaDer
GNN-RL圧縮:多段階グラフ埋め込みと強化学習を使用したトポロジ対応ネットワークプルーニング
GNN-RL Compression: Topology-Aware Network Pruning using Multi-stage Graph Embedding and Reinforcement Learning
モデルの圧縮は、電力とメモリに制約のあるリソースにディープニューラルネットワーク(DNN)を展開するために不可欠な手法です。ただし、既存のモデル圧縮方法は、多くの場合、人間の専門知識に依存し、DNN内の豊富なトポロジ情報を無視して、パラメータのローカルな重要性に焦点を合わせています。この論文では、グラフニューラルネットワーク(GNN)に基づく新しい多段階グラフ埋め込み手法を提案してDNNのトポロジを識別し、強化学習(RL)を使用して適切な圧縮ポリシーを見つけます。リソースに制約のある(つまり、FLOPs)チャネルプルーニングを実行し、過剰にパラメーター化されたDNN(ResNetやVGG-16など)やモバイルフレンドリーなDNN(MobileNetなど)を使用した最新の圧縮方法とアプローチを比較しました。 ShuffleNet)。 ResNetファミリ、VGG-16、MobileNet-v1 / v2、ShuffleNetなど、一般的なネットワークからモバイルフレンドリーなネットワークまで、さまざまなモデルでこの方法を評価しました。結果は、私たちの方法が高密度ネットワーク(VGG-16など)を元のFLOPの最大80%まで除去できることを示しています。さらに重要なことに、私たちの方法は最先端の方法を上回り、ShuffleNet-v1で最大1.84%高い精度を達成しました。さらに、私たちのアプローチに従って、プルーニングされたVGG-16は、顕著な1.38倍の速度向上と141MBのGPUメモリ削減を達成しました。
Model compression is an essential technique for deploying deep neural networks (DNNs) on power and memory-constrained resources. However, existing model-compression methods often rely on human expertise and focus on parameters' local importance, ignoring the rich topology information within DNNs. In this paper, we propose a novel multi-stage graph embedding technique based on graph neural networks (GNNs) to identify the DNNs' topology and use reinforcement learning (RL) to find a suitable compression policy. We performed resource-constrained (i.e., FLOPs) channel pruning and compared our approach with state-of-the-art compression methods using over-parameterized DNNs (e.g., ResNet and VGG-16) and mobile-friendly DNNs (e.g., MobileNet and ShuffleNet). We evaluated our method on various models from typical to mobile-friendly networks, such as ResNet family, VGG-16, MobileNet-v1/v2, and ShuffleNet. The results demonstrate that our method can prune dense networks (e.g., VGG-16) by up to 80% of their original FLOPs. More importantly, our method outperformed state-of-the-art methods and achieved a higher accuracy by up to 1.84% for ShuffleNet-v1. Furthermore, following our approach, the pruned VGG-16 achieved a noticeable 1.38× speed up and 141 MB GPU memory reduction.
updated: Fri Feb 05 2021 14:59:32 GMT+0000 (UTC)
published: Fri Feb 05 2021 14:59:32 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アソシエイト