arXiv reaDer
ResRep:記憶と忘却の分離によるロスレスCNNプルーニング
ResRep: Lossless CNN Pruning via Decoupling Remembering and Forgetting
畳み込み層の幅(出力チャネルの数)を減らすことによってCNNをスリム化する、ロスレスチャネルプルーニング(別名フィルタープルーニング)の新しい方法であるResRepを提案します。記憶と忘却の独立性に関する神経生物学の研究に触発されて、CNNを記憶部分と忘却部分に再パラメーター化することを提案します。前者はパフォーマンスの維持を学び、後者は剪定を学びます。前者については通常のSGDを使用し、後者についてはペナルティ勾配を使用した新しい更新ルールを使用したトレーニングにより、構造化されたスパース性を実現します。次に、記憶と忘却の部分を、より狭いレイヤーを持つ元のアーキテクチャに同等にマージします。この意味で、ResRepは構造再パラメータ化の成功したアプリケーションと見なすことができます。このような方法論は、ResRepを、パラメーターにペナルティを適用してスパース性を生成する従来の学習ベースのプルーニングパラダイムと区別します。これにより、記憶に不可欠なパラメーターが抑制される可能性があります。 ResRepは、ImageNetで76.15%の精度を持つ標準のResNet-50を、わずか45%のFLOPと精度の低下のない狭いものにスリム化します。これは、このような高い圧縮率でロスレスプルーニングを実現した最初の製品です。コードとモデルはhttps://github.com/DingXiaoH/ResRepにあります。
We propose ResRep, a novel method for lossless channel pruning (a.k.a. filter pruning), which slims down a CNN by reducing the width (number of output channels) of convolutional layers. Inspired by the neurobiology research about the independence of remembering and forgetting, we propose to re-parameterize a CNN into the remembering parts and forgetting parts, where the former learn to maintain the performance and the latter learn to prune. Via training with regular SGD on the former but a novel update rule with penalty gradients on the latter, we realize structured sparsity. Then we equivalently merge the remembering and forgetting parts into the original architecture with narrower layers. In this sense, ResRep can be viewed as a successful application of Structural Re-parameterization. Such a methodology distinguishes ResRep from the traditional learning-based pruning paradigm that applies a penalty on parameters to produce sparsity, which may suppress the parameters essential for the remembering. ResRep slims down a standard ResNet-50 with 76.15% accuracy on ImageNet to a narrower one with only 45% FLOPs and no accuracy drop, which is the first to achieve lossless pruning with such a high compression ratio. The code and models are at https://github.com/DingXiaoH/ResRep.
updated: Sat Aug 14 2021 19:36:54 GMT+0000 (UTC)
published: Tue Jul 07 2020 07:56:45 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アソシエイト