UM  > Faculty of Science and Technology
Residential Collegefalse
Status已發表Published
RIFLE: Backpropagation in depth for deep transfer learning through re-initializing the fully-connected layer
Li,Xingjian1,2; Xiong,Haoyi1; An,Haozhe1; Xu,Chengzhong2,3; Dou,Dejing1
2020
Conference Name37th International Conference on Machine Learning, ICML 2020
Source Publication37th International Conference on Machine Learning, ICML 2020
VolumePartF168147-8
Pages5966-5975
Conference Date13 July 2020 - 18 July 2020
Conference PlaceVirtual, Online
Abstract

Fine-tuning the deep convolution neural network (CNN) using a pre-trained model helps transfer knowledge learned from larger datasets to the target task. While the accuracy could be largely improved even when the training dataset is small, the transfer learning outcome is usually constrained by the pre-trained model with close CNN weights (Liu et al., 2019), as the backpropagation here brings smaller updates to deeper CNN layers. In this work, we propose RI- FLE- a simple yet effective strategy that deepens backpropagation in transfer learning settings, through periodically Re-Initializing the Fullyconnected LayEr with random scratch during the fine-tuning procedure. RIFLE brings meaningful updates to the weights of deep CNN layers and improves low-level feature learning, while the effects of randomization can be easily converged throughout the overall learning procedure. The experiments show that the use of RI- FLE significantly improves deep transfer learning accuracy on a wide range of datasets, outperforming known tricks for the similar purpose, such as Dropout, DropConnect, Stochastic Depth, Disturb Label and Cyclic Learning Rate, under the same settings with 0.5%-2% higher testing accuracy. Empirical cases and ablation studies further indicate RIFLE brings meaningful updates to deep CNN layers with accuracy improved.

URLView the original
Language英語English
Scopus ID2-s2.0-85105596195
Fulltext Access
Citation statistics
Document TypeConference paper
CollectionFaculty of Science and Technology
Corresponding AuthorLi,Xingjian
Affiliation1.Big Data Lab,Baidu Research,Beijing,China
2.Faculty of Science and Technology,University of Macau,Macao
3.State Key Lab of IOTSC,Department of Computer Science,University of Macau,Macao
First Author AffilicationFaculty of Science and Technology
Corresponding Author AffilicationFaculty of Science and Technology
Recommended Citation
GB/T 7714
Li,Xingjian,Xiong,Haoyi,An,Haozhe,et al. RIFLE: Backpropagation in depth for deep transfer learning through re-initializing the fully-connected layer[C], 2020, 5966-5975.
APA Li,Xingjian., Xiong,Haoyi., An,Haozhe., Xu,Chengzhong., & Dou,Dejing (2020). RIFLE: Backpropagation in depth for deep transfer learning through re-initializing the fully-connected layer. 37th International Conference on Machine Learning, ICML 2020, PartF168147-8, 5966-5975.
Files in This Item:
There are no files associated with this item.
Related Services
Recommend this item
Bookmark
Usage statistics
Export to Endnote
Google Scholar
Similar articles in Google Scholar
[Li,Xingjian]'s Articles
[Xiong,Haoyi]'s Articles
[An,Haozhe]'s Articles
Baidu academic
Similar articles in Baidu academic
[Li,Xingjian]'s Articles
[Xiong,Haoyi]'s Articles
[An,Haozhe]'s Articles
Bing Scholar
Similar articles in Bing Scholar
[Li,Xingjian]'s Articles
[Xiong,Haoyi]'s Articles
[An,Haozhe]'s Articles
Terms of Use
No data!
Social Bookmark/Share
All comments (0)
No comment.
 

Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.