HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

MV-MR: multi-views and multi-representations for self-supervised learning and knowledge distillation

Vitaliy Kinakh; Mariia Drozdova; Slava Voloshynovskiy

MV-MR: multi-views and multi-representations for self-supervised learning and knowledge distillation

Abstract

We present a new method of self-supervised learning and knowledge distillation based on the multi-views and multi-representations (MV-MR). The MV-MR is based on the maximization of dependence between learnable embeddings from augmented and non-augmented views, jointly with the maximization of dependence between learnable embeddings from augmented view and multiple non-learnable representations from non-augmented view. We show that the proposed method can be used for efficient self-supervised classification and model-agnostic knowledge distillation. Unlike other self-supervised techniques, our approach does not use any contrastive learning, clustering, or stop gradients. MV-MR is a generic framework allowing the incorporation of constraints on the learnable embeddings via the usage of image multi-representations as regularizers. Along this line, knowledge distillation is considered a particular case of such a regularization. MV-MR provides the state-of-the-art performance on the STL10 and ImageNet-1K datasets among non-contrastive and clustering-free methods. We show that a lower complexity ResNet50 model pretrained using proposed knowledge distillation based on the CLIP ViT model achieves state-of-the-art performance on STL10 linear evaluation. The code is available at: https://github.com/vkinakh/mv-mr

Code Repositories

vkinakh/mv-mr
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
self-supervised-image-classification-onMV-MR
Top 1 Accuracy: 74.5%
Top 5 Accuracy: 92.1
self-supervised-learning-on-stl-10MV-MR
Accuracy: 89.67
unsupervised-image-classification-on-cifar-20MV-MR
Accuracy: 73.2
unsupervised-image-classification-on-stl-10MV-MR
Accuracy: 89.67

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp