arXiv reaDer
ディープメトリック学習のための階層プロキシベースの損失
Hierarchical Proxy-based Loss for Deep Metric Learning
プロキシベースのメトリック学習損失は、収束が速く、トレーニングの複雑さが低いため、ペアベースの損失よりも優れています。ただし、既存のプロキシベースの損失は、サンプルの記述と照合に役立つ可能性のあるクラス間で共有される共通性を見落としながら、クラス識別機能の学習に焦点を合わせています。さらに、類似した従属クラスをグループ化できる実際のデータセットのカテゴリの暗黙的な階層を無視します。このホワイトペーパーでは、プロキシに階層構造を課すことでこの暗黙の階層を活用し、既存のプロキシベースの損失で使用できるフレームワークを紹介します。これにより、モデルは、暗黙的なデータ階層を壊すことなく、クラス識別機能とクラス共有特性の両方をキャプチャできます。インショップやSOPなどの5つの確立された画像検索データセットで私たちの方法を評価します。結果は、階層型プロキシベースの損失フレームワークが、特に強力な階層構造を示す大規模なデータセットで、既存のプロキシベースの損失のパフォーマンスを向上させることを示しています。
Proxy-based metric learning losses are superior to pair-based losses due to their fast convergence and low training complexity. However, existing proxy-based losses focus on learning class-discriminative features while overlooking the commonalities shared across classes which are potentially useful in describing and matching samples. Moreover, they ignore the implicit hierarchy of categories in real-world datasets, where similar subordinate classes can be grouped together. In this paper, we present a framework that leverages this implicit hierarchy by imposing a hierarchical structure on the proxies and can be used with any existing proxy-based loss. This allows our model to capture both class-discriminative features and class-shared characteristics without breaking the implicit data hierarchy. We evaluate our method on five established image retrieval datasets such as In-Shop and SOP. Results demonstrate that our hierarchical proxy-based loss framework improves the performance of existing proxy-based losses, especially on large datasets which exhibit strong hierarchical structure.
updated: Wed Aug 18 2021 21:21:02 GMT+0000 (UTC)
published: Thu Mar 25 2021 00:38: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アソシエイト