arXiv reaDer
PnP-DETR:トランスフォーマーを使用した効率的な視覚分析に向けて
PnP-DETR: Towards Efficient Visual Analysis with Transformers
最近、DETRはトランスフォーマーを使用した視覚タスクのソリューションを開拓し、画像の特徴マップをオブジェクト検出結果に直接変換します。効果的ではありますが、背景などの一部の領域で計算が冗長になるため、フィーチャマップ全体の変換にコストがかかる可能性があります。この作業では、空間の冗長性を減らすというアイデアを新しいポーリングおよびプール(PnP)サンプリングモジュールにカプセル化します。これにより、計算を空間的に適応的に割り当てて効率を高めるエンドツーエンドのPnP-DETRアーキテクチャを構築します。具体的には、PnPモジュールは、画像の特徴マップを細かい前景オブジェクトの特徴ベクトルと少数の粗い背景のコンテキスト特徴ベクトルに抽象化します。トランスフォーマーは、細かい粗い特徴空間内の情報の相互作用をモデル化し、特徴を検出結果に変換します。さらに、PnP拡張モデルは、既存の方法として複数のモデルをトレーニングする必要なしに、サンプリングされた特徴の長さを変えることにより、単一のモデルでパフォーマンスと計算の間のさまざまな望ましいトレードオフを即座に達成できます。したがって、さまざまな計算制約のあるさまざまなシナリオでの展開に、より高い柔軟性を提供します。さらに、パノプティコンセグメンテーションと最近のトランスベースの画像認識モデルViTでのPnPモジュールの一般化可能性を検証し、一貫した効率の向上を示します。私たちの方法は、空間的な冗長性が一般的に観察されるトランスフォーマーを使用した効率的な視覚分析の一歩を踏み出すと信じています。コードはhttps://github.com/twangnh/pnp-detrで入手できます。
Recently, DETR pioneered the solution of vision tasks with transformers, it directly translates the image feature map into the object detection result. Though effective, translating the full feature map can be costly due to redundant computation on some area like the background. In this work, we encapsulate the idea of reducing spatial redundancy into a novel poll and pool (PnP) sampling module, with which we build an end-to-end PnP-DETR architecture that adaptively allocates its computation spatially to be more efficient. Concretely, the PnP module abstracts the image feature map into fine foreground object feature vectors and a small number of coarse background contextual feature vectors. The transformer models information interaction within the fine-coarse feature space and translates the features into the detection result. Moreover, the PnP-augmented model can instantly achieve various desired trade-offs between performance and computation with a single model by varying the sampled feature length, without requiring to train multiple models as existing methods. Thus it offers greater flexibility for deployment in diverse scenarios with varying computation constraint. We further validate the generalizability of the PnP module on panoptic segmentation and the recent transformer-based image recognition model ViT and show consistent efficiency gain. We believe our method makes a step for efficient visual analysis with transformers, wherein spatial redundancy is commonly observed. Code will be available at https://github.com/twangnh/pnp-detr.
updated: Fri Jan 07 2022 11:51:10 GMT+0000 (UTC)
published: Wed Sep 15 2021 01:10:30 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アソシエイト