モデルの圧縮は、電力とメモリに制約のあるリソースにディープニューラルネットワーク(DNN)を展開するために不可欠な手法です。ただし、既存のモデル圧縮方法は、多くの場合、人間の専門知識に依存し、DNN内の豊富なトポロジ情報を無視して、パラメータのローカルな重要性に焦点を合わせています。この論文では、グラフニューラルネットワーク(GNN)に基づく新しい多段階グラフ埋め込み手法を提案してDNNトポロジを識別し、強化学習(RL)を使用して適切な圧縮ポリシーを見つけます。リソースに制約のある(つまり、FLOP)チャネルプルーニングを実行し、アプローチを最先端のモデル圧縮方法と比較しました。 ResNetファミリ、VGG-16、MobileNet-v1 / v2、ShuffleNetなど、一般的なネットワークからモバイルフレンドリーなネットワークまで、さまざまなモデルでこの方法を評価しました。結果は、私たちの方法が最小限の微調整コストでより高い圧縮率を達成できる一方で、卓越した競争力のあるパフォーマンスを生み出すことを示しています。
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 DNN topologies 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 model compression methods. We evaluated our method on various models from typical to mobile-friendly networks, such as ResNet family, VGG-16, MobileNet-v1/v2, and ShuffleNet. Results show that our method can achieve higher compression ratios with a minimal fine-tuning cost yet yields outstanding and competitive performance.