arXiv reaDer
対照的な損失の単純で効果的なバランス
Simple and Effective Balance of Contrastive Losses
対照的な損失は、長い間、深い計量学習の重要な要素であり、自己監視学習の成功により、現在、より一般的になっています。最近の研究では、このような損失を、表現ネットワークを学習するときに補完的に機能する2つのサブ損失(正の項とエントロピーの項)に分解することの利点が示されています。したがって、全体的な損失は2つの用語の組み合わせとして定義されますが、これら2つの用語のバランスは、実装の詳細の背後に隠れていることが多く、実際にはほとんど無視され、最適ではありません。この作業では、ハイパーパラメータ最適化問題として対照損失のバランスにアプローチし、評価パフォーマンスを最適化するハイパーパラメータを効率的に見つける座標降下ベースの検索方法を提案します。このプロセスでは、既存の残高分析を対照的なマージン損失に拡張し、残高にバッチサイズを含め、バッチから損失要素を集計して、より広い範囲のバッチサイズでほぼ最適なパフォーマンスを維持する方法を説明します。ディープメトリック学習と自己監視学習のベンチマークを使用した広範な実験では、他の一般的な検索方法よりも、この方法の方が最適なハイパーパラメーターがすばやく見つかることが示されています。
Contrastive losses have long been a key ingredient of deep metric learning and are now becoming more popular due to the success of self-supervised learning. Recent research has shown the benefit of decomposing such losses into two sub-losses which act in a complementary way when learning the representation network: a positive term and an entropy term. Although the overall loss is thus defined as a combination of two terms, the balance of these two terms is often hidden behind implementation details and is largely ignored and sub-optimal in practice. In this work, we approach the balance of contrastive losses as a hyper-parameter optimization problem, and propose a coordinate descent-based search method that efficiently find the hyper-parameters that optimize evaluation performance. In the process, we extend existing balance analyses to the contrastive margin loss, include batch size in the balance, and explain how to aggregate loss elements from the batch to maintain near-optimal performance over a larger range of batch sizes. Extensive experiments with benchmarks from deep metric learning and self-supervised learning show that optimal hyper-parameters are found faster with our method than with other common search methods.
updated: Wed Dec 22 2021 09:23:27 GMT+0000 (UTC)
published: Wed Dec 22 2021 09:23:27 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アソシエイト