arXiv reaDer
EPSANet: 畳み込みニューラル ネットワーク上の効率的なピラミッド分割アテンション ブロック
EPSANet: An Efficient Pyramid Split Attention Block on Convolutional Neural Network
最近、深層畳み込みニューラル ネットワークにアテンション モジュールを埋め込むことで、そのパフォーマンスを効果的に改善できることが実証されています。この作業では、ピラミッド分割アテンション (PSA) モジュールと呼ばれる軽量で効果的な新しいアテンション方法が提案されています。 ResNet のボトルネック ブロック内の 3x3 コンボリューションを PSA モジュールに置き換えることにより、効率的なピラミッド分割アテンション (EPSA) という名前の新しい表現ブロックが得られます。 EPSA ブロックは、確立されたバックボーン ネットワークにプラグ アンド プレイ コンポーネントとして簡単に追加でき、モデルのパフォーマンスを大幅に向上させることができます。したがって、この作業では、これらの ResNet スタイルの EPSA ブロックをスタックすることにより、EPSANet という名前のシンプルで効率的なバックボーン アーキテクチャが開発されます。それに応じて、提案された EPSANet によって、画像分類、オブジェクト検出、インスタンス セグメンテーションなどを含むがこれらに限定されないさまざまなコンピューター ビジョン タスクに対して、より強力なマルチスケール表現能力を提供できます。最先端のチャネル アテンション メソッドのほとんど。 SENet-50 と比較して、ImageNet データセットでトップ 1 の精度が 1.93 % 向上し、オブジェクト検出用の +2.7 ボックス AP のマージンが大きく、Mask-RCNN を使用したインスタンス セグメンテーション用に +1.7 マスク AP が向上しています。 MS-COCO のデータセットが取得されます。ソース コードは、https://github.com/murufeng/EPSANet で入手できます。
Recently, it has been demonstrated that the performance of a deep convolutional neural network can be effectively improved by embedding an attention module into it. In this work, a novel lightweight and effective attention method named Pyramid Split Attention (PSA) module is proposed. By replacing the 3x3 convolution with the PSA module in the bottleneck blocks of the ResNet, a novel representational block named Efficient Pyramid Split Attention (EPSA) is obtained. The EPSA block can be easily added as a plug-and-play component into a well-established backbone network, and significant improvements on model performance can be achieved. Hence, a simple and efficient backbone architecture named EPSANet is developed in this work by stacking these ResNet-style EPSA blocks. Correspondingly, a stronger multi-scale representation ability can be offered by the proposed EPSANet for various computer vision tasks including but not limited to, image classification, object detection, instance segmentation, etc. Without bells and whistles, the performance of the proposed EPSANet outperforms most of the state-of-the-art channel attention methods. As compared to the SENet-50, the Top-1 accuracy is improved by 1.93 % on ImageNet dataset, a larger margin of +2.7 box AP for object detection and an improvement of +1.7 mask AP for instance segmentation by using the Mask-RCNN on MS-COCO dataset are obtained. Our source code is available at:https://github.com/murufeng/EPSANet.
updated: Sun May 30 2021 07:26:41 GMT+0000 (UTC)
published: Sun May 30 2021 07:26:41 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アソシエイト