SpotTune: Transfer Learning Through Adaptive Fine-Tuning
University of California San Diego · Google (United States) · +3 more institutions
Abstract
Transfer learning, which allows a source task to affect the inductive bias of the target task, is widely used in computer vision. The typical way of conducting transfer learning with deep neural networks is to fine-tune a model pretrained on the source task using data from the target task. In this paper, we propose an adaptive fine-tuning approach, called SpotTune, which finds the optimal fine-tuning strategy per instance for the target data. In SpotTune, given an image from the target task, a policy network is used to make routing decisions on whether to pass the image through the fine-tuned layers or the pre-trained layers. We conduct extensive experiments to demonstrate the effectiveness of the proposed…
Citation impact
- FWCI
- 31.07
- Percentile
- 100%
- References
- 96
Authors
6Topics & keywords
- Computer science
- Task (project management)
- Fine-tuning
- Transfer of learning
- Artificial intelligence
- Artificial neural network
- Multi-task learning
- Machine learning