指数移動平均正規化(EMAN)と呼ばれるバッチ正規化(BN)のプラグインの代替品を紹介します。これにより、既存の学生教師ベースの自己および半教師あり学習手法のパフォーマンスが向上します。統計が各バッチ内で計算される標準のBNとは異なり、教師で使用されるEMANは、生徒のBN統計からの指数移動平均によって統計を更新します。この設計は、BNの固有のサンプル間の依存関係を減らし、教師の一般化を強化します。 EMANは、ImageNetで1%/ 10%の教師ありラベルが利用できる場合、自己教師あり学習の強力なベースラインを4-6 / 1-2ポイント、半教師あり学習を約7/2ポイント改善します。これらの改善は、メソッド、ネットワークアーキテクチャ、トレーニング期間、およびデータセット全体で一貫しており、この手法の一般的な有効性を示しています。コードはhttps://github.com/amazon-research/exponential-moving-average-normalizationで入手できます。
We present a plug-in replacement for batch normalization (BN) called exponential moving average normalization (EMAN), which improves the performance of existing student-teacher based self- and semi-supervised learning techniques. Unlike the standard BN, where the statistics are computed within each batch, EMAN, used in the teacher, updates its statistics by exponential moving average from the BN statistics of the student. This design reduces the intrinsic cross-sample dependency of BN and enhances the generalization of the teacher. EMAN improves strong baselines for self-supervised learning by 4-6/1-2 points and semi-supervised learning by about 7/2 points, when 1%/10% supervised labels are available on ImageNet. These improvements are consistent across methods, network architectures, training duration, and datasets, demonstrating the general effectiveness of this technique. The code is available at https://github.com/amazon-research/exponential-moving-average-normalization.