Deep Transfer Learning with Ridge Regression

Abstract

The large amount of online data and vast array of computing resources enable current researchers in both industry and academia to employ the power of deep learning with neural networks. While deep models trained with massive amounts of data demonstrate promising generalisation ability on unseen data from relevant domains, the computational cost of finetuning gradually becomes a bottleneck in transfering the learning to new domains. We address this issue by leveraging the low-rank property of learnt feature vectors produced from deep neural networks (DNNs) with the closed-form solution provided in kernel ridge regression (KRR). This frees transfer learning from finetuning and replaces it with an ensemble of linear systems with many fewer hyperparameters. Our method is successful on supervised and semi-supervised transfer learning tasks.

References in corpus (7)

• Distilling the Knowledge in a Neural Network
• Deep Residual Learning for Image Recognition
• On Calibration of Modern Neural Networks
• Do Deep Nets Really Need to be Deep?
• Algorithms for Learning Kernels Based on Centered Alignment
• Distilling a Neural Network Into a Soft Decision Tree
• Understanding Sparse JL for Feature Hashing