MULTI-LOSS-AWARE CHANNEL PRUNING OF DEEP NETWORKS
文献类型:会议论文
作者 | Hu YM(胡一鸣)![]() |
出版日期 | 2019 |
会议日期 | 2019.9.23 |
会议地点 | 台湾,台北国际会议中心 |
英文摘要 | Channel pruning, which seeks to reduce the model size by removing redundant channels, is a popular solution for deep networks compression. Existing channel pruning methods usually conduct layer-wise channel selection by directly minimizing the reconstruction error of feature maps between the baseline model and the pruned one. However, they ignore the feature and semantic distributions within feature maps and real contribution of channels to the overall performance. In this paper, we propose a new channel pruning method by explicitly using both intermediate outputs of the baseline model and the classification loss of the pruned model to supervise layer-wise channel selection. Particularly, we introduce an additional loss to encode the differences in the feature and semantic distributions within feature maps between the baseline model and the pruned one. By considering the reconstruction error, the additional loss and the classification loss at the same time, our approach can significantly improve the performance of the pruned model. Comprehensive experiments on benchmark datasets demonstrate the effectiveness of the proposed method. |
语种 | 英语 |
源URL | [http://ir.ia.ac.cn/handle/173211/44838] ![]() |
专题 | 精密感知与控制研究中心_精密感知与控制 |
作者单位 | 1.School of Computer and Control Engineering, University of Chinese Academy of Sciences 2.Institute of Automation, Chinese Academy of Sciences |
推荐引用方式 GB/T 7714 | Hu YM. MULTI-LOSS-AWARE CHANNEL PRUNING OF DEEP NETWORKS[C]. 见:. 台湾,台北国际会议中心. 2019.9.23. |
入库方式: OAI收割
来源:自动化研究所
其他版本
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。