arXiv reaDer
大規模なミニバッチ、一貫性、ハードネガティブマイニングによる教師なし学習のためのメモリバンクの改善
Improving memory banks for unsupervised learning with large mini-batch, consistency and hard negative mining
インスタンスベースの識別による教師なし学習の重要なコンポーネントは、データセット内の各トレーニングサンプルの特徴表現を格納するためのメモリバンクです。このホワイトペーパーでは、バニラメモリバンクベースの定式化に3つの改善点を導入し、大幅な精度の向上をもたらします。(a)大規模なミニバッチ:同じバッチ内の各サンプルに対して複数の拡張をプルし、これがより良いモデルにつながることを示します。強化されたメモリバンクの更新。 (b)一貫性:以前のアプローチで提案されたように、ネガティブサンプルに関して識別を強制しようとせずに、同じサンプルの異なる拡張によって取得されたロジットを近くに強制します。 (c)ハードネガティブマイニング:インスタンスの識別は視覚的に類似しすぎるサンプルには意味がないため、以前はインスタンスによって分離を余儀なくされた非常に類似したデータサンプルを徐々にマージするメモリバンクを改善するための新しい最近傍アプローチを考案しますレベル分類の損失。全体として、私たちのアプローチは、バニラメモリバンクベースのインスタンス識別を大幅に改善し、コサイン類似性を備えた表示および非表示の両方のテストカテゴリの既存のすべての方法よりも優れています。
An important component of unsupervised learning by instance-based discrimination is a memory bank for storing a feature representation for each training sample in the dataset. In this paper, we introduce 3 improvements to the vanilla memory bank-based formulation which brings massive accuracy gains: (a) Large mini-batch: we pull multiple augmentations for each sample within the same batch and show that this leads to better models and enhanced memory bank updates. (b) Consistency: we enforce the logits obtained by different augmentations of the same sample to be close without trying to enforce discrimination with respect to negative samples as proposed by previous approaches. (c) Hard negative mining: since instance discrimination is not meaningful for samples that are too visually similar, we devise a novel nearest neighbour approach for improving the memory bank that gradually merges extremely similar data samples that were previously forced to be apart by the instance level classification loss. Overall, our approach greatly improves the vanilla memory-bank based instance discrimination and outperforms all existing methods for both seen and unseen testing categories with cosine similarity.
updated: Mon Feb 08 2021 18:56:33 GMT+0000 (UTC)
published: Mon Feb 08 2021 18:56:33 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト