arXiv reaDer
多段階のクロススケールの注意による視覚バックボーンの強化
Vision Backbone Enhancement via Multi-Stage Cross-Scale Attention
畳み込みニューラル ネットワーク (CNN) とビジョン トランスフォーマー (ViT) は、さまざまなビジョン タスクで目覚ましい成功を収めています。ただし、多くのアーキテクチャでは、さまざまなステージやスケールの機能マップ間の相互作用が考慮されていないため、パフォーマンスが制限される可能性があります。この研究では、多段階かつクロススケールのインタラクションを通じてこれらの制限を克服するための、シンプルなアドオン アテンション モジュールを提案します。具体的には、提案された多段階クロススケール アテンション (\meth) モジュールは、さまざまな段階から特徴マップを取得して多段階インタラクションを可能にし、多段階特徴マップに基づいてさまざまなスケールで自己注意を計算することでクロススケール インタラクションを実現します。 。いくつかのダウンストリーム タスクに関する実験では、\meth~ が適度な追加の FLOP と実行時間で大幅なパフォーマンス向上を実現することを示しています。
Convolutional neural networks (CNNs) and vision transformers (ViTs) have achieved remarkable success in various vision tasks. However, many architectures do not consider interactions between feature maps from different stages and scales, which may limit their performance. In this work, we propose a simple add-on attention module to overcome these limitations via multi-stage and cross-scale interactions. Specifically, the proposed Multi-Stage Cross-Scale Attention (\meth) module takes feature maps from different stages to enable multi-stage interactions and achieves cross-scale interactions by computing self-attention at different scales based on the multi-stage feature maps. Our experiments on several downstream tasks show that \meth~provides a significant performance boost with modest additional FLOPs and runtime.
updated: Thu Aug 10 2023 22:57:31 GMT+0000 (UTC)
published: Thu Aug 10 2023 22:57:31 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アソシエイト