The Effect of Task Similarity on Deep Transfer Learning

被引:4
作者
Zhang, Wei [1 ]
Fang, Yuchun [1 ]
Ma, Zhengyan [1 ]
机构
[1] Shanghai Univ, Sch Comp Engn & Sci, Shanghai, Peoples R China
来源
NEURAL INFORMATION PROCESSING (ICONIP 2017), PT II | 2017年 / 10635卷
基金
中国国家自然科学基金;
关键词
Deep learning; Transfer learning; CNN;
D O I
10.1007/978-3-319-70096-0_27
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In recent years, with deep learning achieving a great success, deep transfer learning gradually becomes a new issue. Fine-tuning as a simple transfer learning method can be used to help train deep network and improve the performance of network. In our paper, we use two fine-tuning strategies on deep convolutional neural network and compare their results. There are many influencing factors, such as the depth and width of the network, the amount of data, the similarity of the source and target domain, and so on. Then we keep the network structure and other related factors consistent and use the fine fine-tuning strategy to find the effect of cross-domain factor and similarity of task. Specifically, we use source network and target test data to calculate the similarity. The results of experiments show that when we use fine-tune strategy, using different dataset in source and target domain would affect the target task a lot. Besides the similarity of tasks has direction, and to some extent the similarity would reflect the increment of performance of target task when the source and target task use the same dataset.
引用
收藏
页码:256 / 265
页数:10
相关论文
共 8 条
[1]  
[Anonymous], 2016, IEEE T PATTERN ANAL
[2]  
[Anonymous], 2017, ARXIV170208690
[3]   Factors of Transferability for a Generic ConvNet Representation [J].
Azizpour, Hossein ;
Razavian, Ali Sharif ;
Sullivan, Josephine ;
Maki, Atsuto ;
Carlsson, Stefan .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2016, 38 (09) :1790-1802
[4]  
Ding ZM, 2016, INT CONF ACOUST SPEE, P2414, DOI 10.1109/ICASSP.2016.7472110
[5]   ImageNet Classification with Deep Convolutional Neural Networks [J].
Krizhevsky, Alex ;
Sutskever, Ilya ;
Hinton, Geoffrey E. .
COMMUNICATIONS OF THE ACM, 2017, 60 (06) :84-90
[6]  
Long Mingsheng, 2016, ARXIV PREPRINT ARXIV
[7]   A Survey on Transfer Learning [J].
Pan, Sinno Jialin ;
Yang, Qiang .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2010, 22 (10) :1345-1359
[8]  
Szegedy C., 2015, IEEE CVPR 2015