arXiv reaDer
TransFG:きめ細かい認識のためのTransformerアーキテクチャ
TransFG: A Transformer Architecture for Fine-grained Recognition
サブカテゴリからオブジェクトを認識することを目的としたきめ細かい視覚分類(FGVC)は、本質的に微妙なクラス間の違いのため、非常に困難な作業です。ほとんどの既存の作業は、主にバックボーンネットワークを再利用して、検出された識別領域の特徴を抽出することにより、この問題に取り組んでいます。ただし、この戦略は必然的にパイプラインを複雑にし、提案された領域にオブジェクトのほとんどの部分を含めるようにプッシュするため、本当に重要な部分を見つけることができません。最近、ビジョントランスフォーマー(ViT)は、従来の分類タスクでその強力なパフォーマンスを示しています。トランスフォーマーの自己注意メカニズムは、すべてのパッチトークンを分類トークンにリンクします。この作業では、最初に、きめ細かい認識設定でのViTフレームワークの有効性を評価します。次に、アテンションリンクの強さに動機付けられて、トークンの重要性の指標として直感的に考えることができます。さらに、トランスフォーマーのすべての生のアテンションウェイトを統合するほとんどのトランスフォーマーアーキテクチャに適用できる新しいパーツ選択モジュールを提案します。識別可能な画像パッチを効果的かつ正確に選択し、それらの関係を計算するようにネットワークを導くための注意マップに。対照的な損失は、紛らわしいクラスの特徴表現間の距離を拡大するために適用されます。拡張トランスベースモデルをTransFGと名付け、最先端のパフォーマンスを実現する5つの人気のあるきめ細かいベンチマークで実験を行うことにより、その価値を実証します。モデルをよりよく理解するために、定性的な結果が示されています。
Fine-grained visual classification (FGVC) which aims at recognizing objects from subcategories is a very challenging task due to the inherently subtle inter-class differences. Most existing works mainly tackle this problem by reusing the backbone network to extract features of detected discriminative regions. However, this strategy inevitably complicates the pipeline and pushes the proposed regions to contain most parts of the objects thus fails to locate the really important parts. Recently, vision transformer (ViT) shows its strong performance in the traditional classification task. The self-attention mechanism of the transformer links every patch token to the classification token. In this work, we first evaluate the effectiveness of the ViT framework in the fine-grained recognition setting. Then motivated by the strength of the attention link can be intuitively considered as an indicator of the importance of tokens, we further propose a novel Part Selection Module that can be applied to most of the transformer architectures where we integrate all raw attention weights of the transformer into an attention map for guiding the network to effectively and accurately select discriminative image patches and compute their relations. A contrastive loss is applied to enlarge the distance between feature representations of confusing classes. We name the augmented transformer-based model TransFG and demonstrate the value of it by conducting experiments on five popular fine-grained benchmarks where we achieve state-of-the-art performance. Qualitative results are presented for better understanding of our model.
updated: Wed Dec 01 2021 22:24:53 GMT+0000 (UTC)
published: Sun Mar 14 2021 17:03:53 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アソシエイト