KNOWLEDGE DISTILLATION INSPIRED FINE-TUNING OF TUCKER DECOMPOSED CNNs AND ADVERSARIAL ROBUSTNESS ANALYSIS

被引:0
作者
Sadhukhan, Ranajoy [1 ]
Saha, Avinab [1 ]
Mukhopadhyay, Jayanta [1 ]
Patra, Amit [1 ]
机构
[1] Indian Inst Technol Kharagpur, Kharagpur, W Bengal, India
来源
2020 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP) | 2020年
关键词
Tucker Decomposition; Accelerating CNNs; Network Decomposition; Knowledge Distillation; Adversarial Robustness;
D O I
10.1109/icip40778.2020.9190672
中图分类号
TB8 [摄影技术];
学科分类号
0804 ;
摘要
The recent works in tensor decomposition of convolutional neural networks have paid little attention to fine-tuning the decomposed models more effectively. We propose to improve the accuracy as well as the adversarial robustness of decomposed networks over existing non-iterative methods by distilling knowledge from the computationally intensive undecomposed (teacher) model to the decomposed (student) model. Through a series of experiments, we demonstrate the effectiveness of knowledge distillation with different loss functions and compare it to the existing fine-tuning strategy of minimizing cross-entropy loss with ground truth labels. Finally, we conclude that the student networks obtained by the proposed approach are superior not only in terms of accuracy but also adversarial robustness, which is often compromised in the existing methods.
引用
收藏
页码:1876 / 1880
页数:5
相关论文
共 22 条
[1]  
[Anonymous], 2014, CORR
[2]  
[Anonymous], 2014, Speeding up convolutional neural networks with low rank expansions
[3]  
Ba L. J., 2013, CORR
[4]  
Denton E, 2014, ADV NEUR IN, V27
[5]   More is Less: A More Complicated Network with Less Inference Complexity [J].
Dong, Xuanyi ;
Huang, Junshi ;
Yang, Yi ;
Yan, Shuicheng .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :1895-1903
[6]   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
[7]  
Heo B., 2018, CORR
[8]  
Hinton G., 2009, Handbook of Systemic Autoimmune Diseases
[9]  
Hinton G., 2015, ARXIV150302531
[10]  
Kim Y.-D., 2015, ARXIV PREPRINT ARXIV