Project - Introduction to Transfer Learning (Cat vs Non-cats Project)

3 / 26

Cats vs Non-cats using Transfer Learning - How do we do Transfer Learning?

Transfer learning makes sense when the data in training for task A is quite large and that of task B is relatively smaller.

  • By getting trained on such vast amounts of data and showing excellent performance on its test data, this implies that the neural network has a good knowledge of extracting useful features from the input images. This is essential and powerful for achieving a task.

  • Now that we have such powerful features from these layers (whose weights from task A are frozen), we just need to make use of these extracted features to achieve task B. So, these features from frozen layers are fed to the new layers, and the parameters for these layers are trained on the data of task B.

So basically, we store the knowledge from the previous task in the form of the weights of the frozen layers (called pre-training). Then we make the neural network task B-specific by training (called fine-tuning) the latter layers on the new data.

  • In our project, we will be using pre-trained models which were trained to classify a large set of images(which is the task A), and fine-tune it to make the classifier specific to our classification task of cats vs non-cats(which is task B).

No hints are availble for this assesment

Answer is not availble for this assesment

Loading comments...