wAIpoint
Italiano

Unsupervised & transfer learning

Level: Advanced

What it is

Unsupervised learning finds structure in data without labels. Transfer learning reuses what a model learned on one big task as the starting point for another. Together they are the idea behind pre-training: learn from huge amounts of unlabelled data first, then adapt.

What it is for

When you have few labelled examples, start from a model pre-trained on lots of data (a language model, an image model) and adapt it. This is how most practical AI is built today.

How to do it

The process, step by step. Libraries and tools change; these steps stay.

  1. Find a pre-trained model for your kind of data (text, images, audio) on a model hub.
  2. Decide how much to change: use it as it is, train only a small new part on top, or fine-tune all of it.
  3. Prepare your small labelled dataset.
  4. Train with a small learning rate, so you adapt the model without erasing what it already knows.
  5. Evaluate on data from your real use case: if it is very different from the pre-training data, expect to need more care.

Timeline

  1. 2018BERT: pre-training deep bidirectional transformers on unlabelled text, then adapting them (Devlin et al.). source

Sources