1. 首页
  2. 人工智能
  3. 论文/代码
  4. 动量^ 2老师:动量老师具有动量统计的自我指导学习

动量^ 2老师:动量老师具有动量统计的自我指导学习

上传者: 2021-01-22 01:15:24上传 .PDF文件 533.33 KB 热度 23次

在本文中,我们提出了一种新颖的方法,即动量 2 老师,用于基于学生老师的自我监督学习。该方法对网络权重和批次归一化(BN)统计信息执行动量更新。..

Momentum^2 Teacher: Momentum Teacher with Momentum Statistics for Self-Supervised Learning

In this paper, we present a novel approach, Momentum$^2$ Teacher, for student-teacher based self-supervised learning. The approach performs momentum update on both network weights and batch normalization (BN) statistics.The teacher's weight is a momentum update of the student, and the teacher's BN statistics is a momentum update of those in history. The Momentum$^2$ Teacher is simple and efficient. It can achieve the state of the art results (74.5\%) under ImageNet linear evaluation protocol using small-batch size(\eg, 128), without requiring large-batch training on special hardware like TPU or inefficient across GPU operation (\eg, shuffling BN, synced BN). Our implementation and pre-trained models will be given on GitHub\footnote{https://github.com/zengarden/momentum2-teacher}.

下载地址
用户评论