Does Optimal Source Task Performance Imply Optimal Pre-training for a Target Task?
Machine Learning
2022-04-13 v2 Computer Vision and Pattern Recognition
Abstract
Fine-tuning of pre-trained deep nets is commonly used to improve accuracies and training times for neural nets. It is generally assumed that pre-training a net for optimal source task performance best prepares it for fine-tuning to learn an arbitrary target task. This is generally not true. Stopping source task training, prior to optimal performance, can create a pre-trained net better suited for fine-tuning to learn a new task. We perform several experiments demonstrating this effect, as well as the influence of the amount of training and of learning rate. Additionally, our results indicate that this reflects a general loss of learning ability that even extends to relearning the source task.
Cite
@article{arxiv.2106.11174,
title = {Does Optimal Source Task Performance Imply Optimal Pre-training for a Target Task?},
author = {Steven Gutstein and Brent Lance and Sanjay Shakkottai},
journal= {arXiv preprint arXiv:2106.11174},
year = {2022}
}