arXiv reaDer
教師なし個人の再識別のためのクラスターコントラスト
Cluster Contrast for Unsupervised Person Re-Identification
最先端の教師なしre-ID法は、メモリベースのノンパラメトリックソフトマックス損失を使用してニューラルネットワークをトレーニングします。メモリに保存されたインスタンス特徴ベクトルには、クラスタリングによって疑似ラベルが割り当てられ、インスタンスレベルで更新されます。ただし、クラスターサイズが異なると、各クラスターの更新の進行状況に一貫性がなくなります。この問題を解決するために、特徴ベクトルを格納し、クラスターレベルでコントラスト損失を計算するクラスターコントラストを提示します。私たちのアプローチでは、一意のクラスター表現を使用して各クラスターを記述し、クラスターレベルのメモリディクショナリを作成します。このようにして、クラスタリングの一貫性をパイプライン全体で効果的に維持し、GPUメモリの消費を大幅に削減できます。したがって、私たちの方法は、クラスターの不整合の問題を解決し、より大きなデータセットに適用できます。さらに、フレームワークの堅牢性と一般化を実証するために、さまざまなクラスタリングアルゴリズムを採用しています。 Cluster Contrastを標準の教師なしre-IDパイプラインに適用すると、最新の純粋な教師なしre-IDメソッドおよび5.5%、4.8%、4.4%mAPと比較して、9.9%、8.3%、12.1%の大幅な改善が達成されます。 Market、Duke、およびMSMT17データセットの最先端の教師なしドメイン適応re-IDメソッドと比較。コードはhttps://github.com/alibaba/cluster-contrastで入手できます。
State-of-the-art unsupervised re-ID methods train the neural networks using a memory-based non-parametric softmax loss. Instance feature vectors stored in memory are assigned pseudo-labels by clustering and updated at instance level. However, the varying cluster sizes leads to inconsistency in the updating progress of each cluster. To solve this problem, we present Cluster Contrast which stores feature vectors and computes contrast loss at the cluster level. Our approach employs a unique cluster representation to describe each cluster, resulting in a cluster-level memory dictionary. In this way, the consistency of clustering can be effectively maintained throughout the pipline and the GPU memory consumption can be significantly reduced. Thus, our method can solve the problem of cluster inconsistency and be applicable to larger data sets. In addition, we adopt different clustering algorithms to demonstrate the robustness and generalization of our framework. The application of Cluster Contrast to a standard unsupervised re-ID pipeline achieves considerable improvements of 9.9%, 8.3%, 12.1% compared to state-of-the-art purely unsupervised re-ID methods and 5.5%, 4.8%, 4.4% mAP compared to the state-of-the-art unsupervised domain adaptation re-ID methods on the Market, Duke, and MSMT17 datasets. Code is available at https://github.com/alibaba/cluster-contrast.
updated: Fri Feb 10 2023 08:09:54 GMT+0000 (UTC)
published: Mon Mar 22 2021 03:41:19 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アソシエイト