← Back to Data Science

All Topics

Advertisement

Learn/Data Science/Deep Learning

Transfer Learning

Topic: Transfer Learning

Advertisement

Transfer Learning

Use pre-trained models for new tasks.

Approaches

Feature extraction: freeze base, train new classifier. Fine-tuning: unfreeze some layers, train end-to-end.

Full fine-tuning: unfreeze all, lower learning rate.

Image Models

ImageNet pre-trained: VGG, ResNet, Inception, EfficientNet. Keras: applications module.

base_model.trainable = False for feature extraction. Then unfreeze for fine-tuning.

Text Models

BERT, RoBERTa, ALBERT pre-trained. Hugging Face Transformers.

Fine-tune on task-specific data. Lower learning rate than training from scratch.

Key Takeaways

  1. Transfer learning uses pre-trained models
  2. Feature extraction or fine-tuning
  3. Lower learning rate for fine-tuning

Advertisement

Advertisement

Need More Practice?

Get personalized data science help from ChatWhole's AI-powered platform.

Get Expert Help →