arXiv reaDer
一般化可能な人物の再識別のためのグラフサンプリングベースのディープメトリック学習
Graph Sampling Based Deep Metric Learning for Generalizable Person Re-Identification
最近の研究によると、明示的な深い特徴のマッチングと大規模で多様なトレーニングデータの両方が、個人の再識別の一般化を大幅に改善できることが示されています。ただし、大規模なデータでディープマッチャーを学習する効率はまだ十分に研究されていません。分類パラメータまたはクラスメモリを使用した学習は一般的な方法ですが、大量のメモリと計算コストが発生します。対照的に、ミニバッチ内のペアワイズディープメトリック学習はより良い選択です。ただし、最も一般的なランダムサンプリング方法であるよく知られているPKサンプラーは、詳細なメトリック学習には有益で効率的ではありません。オンラインのハードサンプルマイニングは学習効率をある程度改善しましたが、ランダムサンプリング後のミニバッチでのマイニングはまだ制限されています。これにより、データサンプリングの段階で、ハードサンプルマイニングの使用を早期に検討するようになります。そのために、本論文では、大規模な深距離計量学習のために、グラフサンプリング(GS)と呼ばれる効率的なミニバッチサンプリング法を提案します。基本的な考え方は、各エポックの開始時にすべてのクラスの最近傍関係グラフを作成することです。次に、各ミニバッチは、ランダムに選択されたクラスとそれに最も近い隣接するクラスで構成され、学習のための有益でやりがいのある例を提供します。適応された競争力のあるベースラインとともに、RandPersonでトレーニングされた場合、MSMT17のランク1で25.1%、一般化可能な人物の再識別の最先端を大幅に改善します。さらに、提案された方法は、MSMT17でトレーニングされた場合、CUHK03-NPのランク1で6.8%、競合ベースラインを上回ります。一方、トレーニング時間は、8,000個のIDを持つRandPersonでトレーニングした場合の25.4時間から2時間に大幅に短縮されます。コードはhttps://github.com/ShengcaiLiao/QAConvで入手できます。
Recent studies show that, both explicit deep feature matching as well as large-scale and diverse training data can significantly improve the generalization of person re-identification. However, the efficiency of learning deep matchers on large-scale data has not yet been adequately studied. Though learning with classification parameters or class memory is a popular way, it incurs large memory and computational costs. In contrast, pairwise deep metric learning within mini batches would be a better choice. However, the most popular random sampling method, the well-known PK sampler, is not informative and efficient for deep metric learning. Though online hard example mining has improved the learning efficiency to some extent, the mining in mini batches after random sampling is still limited. This inspires us to explore the use of hard example mining earlier, in the data sampling stage. To do so, in this paper, we propose an efficient mini-batch sampling method, called graph sampling (GS), for large-scale deep metric learning. The basic idea is to build a nearest neighbor relationship graph for all classes at the beginning of each epoch. Then, each mini batch is composed of a randomly selected class and its nearest neighboring classes so as to provide informative and challenging examples for learning. Together with an adapted competitive baseline, we improve the state of the art in generalizable person re-identification significantly, by 25.1% in Rank-1 on MSMT17 when trained on RandPerson. Besides, the proposed method also outperforms the competitive baseline, by 6.8% in Rank-1 on CUHK03-NP when trained on MSMT17. Meanwhile, the training time is significantly reduced, from 25.4 hours to 2 hours when trained on RandPerson with 8,000 identities. Code is available at https://github.com/ShengcaiLiao/QAConv.
updated: Wed Apr 06 2022 10:52:30 GMT+0000 (UTC)
published: Sun Apr 04 2021 06:44:15 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アソシエイト