An Entropy-based Pruning Method for CNN Compression

這篇文章也是網絡剪枝的早期文章,其算法是filter級別的裁剪。文章認爲weights的大小不能決定一個filter的重要性,應該用熵來決定: 而對於這個概率pi,其得到方法如下: 簡單來說,就是用全局平均池化來轉化一個layer的output,得到一個向量。用驗證集的一些圖片進入網絡,就可以得到一個矩陣,這個filter的熵就可以用這個矩陣計算出來。 實驗上,文章針對VGG16和ResNet18
相關文章
相關標籤/搜索