arXiv reaDer
TDAM:CNNでのコンテキストガイド付き特徴選択のためのトップダウンアテンションモジュール
TDAM: Top-Down Attention Module for Contextually Guided Feature Selection in CNNs
畳み込みニューラルネットワーク(CNN)の注意モジュールは、複数のコンピュータービジョンタスクのパフォーマンスを向上させる効果的な方法です。既存の方法は、チャネル、空間、および自己注意を適切にモデル化しますが、主にフィードフォワードボトムアップ方式で動作します。その結果、注意メカニズムは、単一の入力特徴マップのローカル情報に強く依存し、上位層から下位レベルの特徴マップで「何をどこで見るか」を指定できる、上位層で利用可能な比較的意味的に豊富なコンテキスト情報を組み込みません。ダウン情報フロー。したがって、この作業では、「視覚的サーチライト」を繰り返し生成して入力のチャネルおよび空間変調を実行し、各計算ステップでよりコンテキストに関連する特徴マップを出力する軽量トップダウン注意モジュール(TDAM)を提案します。私たちの実験は、TDAMが複数のオブジェクト認識ベンチマーク全体でCNNのパフォーマンスを向上させ、パラメーターとメモリの効率を高めながら、著名な注意モジュールよりも優れていることを示しています。さらに、TDAMベースのモデルは、明示的な監視なしで各計算ステップで個々のオブジェクトまたは機能をローカライズすることによって「注意をシフト」することを学習します。その結果、弱く監視されたオブジェクトのローカリゼーションでResNet50が5%向上します。
Attention modules for Convolutional Neural Networks (CNNs) are an effective method to enhance performance on multiple computer-vision tasks. While existing methods appropriately model channel-, spatial- and self-attention, they primarily operate in a feedforward bottom-up manner. Consequently, the attention mechanism strongly depends on the local information of a single input feature map and does not incorporate relatively semantically-richer contextual information available at higher layers that can specify "what and where to look" in lower-level feature maps through top-down information flow. Accordingly, in this work, we propose a lightweight top-down attention module (TDAM) that iteratively generates a "visual searchlight" to perform channel and spatial modulation of its inputs and outputs more contextually-relevant feature maps at each computation step. Our experiments indicate that TDAM enhances the performance of CNNs across multiple object-recognition benchmarks and outperforms prominent attention modules while being more parameter and memory efficient. Further, TDAM-based models learn to "shift attention" by localizing individual objects or features at each computation step without any explicit supervision resulting in a 5% improvement for ResNet50 on weakly-supervised object localization.
updated: Tue Apr 12 2022 09:53:38 GMT+0000 (UTC)
published: Fri Nov 26 2021 12:35:17 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アソシエイト