arrow
Return

On automated source selection for transfer learning in convolutional neural networks

delete2018-01-01
delete79
delete
OA
AI
A
Arun Ross
E
Erik M. Shapiro
DOI:10.1016/j.patcog.2017.07.019delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Transfer learning, or inductive transfer, refers to the transfer of knowledge from a source task to a target task. In the context of convolutional neural networks (CNNs), transfer learning can be implemented by transplanting the learned feature layers from one CNN (derived from the source task) to initialize another (for the target task). Previous research has shown that the choice of the source CNN impacts the performance of the target task. In the current literature, there is no principled way for selecting a source CNN for a given target task despite the increasing availability of pre-trained source CNNs. In this paper we investigate the possibility of automatically ranking source CNNs prior to utilizing them for a target task. In particular, we present an information theoretic framework to understand the source-target relationship and use this as a basis to derive an approach to automatically rank source CNNs in an efficient, zero shot manner. The practical utility of the approach is thoroughly evaluated using the Places-MIT dataset, MNIST dataset and a real-world MRI database. Experimental results demonstrate the efficacy of the proposed ranking method for transfer learning. (C) 2017 Elsevier Ltd. All rights reserved.
Keywords:
Transfer learning
CNN selection
Deep learning
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Pattern Recognition cover
Pattern Recognition
IF:
7.6
Papers:
1.3W
Citations:
4.5W

Organization

M
michigan state university
Scholars:
3.6W
Papers: 3.2W
Citations: 44