Residential College | false |
Status | 已發表Published |
Pay Attention to Features, Transfer Learn Faster CNNs | |
Wang, Kafeng1,2; Gao, Xitong1; Zhao, Yiren3; Li, Xingjian4; Dou, Dejing4; Xu, Cheng-Zhong5 | |
2020-04-30 | |
Conference Name | 8th International Conference on Learning Representations, ICLR 2020 |
Source Publication | 8th International Conference on Learning Representations, ICLR 2020 |
Conference Date | 2020-04-30 |
Conference Place | Addis Ababa |
Abstract | Deep convolutional neural networks are now widely deployed in vision applications, but a limited size of training data can restrict their task performance. Transfer learning offers the chance for CNNs to learn with limited data samples by transferring knowledge from models pretrained on large datasets. Blindly transferring all learned features from the source dataset, however, brings unnecessary computation to CNNs on the target task. In this paper, we propose attentive feature distillation and selection (AFDS), which not only adjusts the strength of transfer learning regularization but also dynamically determines the important features to transfer. By deploying AFDS on ResNet-101, we achieved a state-of-the-art computation reduction at the same accuracy budget, outperforming all existing transfer learning methods. With a 10× MACs reduction budget, a ResNet-101 equipped with AFDS transfer learned from ImageNet to Stanford Dogs 120, can achieve an accuracy 11.07% higher than its best competitor. |
Keyword | Transfer Learning Pruning Faster Cnns |
Language | 英語English |
Scopus ID | 2-s2.0-85150593657 |
Citation statistics | |
Document Type | Conference paper |
Collection | University of Macau |
Corresponding Author | Wang, Kafeng; Gao, Xitong |
Affiliation | 1.Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, China 2.University of Chinese Academy of Sciences, China 3.University of Cambridge, United Kingdom 4.Big Data Lab, Baidu Research 5.University of Macau, Macao |
Recommended Citation GB/T 7714 | Wang, Kafeng,Gao, Xitong,Zhao, Yiren,et al. Pay Attention to Features, Transfer Learn Faster CNNs[C], 2020. |
APA | Wang, Kafeng., Gao, Xitong., Zhao, Yiren., Li, Xingjian., Dou, Dejing., & Xu, Cheng-Zhong (2020). Pay Attention to Features, Transfer Learn Faster CNNs. 8th International Conference on Learning Representations, ICLR 2020. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment