An Empirical Study on Improving SimCLR's Nonlinear Projection Head using Pretrained Autoencoder Embeddings

TL;DR

This study enhances SimCLR's nonlinear projection head by integrating pretrained autoencoder embeddings, leading to improved classification accuracy and reduced projection dimensionality, with architectural modifications and activation function optimizations.

Contribution

It introduces a novel approach of using pretrained autoencoder embeddings in SimCLR's projection head, demonstrating improved performance and efficiency over standard methods.

Findings

Up to 2.9% increase in classification accuracy

Reduced dimensionality of projection space

Sigmoid and tanh outperform ReLU in accuracy

Abstract

This paper focuses on improving the effectiveness of the standard 2-layer MLP projection head featured in the SimCLR framework through the use of pretrained autoencoder embeddings. Given a contrastive learning task with a largely unlabeled image classification dataset, we first train a shallow autoencoder architecture and extract its compressed representations contained in the encoder's embedding layer. After freezing the weights within this pretrained layer, we use it as a drop-in replacement for the input layer of SimCLR's default projector. Additionally, we also apply further architectural changes to the projector by decreasing its width and changing its activation function. The different projection heads are then used to contrastively train and evaluate a feature extractor following the SimCLR protocol. Our experiments indicate that using a pretrained autoencoder embedding in the projector can not only increase classification accuracy by up to 2.9% or 1.7% on average, but can also significantly decrease the dimensionality of the projection space. Our results also suggest, that using the sigmoid and tanh activation functions within the projector can outperform ReLU in terms of peak and average classification accuracy. All experiments involving our pretrained projectors are conducted with frozen embeddings, since our test results indicate an advantage compared to using their non-frozen counterparts.