Borrowing Treasures from the Wealthy: Deep Transfer Learning through Selective Joint Fine-Tuning

被引:126
作者
Ge, Weifeng [1 ]
Yu, Yizhou [1 ]
机构
[1] Univ Hong Kong, Dept Comp Sci, Hong Kong, Hong Kong, Peoples R China
来源
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017) | 2017年
关键词
D O I
10.1109/CVPR.2017.9
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep neural networks require a large amount of labeled training data during supervised learning. However, collecting and labeling so much data might be infeasible in many cases. In this paper, we introduce a deep transfer learning scheme, called selective joint fine-tuning, for improving the performance of deep learning tasks with insufficient training data. In this scheme, a target learning task with insufficient training data is carried out simultaneously with another source learning task with abundant training data. However, the source learning task does not use all existing training data. Our core idea is to identify and use a subset of training images from the original source learning task whose low-level characteristics are similar to those from the target learning task, and jointly fine-tune shared convolutional layers for both tasks. Specifically, we compute descriptors from linear or nonlinear filter bank responses on training images from both tasks, and use such descriptors to search for a desired subset of training samples for the source learning task. Experiments demonstrate that our deep transfer learning scheme achieves state-of-the-art performance on multiple visual classification tasks with insufficient training data for deep learning. Such tasks include Caltech 256, MIT Indoor 67, and fine-grained classification problems (Oxford Flowers 102 and Stanford Dogs 120). In comparison to fine-tuning without a source domain, the proposed method can improve the classification accuracy by 2% - 10% using a single model. Codes and models are available at https://github.com/ZYYSzj/Selective-Joint-Fine-tuning.
引用
收藏
页码:10 / 19
页数:10
相关论文
共 50 条
[1]  
[Anonymous], 2015, FAST GUIDED FILTER
[2]  
[Anonymous], 2011, P CVPR WORKSH FIN GR
[3]  
[Anonymous], 2007, Multi-Task Feature Learning, DOI DOI 10.7551/MITPRESS/7503.003.0010
[4]  
[Anonymous], PREPRINT ARXIV 1603
[5]  
[Anonymous], PROC CVPR IEEE
[6]  
[Anonymous], 2015, PROC CVPR IEEE
[7]  
[Anonymous], IEEE T PATTERN ANAL
[8]  
[Anonymous], 2014, P 31 INT C INT C MAC
[9]  
[Anonymous], INT C COMP VIS ICCV
[10]  
[Anonymous], 2014, P ADV NEUR INF PROC