弱い教師ありセマンティック セグメンテーションは、人によるアノテーションのコストが低いため、大きな注目を集めています。この論文では、境界ボックスの教師ありセマンティック セグメンテーション、つまり、監視としてバウンディング ボックスの注釈を使用して正確なセマンティック セグメンテーション モデルをトレーニングすることを目指しています。この目的のために、アフィニティ アテンション グラフ ニューラル ネットワーク (A^2GNN) を提案します。以前のプラクティスに従って、最初に疑似セマンティック認識シードを生成し、次に、新しく提案されたアフィニティ畳み込みニューラル ネットワーク (CNN) に基づいてセマンティック グラフに形成します。次に、構築されたグラフが A^2GNN に入力されます。この A^2GNN では、アフィニティ アテンション レイヤーが、ソフト グラフ エッジから短距離および長距離情報を取得して、信頼できるシードからラベルのないピクセルにセマンティック ラベルを正確に伝播するように設計されています。ただし、シードの精度を保証するために、A^2GNN には限られた数の信頼できるピクセル シード ラベルのみを採用しているため、トレーニングの監督が不十分になる可能性があります。この問題を軽減するために、新しい損失関数と、境界ボックスの制約を活用する整合性チェック メカニズムをさらに導入して、モデルの最適化により信頼性の高いガイダンスを含めることができるようにします。実験では、私たちのアプローチが Pascal VOC 2012 データセットで新しい最先端のパフォーマンスを達成することが示されています (値: 76.5%、テスト: 75.2%)。さらに重要なことに、私たちのアプローチは、バウンディング ボックスの教師ありインスタンス セグメンテーション タスクまたはその他の弱い教師ありセマンティック セグメンテーション タスクに容易に適用でき、PASCAL VOC または COCO データセットのすべての弱い教師ありタスク間で最先端または同等のパフォーマンスを実現できます。ソース コードは https://github.com/zbf1991/A2GNN で入手できます。
Weakly supervised semantic segmentation is receiving great attention due to its low human annotation cost. In this paper, we aim to tackle bounding box supervised semantic segmentation, i.e., training accurate semantic segmentation models using bounding box annotations as supervision. To this end, we propose Affinity Attention Graph Neural Network (A^2GNN). Following previous practices, we first generate pseudo semantic-aware seeds, which are then formed into semantic graphs based on our newly proposed affinity Convolutional Neural Network (CNN). Then the built graphs are input to our A^2GNN, in which an affinity attention layer is designed to acquire the short- and long- distance information from soft graph edges to accurately propagate semantic labels from the confident seeds to the unlabeled pixels. However, to guarantee the precision of the seeds, we only adopt a limited number of confident pixel seed labels for A^2GNN, which may lead to insufficient supervision for training. To alleviate this issue, we further introduce a new loss function and a consistency-checking mechanism to leverage the bounding box constraint, so that more reliable guidance can be included for the model optimization. Experiments show that our approach achieves new state-of-the-art performances on Pascal VOC 2012 datasets (val: 76.5%, test: 75.2%). More importantly, our approach can be readily applied to bounding box supervised instance segmentation task or other weakly supervised semantic segmentation tasks, with state-of-the-art or comparable performance among almot all weakly supervised tasks on PASCAL VOC or COCO dataset. Our source code will be available at https://github.com/zbf1991/A2GNN.