arXiv reaDer
大きなバッチと類似性の取り違えによるRecall @ k代理損失
Recall@k Surrogate Loss with Large Batches and Similarity Mixup
最急降下法による評価メトリックの直接最適化は、それが微分不可能である場合には不可能です。これは、検索でのリコールの場合です。この作業では、リコールの微分可能な代理損失が提案されます。 GPUメモリのハードウェア制約を回避する実装を使用して、メソッドは非常に大きなバッチサイズでトレーニングします。これは、取得データベース全体で計算されるメトリックに不可欠です。これは、ペアワイズスカラーの類似性に基づいて動作し、バッチサイズを実質的にさらに増加させる効率的なミックスアップアプローチによって支援されます。深いメトリック学習に使用される場合、提案された方法は、いくつかの画像検索ベンチマークで最先端の結果を達成します。インスタンスレベルの認識の場合、このメソッドは、平均精度の近似を使用してトレーニングする同様のアプローチよりも優れています。実装は公開されます。
Direct optimization, by gradient descent, of an evaluation metric, is not possible when it is non-differentiable, which is the case for recall in retrieval. In this work, a differentiable surrogate loss for the recall is proposed. Using an implementation that sidesteps the hardware constraints of the GPU memory, the method trains with a very large batch size, which is essential for metrics computed on the entire retrieval database. It is assisted by an efficient mixup approach that operates on pairwise scalar similarities and virtually increases the batch size further. When used for deep metric learning, the proposed method achieves state-of-the-art results in several image retrieval benchmarks. For instance-level recognition, the method outperforms similar approaches that train using an approximation of average precision. The implementation will be made public.
updated: Wed Aug 25 2021 11:09:11 GMT+0000 (UTC)
published: Wed Aug 25 2021 11:09:11 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アソシエイト