深いメトリック学習は、サンプルを異なるクラスにグループ化するために使用できる高度に識別可能な特徴の埋め込みを取得するためにニューラルネットワークを活用することにより、クラスタリングや画像検索などのタスクで印象的な結果をもたらしました。多くの研究が、そのようなネットワークをトレーニングするためのスマート損失関数またはデータマイニング戦略の設計に向けられてきました。ほとんどの方法では、損失関数を計算するために、ミニバッチ内のサンプルのペアまたはトリプレットのみが考慮されます。損失関数は、通常、埋め込み間の距離に基づいています。微分可能ラベル伝播法に基づく損失関数であるグループ損失を提案します。これは、グループのすべてのサンプルに類似性を埋め込み、同時に、異なるグループに属するデータポイント間の低密度領域を促進します。 「類似したオブジェクトは同じグループに属する必要がある」という滑らかさの仮定に基づいて、提案された損失は、分類タスクのためにニューラルネットワークをトレーニングし、クラス内のサンプル間で一貫したラベル付けを実施します。モデルの結果をさらに改善する、GroupLoss++という名前のアルゴリズムに合わせた一連の推論戦略を設計します。 4つの検索データセットでのクラスタリングと画像検索に関する最新の結果を示し、2人の再識別データセットで競争力のある結果を示し、検索と再識別のための統一されたフレームワークを提供します。
Deep metric learning has yielded impressive results in tasks such as clustering and image retrieval by leveraging neural networks to obtain highly discriminative feature embeddings, which can be used to group samples into different classes. Much research has been devoted to the design of smart loss functions or data mining strategies for training such networks. Most methods consider only pairs or triplets of samples within a mini-batch to compute the loss function, which is commonly based on the distance between embeddings. We propose Group Loss, a loss function based on a differentiable label-propagation method that enforces embedding similarity across all samples of a group while promoting, at the same time, low-density regions amongst data points belonging to different groups. Guided by the smoothness assumption that "similar objects should belong to the same group", the proposed loss trains the neural network for a classification task, enforcing a consistent labelling amongst samples within a class. We design a set of inference strategies tailored towards our algorithm, named Group Loss++ that further improve the results of our model. We show state-of-the-art results on clustering and image retrieval on four retrieval datasets, and present competitive results on two person re-identification datasets, providing a unified framework for retrieval and re-identification.