arXiv reaDer
TransMatcher:一般化可能な人物の再識別のためのトランスフォーマーを介したディープイメージマッチング
TransMatcher: Deep Image Matching Through Transformers for Generalizable Person Re-identification
トランスフォーマーは最近、コンピュータービジョンでますます注目を集めています。ただし、既存の研究では、画像分類や高密度予測などの特徴表現学習にトランスフォーマーを主に使用しており、トランスフォーマーの一般化可能性は不明です。この作業では、画像のペアが与えられた場合に、画像マッチングとメトリック学習にTransformersを適用する可能性をさらに調査します。ビジョントランスフォーマー(ViT)とデコーダー付きのバニラトランスフォーマーは、画像間の注意が不足しているため、画像のマッチングには不十分であることがわかりました。したがって、さらに2つの単純なソリューション、つまりViTでのクエリギャラリーの連結と、バニラTransformerでのクエリギャラリーのクロスアテンションを設計します。後者はパフォーマンスを向上させますが、それでも制限があります。これは、Transformersのアテンションメカニズムが主にグローバルな特徴の集約のために設計されていることを意味します。これは、画像のマッチングには自然に適していません。したがって、新しい簡略化されたデコーダーを提案します。これは、クエリキーの類似性の計算のみを保持し、softmax重み付けを使用して完全な注意の実装を削除します。さらに、グローバル最大プーリングと多層パーセプトロン(MLP)ヘッドを適用して、マッチング結果をデコードします。このように、簡略化されたデコーダーは計算上より効率的であると同時に、画像マッチングに対してより効果的です。 TransMatcherと呼ばれる提案された方法は、一般化可能な人物の再識別において最先端のパフォーマンスを実現し、いくつかの一般的なデータセットで、ランク1とmAPでそれぞれ最大6.1%と5.7%のパフォーマンス向上を実現します。コードはhttps://github.com/ShengcaiLiao/QAConvで入手できます。
Transformers have recently gained increasing attention in computer vision. However, existing studies mostly use Transformers for feature representation learning, e.g. for image classification and dense predictions, and the generalizability of Transformers is unknown. In this work, we further investigate the possibility of applying Transformers for image matching and metric learning given pairs of images. We find that the Vision Transformer (ViT) and the vanilla Transformer with decoders are not adequate for image matching due to their lack of image-to-image attention. Thus, we further design two naive solutions, i.e. query-gallery concatenation in ViT, and query-gallery cross-attention in the vanilla Transformer. The latter improves the performance, but it is still limited. This implies that the attention mechanism in Transformers is primarily designed for global feature aggregation, which is not naturally suitable for image matching. Accordingly, we propose a new simplified decoder, which drops the full attention implementation with the softmax weighting, keeping only the query-key similarity computation. Additionally, global max pooling and a multilayer perceptron (MLP) head are applied to decode the matching result. This way, the simplified decoder is computationally more efficient, while at the same time more effective for image matching. The proposed method, called TransMatcher, achieves state-of-the-art performance in generalizable person re-identification, with up to 6.1% and 5.7% performance gains in Rank-1 and mAP, respectively, on several popular datasets. Code is available at https://github.com/ShengcaiLiao/QAConv.
updated: Tue Dec 07 2021 07:09:23 GMT+0000 (UTC)
published: Sun May 30 2021 05: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アソシエイト