モデルの圧縮は、電力とメモリに制約のあるリソースにディープニューラルネットワーク(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.