arXiv reaDer
メトリック学習のための適応クロスバッチ正規化
Adaptive Cross Batch Normalization for Metric Learning
メトリック学習はコンピューター ビジョンの基本的な問題であり、ランキング ロスを介して意味的に有用な埋め込み空間を学習するようにモデルをトレーニングします。伝統的に、ランキングの損失の効果はミニバッチのサイズに依存するため、基盤となるハードウェアのメモリの制約によって本質的に制限されます。ミニバッチ全体で埋め込みを単純に蓄積することは有用であることが証明されていますが (Wang et al. [2020])、蓄積された埋め込みが最新であることを確認することも同様に重要であることを示しています。特に、学習可能なパラメーターが更新されているため、現在のトレーニング反復で蓄積された埋め込みと特徴埋め込みの間の表現上のドリフトを回避する必要があります。この論文では、表現のドリフトを分布の不整合としてモデル化し、モーメントマッチングを使用してそれに取り組みます。その結果、保存された埋め込みを更新して、トレーニングの反復ごとに現在の埋め込みの最初と 2 番目の瞬間に一致する簡単な方法が得られます。 SOP、In-Shop、DeepFashion2 という 3 つの一般的な画像検索データセットでの実験では、すべてのシナリオでこのアプローチがパフォーマンスを大幅に向上させることが示されています。
Metric learning is a fundamental problem in computer vision whereby a model is trained to learn a semantically useful embedding space via ranking losses. Traditionally, the effectiveness of a ranking loss depends on the minibatch size, and is, therefore, inherently limited by the memory constraints of the underlying hardware. While simply accumulating the embeddings across minibatches has proved useful (Wang et al. [2020]), we show that it is equally important to ensure that the accumulated embeddings are up to date. In particular, it is necessary to circumvent the representational drift between the accumulated embeddings and the feature embeddings at the current training iteration as the learnable parameters are being updated. In this paper, we model representational drift as distribution misalignment and tackle it using moment matching. The result is a simple method for updating the stored embeddings to match the first and second moments of the current embeddings at each training iteration. Experiments on three popular image retrieval datasets, namely, SOP, In-Shop, and DeepFashion2, demonstrate that our approach significantly improves the performance in all scenarios.
updated: Thu Mar 30 2023 03:22:52 GMT+0000 (UTC)
published: Thu Mar 30 2023 03:22:52 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アソシエイト