arXiv reaDer
Lawin Transformer:大きなウィンドウの注意によるマルチスケール表現によるセマンティックセグメンテーショントランスフォーマーの改善
Lawin Transformer: Improving Semantic Segmentation Transformer with Multi-Scale Representations via Large Window Attention
マルチスケール表現は、セマンティックセグメンテーションにとって非常に重要です。コミュニティは、マルチスケールのコンテキスト情報を活用するセマンティックセグメンテーション畳み込みニューラルネットワーク(CNN)の繁栄を目の当たりにしてきました。ビジョントランスフォーマー(ViT)が画像分類で強力であることに動機付けられて、いくつかのセマンティックセグメンテーションViTが最近提案されました。それらのほとんどは印象的な結果を達成しますが、計算コストがかかります。この論文では、ウィンドウアテンションメカニズムを介してマルチスケール表現をセマンティックセグメンテーションViTに導入することに成功し、パフォーマンスと効率をさらに向上させます。この目的のために、ローカルウィンドウがわずかな計算オーバーヘッドでコンテキストウィンドウのより広い領域をクエリできるようにする大きなウィンドウアテンションを導入します。クエリ領域に対するコンテキスト領域の比率を調整することにより、大きなウィンドウの注意が複数のスケールでコンテキスト情報をキャプチャできるようにします。さらに、空間ピラミッドプーリングのフレームワークは、大ウィンドウアテンションと連携するために採用されています。これは、セマンティックセグメンテーションViT用の大ウィンドウアテンション空間ピラミッドプーリング(LawinASPP)という名前の新しいデコーダーを提供します。結果として得られるViTであるLawinTransformerは、エンコーダーとしての効率的な階層型ビジョントランスフォーマー(HVT)とデコーダーとしてのLawinASPPで構成されています。経験的結果は、LawinTransformerが既存の方法と比較して改善された効率を提供することを示しています。 Lawin Transformerは、Cityscapes(84.4%mIoU)、ADE20K(56.2%mIoU)、およびCOCO-Stuffデータセットに新しい最先端のパフォーマンスをさらに設定します。コードはhttps://github.com/yan-hao-tian/lawinでリリースされます。
Multi-scale representations are crucial for semantic segmentation. The community has witnessed the flourish of semantic segmentation convolutional neural networks (CNN) exploiting multi-scale contextual information. Motivated by that the vision transformer (ViT) is powerful in image classification, some semantic segmentation ViTs are recently proposed, most of them attaining impressive results but at a cost of computational economy. In this paper, we succeed in introducing multi-scale representations into semantic segmentation ViT via window attention mechanism and further improves the performance and efficiency. To this end, we introduce large window attention which allows the local window to query a larger area of context window at only a little computation overhead. By regulating the ratio of the context area to the query area, we enable the large window attention to capture the contextual information at multiple scales. Moreover, the framework of spatial pyramid pooling is adopted to collaborate with the large window attention, which presents a novel decoder named large window attention spatial pyramid pooling (LawinASPP) for semantic segmentation ViT. Our resulting ViT, Lawin Transformer, is composed of an efficient hierachical vision transformer (HVT) as encoder and a LawinASPP as decoder. The empirical results demonstrate that Lawin Transformer offers an improved efficiency compared to the existing method. Lawin Transformer further sets new state-of-the-art performance on Cityscapes (84.4% mIoU), ADE20K (56.2% mIoU) and COCO-Stuff datasets. The code will be released at https://github.com/yan-hao-tian/lawin.
updated: Wed Jan 05 2022 13:51:20 GMT+0000 (UTC)
published: Wed Jan 05 2022 13:51:20 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アソシエイト