English

Towards Sustainable Self-supervised Learning

Computer Vision and Pattern Recognition 2022-10-21 v1

Abstract

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.

Keywords

Cite

@article{arxiv.2210.11016,
  title  = {Towards Sustainable Self-supervised Learning},
  author = {Shanghua Gao and Pan Zhou and Ming-Ming Cheng and Shuicheng Yan},
  journal= {arXiv preprint arXiv:2210.11016},
  year   = {2022}
}

Comments

Extension of NeurIPS 2022 workshop

R2 v1 2026-06-28T04:03:24.874Z