arXiv reaDer
マルチラベル画像分類のためのプライマルオブジェクトクエリを備えたビジュアルトランスフォーマー
Visual Transformers with Primal Object Queries for Multi-Label Image Classification
マルチラベル画像分類は、順序のないシーケンシャルデータと見なすことができるクラスラベルのセットを予測することです。トランスフォーマーはシーケンシャルデータ全体を処理するため、本質的にセット予測に優れています。オブジェクト検出タスク用に提案された最初のビジョンベースのトランスフォーマーモデルは、オブジェクトクエリの概念を導入しました。オブジェクトクエリは、画像内の関心領域を使用してオブジェクトクラスまたはバウンディングボックスをデコードするためにデコーダーレイヤーのアテンションモジュールによって使用される学習可能な位置エンコーディングです。ただし、同じオブジェクトクエリのセットを異なるデコーダーレイヤーに入力すると、トレーニングが妨げられます。その結果、パフォーマンスが低下し、収束が遅れます。この論文では、トランスフォーマデコーダスタックの開始時にのみ提供されるプライマリオブジェクトクエリの使用法を提案します。さらに、マルチラベル分類のために提案された混合手法を改善します。プライマリオブジェクトクエリを使用して提案されたトランスフォーマーモデルは、最先端のクラスごとのF1メトリックを2.1%および1.8%改善します。また、MS-COCOデータセットとNUS-WIDEデータセットでそれぞれ79.0%と38.6%の収束を高速化します。
Multi-label image classification is about predicting a set of class labels that can be considered as orderless sequential data. Transformers process the sequential data as a whole, therefore they are inherently good at set prediction. The first vision-based transformer model, which was proposed for the object detection task introduced the concept of object queries. Object queries are learnable positional encodings that are used by attention modules in decoder layers to decode the object classes or bounding boxes using the region of interests in an image. However, inputting the same set of object queries to different decoder layers hinders the training: it results in lower performance and delays convergence. In this paper, we propose the usage of primal object queries that are only provided at the start of the transformer decoder stack. In addition, we improve the mixup technique proposed for multi-label classification. The proposed transformer model with primal object queries improves the state-of-the-art class wise F1 metric by 2.1% and 1.8%; and speeds up the convergence by 79.0% and 38.6% on MS-COCO and NUS-WIDE datasets respectively.
updated: Sun May 15 2022 10:21:09 GMT+0000 (UTC)
published: Fri Dec 10 2021 12:29:07 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アソシエイト