arXiv reaDer
補助回転損失による教師なしディープメトリックラーニング
Unsupervised Deep Metric Learning via Auxiliary Rotation Loss
 ディープメトリックラーニングは、画像検索や個人の再識別など、多くのドメインに適用できるため、重要な分野です。このようなモデルの主な欠点は、ラベル付きデータが必要なことです。この作業では、クラスタリングの割り当てから直接ディープメトリック学習用の擬似ラベルを生成することを提案し、自己監視(SS)タスクによって正則化された教師なしディープメトリック学習(UDML)を導入します。特に、画像の回転を予測することにより、トレーニングプロセスを正規化することを提案します。私たちの方法(UDML-SS)は、識別的埋め込み、埋め込みの教師なしクラスタリング割り当て、および自己教師付き口実タスクを共同で学習します。 UDML-SSは、従来のクラスタリングアルゴリズム(k-meansなど)を使用して埋め込みを繰り返しクラスタリングし、メトリック学習のためのクラスター割り当てに基づいてトレーニングペアをサンプリングします。同時に、マルチタスク方式で自己監視プリテキストタスクを最適化します。自己監視の役割は、トレーニングプロセスを安定させることであり、モデルが信頼性の低いクラスタリング割り当てのために歪んでいない意味のある特徴表現を学習することを奨励します。提案された方法は、メトリック学習の標準ベンチマークで優れた性能を発揮し、現在の最先端のアプローチを大幅に上回るパフォーマンスを発揮し、さまざまなメトリック学習損失関数との競合パフォーマンスも示します。
Deep metric learning is an important area due to its applicability to many domains such as image retrieval and person re-identification. The main drawback of such models is the necessity for labeled data. In this work, we propose to generate pseudo-labels for deep metric learning directly from clustering assignment and we introduce unsupervised deep metric learning (UDML) regularized by a self-supervision (SS) task. In particular, we propose to regularize the training process by predicting image rotations. Our method (UDML-SS) jointly learns discriminative embeddings, unsupervised clustering assignments of the embeddings, as well as a self-supervised pretext task. UDML-SS iteratively cluster embeddings using traditional clustering algorithm (e.g., k-means), and sampling training pairs based on the cluster assignment for metric learning, while optimizing self-supervised pretext task in a multi-task fashion. The role of self-supervision is to stabilize the training process and encourages the model to learn meaningful feature representations that are not distorted due to unreliable clustering assignments. The proposed method performs well on standard benchmarks for metric learning, where it outperforms current state-of-the-art approaches by a large margin and it also shows competitive performance with various metric learning loss functions.
updated: Sat Nov 16 2019 18:28:45 GMT+0000 (UTC)
published: Sat Nov 16 2019 18:28:45 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アソシエイト