剪枝小白看這里


論文:《AutoCompress: An Automatic DNN Structured Pruning Framework for Ultra-High Compression Rates》

論文地址:https://arxiv.org/abs/1907.03141

論文的三個創新點:

  1.提出混合型的結構化剪枝維度。

  2.采用剪枝算法ADMM(交替乘子優化算法)對訓練過程中的正則項進行動態更新。

  3.利用了增強型引導啟發式搜索的方式進行行為抽樣。

結構化剪枝包括三種剪枝維度,輸出通道剪枝(filter pruning)、輸入通道剪枝(channel pruning)、輸出通道形狀剪枝(filter shape/column pruning)。如上圖所示可以更加理解剪枝的不同類型。

輸出通道剪枝(filter pruning):直接刪除一個卷積核。

輸入通道剪枝(channel pruning):刪除每個卷積核對應的輸入通道。

輸出通道形狀剪枝(filter shape/column pruning):刪除每個卷積核上相同位置的權重。

 

主要想要理解各個剪枝過程。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM