arXiv reaDer
微細解像度リモートセンシング画像のセマンティックセグメンテーションのための多段注意ResU-Net
Multistage Attention ResU-Net for Semantic Segmentation of Fine-Resolution Remote Sensing Images
視覚および言語タスクで広く使用されている内積注意メカニズムのメモリおよび計算コストは、入力の時空間サイズとともに二次関数的に増加します。このような成長は、入力が多いアプリケーションシナリオでの注意メカニズムの使用を妨げます。このレターでは、この欠陥を修正するために、メモリと計算のコストを大幅に削減して、ドット積の注意に近似する線形注意メカニズム(LAM)を提案します。効率的な設計により、注意メカニズムとニューラルネットワーク間の組み込みがより柔軟で用途が広くなります。提案されたLAMに基づいて、生のU-Netのスキップ接続をリファクタリングし、高解像度のリモートセンシング画像を使用したセマンティックセグメンテーション用のMultistage Attention ResU-Net(MAResU-Net)を設計します。 Vaihingenデータセットで実施された実験により、MAResU-Netの有効性が実証されました。コードはhttps://github.com/lironui/Multistage-Attention-ResU-Netで入手できます。
The memory and computation costs of the dot-product attention mechanism widely used in vision and language tasks increase quadratically with the spatio-temporal size of the input. Such growth hinders the usage of attention mechanisms in application scenarios with large inputs. In this Letter, to remedy this deficiency, we propose a Linear Attention Mechanism (LAM) which is approximate to dot-product attention with dramatically less memory and computation costs. The efficient design makes the incorporation between attention mechanisms and neural networks more flexible and versatile. Based on the proposed LAM, we refactor the skip connections in the raw U-Net and design a Multistage Attention ResU-Net (MAResU-Net) for semantic segmentation using fine-resolution remote sensing images. Experiments conducted on the Vaihingen dataset demonstrated the effectiveness of our MAResU-Net. Code is available at https://github.com/lironui/Multistage-Attention-ResU-Net.
updated: Sun Nov 29 2020 07:24:21 GMT+0000 (UTC)
published: Sun Nov 29 2020 07:24:21 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アソシエイト