Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Towards Sustainable Self-supervised Learning

About

Although increasingly training-expensive, most self-supervised learning (SSL) models have repeatedly been trained from scratch but not fully utilized, since only a few SOTAs are employed for downstream tasks. In this work, we explore a sustainable SSL framework with two major challenges: i) learning a stronger new SSL model based on the existing pretrained SSL model, also called as "base" model, in a cost-friendly manner, ii) allowing the training of the new model to be compatible with various base models. We propose a Target-Enhanced Conditional (TEC) scheme which introduces two components to the existing mask-reconstruction based SSL. Firstly, we propose patch-relation enhanced targets which enhances the target given by base model and encourages the new model to learn semantic-relation knowledge from the base model by using incomplete inputs. This hardening and target-enhancing help the new model surpass the base model, since they enforce additional patch relation modeling to handle incomplete input. Secondly, we introduce a conditional adapter that adaptively adjusts new model prediction to align with the target of different base models. Extensive experimental results show that our TEC scheme can accelerate the learning speed, and also improve SOTA SSL base models, e.g., MAE and iBOT, taking an explorative step towards sustainable SSL.

Shanghua Gao, Pan Zhou, Ming-Ming Cheng, Shuicheng Yan• 2022

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K (val)
mIoU51
2731
Image ClassificationImageNet-1k (val)
Top-1 Acc85.2
287
Instance SegmentationCOCO
APmask47.2
279
Image ClassificationImageNet 1K (train val)
Top-1 Accuracy86.5
107
Object DetectionCOCO
AP (bbox)54.6
59
Semantic segmentationImageNet-S (val)
mIoU62
21
Showing 6 of 6 rows

Other info

Follow for update