Transfer from Task A to Task B

- Tasks A and B have the same input x
- You have a lot more data for Task A than Task B
- Low-level features from A could be helpful for learning B

University of Michigan - Ann Arbor

The longest part of the development of a deep learning model is that of the training stage. However, if the problem that is being tackled has already been studied and worked on by someone else and the model has been released to the public including the trained weights then a technique to substantially increase the training process is to utilize transfer learning.

Transfer Learning in Deep Learning

https://towardsdatascience.com/transfer-learning-using-the-fastai-library-d686b238213e

Transfer Learning Reference

- Need very large dataset
- Data must be general and not domain-specific
- Only works if initial and current problem are similar enough


Disadvantages of Transfer Learning

According to DeepMind CEO Demis Hassabis, transfer learning is also one of the most promising techniques that could lead to artificial general intelligence (AGI) someday.

ideas about transfer learning from the specialist

When Transfer Learning in Deep Learning Makes Sense

- Developing your own model
- Using a pre-trained model

Two Approaches to Transfer Learning in Deep Learning

Transfer learning has several benefits, but the main advantages are  saving training time, better performance of neural networks (in most cases), and not needing a lot of data. 

Usually, a lot of data is needed to train a neural network from scratch but access to that data isn't always available — this is where transfer learning comes in handy. With transfer learning a solid machine learning model can be built with comparatively little training data because the model is already pre-trained. This is especially valuable in natural language processing because mostly expert knowledge is required to create large labeled datasets. Additionally, training time is reduced because it can sometimes take days or even weeks to train a deep neural network from scratch on a complex task.

Advantages of Transfer Learning in Deep Learning

A practical example of fine-tuning involves adapting an image classification model. First, a source model is pre-trained on a large source dataset, such as ImageNet. To adapt this model for a specific task, like classifying images in a smaller target dataset of chairs, a target model is constructed by copying the source model's parameters and replacing its output layer. The new output layer is then trained from scratch on the chair dataset, while the remaining layers are fine-tuned, allowing the target model to leverage the broad visual features learned from the source dataset.

Example of Fine-Tuning for Image Classification

To classify a subset of a large dataset—such as a specific group of classes from ImageNet—it is computationally advantageous to use models pre-trained on the full dataset to extract features. Rather than training a model from scratch or fine-tuning all layers, the pre-trained architecture can be utilized, and only a custom, small-scale output network needs to be trained on the subset. This approach leads to less computational time and memory cost while capitalizing on the robust visual representations already learned by the pre-trained model.

Learn Before

Related