中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
Towards Automatic Model Compression via A Unified Two-Stage Framework

文献类型:期刊论文

作者Weihan Chen1,3; Peisong Wang1; Jian Cheng1,2,4
刊名Pattern Recognition (PR)
出版日期2023
卷号140页码:109527
关键词Deep Neural Networks Model Compression Quantization Pruning
英文摘要

Deep Neural Networks have become ubiquitous in various domains. Meanwhile, the problems of massive storage and computation costs have hindered the deploy- ment of these models to real-world applications. This paper proposes a novel and unified two-stage framework for automatic model compression. To determine the compression ratio of each layer, we improve the optimization from two aspects. First, to predict the performance of each compression policy, we propose Dynamic BN, which improves the correlation significantly with little computation overhead. Second, to search for the compression ratio allocation, we propose an efficient and hyperparameter-free solving algorithm based on the proposed Hessian matrix approximation and Knapsack problem reformulation. Moreover, comprehensive experiments and analyses are conducted on the CIFAR-100&ImageNet datasets and various network architectures to demonstrate its performance advantages over existing model compression methods under the quantization-only, pruning-only, and pruning-quantization settings.

语种英语
源URL[http://ir.ia.ac.cn/handle/173211/52076]  
专题类脑芯片与系统研究
通讯作者Jian Cheng
作者单位1.Institute of Automation, Chinese Academy of Sciences, Beijing, 100190, China
2.AiRiA
3.School of Artificial Intelligence, University of Chinese Academy of Sciences, Beijing, 100049, China
4.School of Future Technology, University of Chinese Academy of Sciences, Beijing, 100049, China
推荐引用方式
GB/T 7714
Weihan Chen,Peisong Wang,Jian Cheng. Towards Automatic Model Compression via A Unified Two-Stage Framework[J]. Pattern Recognition (PR),2023,140:109527.
APA Weihan Chen,Peisong Wang,&Jian Cheng.(2023).Towards Automatic Model Compression via A Unified Two-Stage Framework.Pattern Recognition (PR),140,109527.
MLA Weihan Chen,et al."Towards Automatic Model Compression via A Unified Two-Stage Framework".Pattern Recognition (PR) 140(2023):109527.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。