Exploring the Equivalence of Siamese Self-Supervised Learning via A Unified Gradient Framework
Chenxin Tao Honghui Wang Xizhou Zhu Affiliation: Tsinghua University, SenseTime Research, Zhejiang University Jiahua Dong Shiji Song Gao Huang Affiliation: Beijing Academy of Artificial Intelligence, Beijing, China{tcx20, {zhuwalter, Jifeng Dai Affiliation: Beijing Academy of Artificial Intelligence, Beijing, China{tcx20, {zhuwalter,
Abstract
Self-supervised learning has shown its great potential to extract powerful visual representations without human annotations. Various works are proposed to deal with self-supervised learning from different perspectives: (1) contrastive learning methods (e.g., MoCo, SimCLR) utilize both positive and negative samples to guide the training direction; (2) asymmetric network methods (e.g., BYOL, SimSiam) get rid of negative samples via the introduction of a predictor network and the stop-gradient operation; (3) feature decorrelation methods (e.g., Barlow Twins, VICReg) instead aim to reduce the redundancy between feature dimensions. These methods appear to be quite different in the designed loss functions from various motivations. The final accuracy numbers also vary, where different networks and tricks are utilized in different works. In this work, we demonstrate that these methods can be unified into the same form. Instead of comparing their loss functions, we derive a unified formula through gradient analysis. Furthermore, we conduct fair and detailed experiments to compare their performances. It turns out that there is little gap between these methods, and the use of momentum encoder
原文 arXiv:2112.05141;中英对照 + 大白话阅读 https://aha.fim.ai/paper/2112.05141v3