Combining deep learning model compression techniques

被引:1
作者
Santos Silva, Jose Vitor [1 ]
Matos Matos, Leonardo [1 ]
Santos, Flavio [2 ]
Magalhaes Cerqueira, Helisson Oliveira [1 ]
Macedo, Hendrik [1 ]
Piedade Prado, Bruno Otavio [1 ]
Ferreira da Silva, Gilton Jose [1 ]
Bispo, Kalil Araujo [1 ]
机构
[1] Univ Fed Sergipe UFS, Sao Cristovao, Sergipe, Brazil
[2] Univ Fed Pernambuco UFPE, Recife, PE, Brazil
关键词
Pulmonary diseases; Computational modeling; Deep learning; Adaptation models; X-ray imaging; Quantization (signal); Internet; model compression; dark knowledge distillation; pruning; quantization;
D O I
10.1109/TLA.2022.9667144
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In this article, we evaluate the performance of combining several model compression techniques. The techniques assessed were dark knowledge distillation, pruning, and quantization. We found that in the scenario in which we developed the experiments, classification of chest x-rays, the combination of these three techniques yielded a new model capable of aggregating the individual advantages of each one. In the experiments we used a combination of deep models with 95.05% accuracy, a value higher than that reported in some related works but lower than the state of the art, whose accuracy is 96.39%. The accuracy of the compressed model in turn was 90.86%, a small loss compared to the gain obtained from the reduction, in bytes, in relation to the size of the original model. The size has been reduced from 841MB to 40KB, which opens up the possibility for using deep models in edge computing applications.
引用
收藏
页码:458 / 464
页数:7
相关论文
共 21 条
[1]  
Blalock D, 2020, ARXIV PREPRINT ARXIV
[2]  
Caruana R, 2001, ADV NEUR IN, V13, P402
[3]   A Novel Transfer Learning Based Approach for Pneumonia Detection in Chest X-ray Images [J].
Chouhan, Vikash ;
Singh, Sanjay Kumar ;
Khamparia, Aditya ;
Gupta, Deepak ;
Tiwari, Prayag ;
Moreira, Catarina ;
Damasevicius, Robertas ;
de Albuquerque, Victor Hugo C. .
APPLIED SCIENCES-BASEL, 2020, 10 (02)
[4]   Imaging of pneumonia: trends and algorithms [J].
Franquet, T .
EUROPEAN RESPIRATORY JOURNAL, 2001, 18 (01) :196-208
[5]   Understanding and Exploiting Spatial Properties of System Failures on Extreme-Scale HPC Systems [J].
Gupta, Saurabh ;
Tiwari, Devesh ;
Jantzi, Christopher ;
Rogers, James ;
Maxwell, Don .
2015 45TH ANNUAL IEEE/IFIP INTERNATIONAL CONFERENCE ON DEPENDABLE SYSTEMS AND NETWORKS, 2015, :37-44
[6]  
Han S., 2015, P 29 INT C NEURAL IN
[7]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[8]  
Hinton G., 2015, ARXIV
[9]  
HUANG G, 2017, PROC CVPR IEEE, DOI DOI 10.1109/CVPR.2017.243
[10]  
Hubara I., 2016, P 30 INT C NEUR INF, P4114, DOI DOI 10.5555/3157382.3157557