arXiv · 2304.05369
A surprisingly simple technique to control the pretraining bias for better transfer: Expand or Narrow your representation
Abstract
Self-Supervised Learning (SSL) models rely on a pretext task to learn representations. Because this pretext task differs from the downstream tasks used to evaluate the performance of these models, there is an inherent misalignment or pretraining bias. A commonly used trick in SSL, shown to make deep networks more robust to such bias, is the addition of a small projector (usually a 2 or 3 layer multi-layer perceptron) on top of a backbone network during training. In contrast to previous work that studied the impact of the projector architecture, we here focus on a simpler, yet overlooked lever to control the information in the backbone representation. We show that merely changing its dimensionality -- by changing only the size of the backbone's very last block -- is a remarkably effective technique to mitigate the pretraining bias. It significantly improves downstream transfer performance for both Self-Supervised and Supervised pretrained models.
Explore related subjects
Keep this discovery
Florian Bordes, Samuel Lavoie, Randall Balestriero, Nicolas Ballas, Pascal Vincent. 2023-04-11. A surprisingly simple technique to control the pretraining bias for better transfer: Expand or Narrow your representation. https://arxiv.org/abs/2304.05369
Cite the original work for its findings. Save a collection to share your selection of sources.