arXiv reaDer
注意からの親和性の学習:トランスフォーマーを使用したエンドツーエンドの弱教師ありセマンティックセグメンテーション
Learning Affinity from Attention: End-to-End Weakly-Supervised Semantic Segmentation with Transformers
画像レベルのラベルを使用した弱教師ありセマンティックセグメンテーション(WSSS)は、重要で困難な作業です。トレーニング効率が高いため、WSSSのエンドツーエンドソリューションはコミュニティからますます注目を集めています。ただし、現在の方法は主に畳み込みニューラルネットワークに基づいており、グローバル情報を適切に探索できないため、通常、オブジェクト領域が不完全になります。この論文では、前述の問題に対処するために、グローバル情報を自然に統合して、エンドツーエンドWSSSのより統合された初期疑似ラベルを生成するトランスフォーマーを紹介します。トランスフォーマーの自己注意とセマンティックアフィニティの間の固有の一貫性に動機付けられて、トランスフォーマーのマルチヘッドセルフアテンション(MHSA)からセマンティックアフィニティを学習するためのAffinity from Attention(AFA)モジュールを提案します。次に、学習したアフィニティを利用して、セグメンテーションの初期疑似ラベルを改良します。さらに、AFAを監視するための信頼できるアフィニティラベルを効率的に導出し、疑似ラベルのローカル整合性を確保するために、低レベルの画像外観情報を組み込んで疑似ラベルをリファインするPixel-AdaptiveRefinementモジュールを考案します。私たちは広範な実験を行い、PASCAL VOC2012およびMSCOCO 2014データセットでそれぞれ66.0%および38.9%mIoUを達成し、最近のエンドツーエンドの方法およびいくつかの多段階の競合他社を大幅に上回っています。コードはhttps://github.com/rulixiang/afaで入手できます。
Weakly-supervised semantic segmentation (WSSS) with image-level labels is an important and challenging task. Due to the high training efficiency, end-to-end solutions for WSSS have received increasing attention from the community. However, current methods are mainly based on convolutional neural networks and fail to explore the global information properly, thus usually resulting in incomplete object regions. In this paper, to address the aforementioned problem, we introduce Transformers, which naturally integrate global information, to generate more integral initial pseudo labels for end-to-end WSSS. Motivated by the inherent consistency between the self-attention in Transformers and the semantic affinity, we propose an Affinity from Attention (AFA) module to learn semantic affinity from the multi-head self-attention (MHSA) in Transformers. The learned affinity is then leveraged to refine the initial pseudo labels for segmentation. In addition, to efficiently derive reliable affinity labels for supervising AFA and ensure the local consistency of pseudo labels, we devise a Pixel-Adaptive Refinement module that incorporates low-level image appearance information to refine the pseudo labels. We perform extensive experiments and our method achieves 66.0% and 38.9% mIoU on the PASCAL VOC 2012 and MS COCO 2014 datasets, respectively, significantly outperforming recent end-to-end methods and several multi-stage competitors. Code is available at https://github.com/rulixiang/afa.
updated: Thu Sep 08 2022 05:38:37 GMT+0000 (UTC)
published: Sat Mar 05 2022 06:07: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アソシエイト