Context Autoencoder for Self-Supervised Representation Learning

Understand

We present a novel masked image modeling (MIM) approach, context autoencoder (CAE), for self-supervised representation pretraining.

• We pretrain an encoder by making predictions in the encoded representation space.
• The pretraining tasks include two tasks: masked representation prediction - predict the representations for the masked patches, and masked patch reconstruction - reconstruct the masked patches.
• The network is an encoder-regressor-decoder architecture: the encoder takes the visible patches as input; the regressor predicts the representations of the masked patches, which are expected to be aligned with the representations computed from the encoder, using the representations of visible patches and the positions of visible and masked patches; the decoder reconstructs the masked patches from the predicted encoded representations.