arXiv reaDer
GuidedMix-Net:ラベル付き画像を参照として使用して疑似マスクを改善する方法を学ぶ
GuidedMix-Net: Learning to Improve Pseudo Masks Using Labeled Images as Reference
半教師あり学習は、限られた数のラベル付きの例から学習することによってモデルを構築することを目的とした挑戦的な問題です。この問題に取り組むために多くの方法が提案されており、ネットワークを正規化するためにラベルのないインスタンスの一貫性の予測のみを利用することに最も焦点が当てられています。ただし、ラベル付きデータとラベルなしデータを別々に処理すると、ラベル付きの例から学習した大量の事前知識が破棄され、ラベル付きとラベルなしの画像ペア間の特徴の相互作用をマイニングできなくなることがよくあります。この論文では、ラベル付き情報を活用してラベルなしインスタンスの学習をガイドすることにより、GuidedMix-Netという名前の半教師ありセマンティックセグメンテーションの新しい方法を提案します。具体的には、最初にラベル付きデータとラベルなしデータの間に特徴の位置合わせの目的を導入して、潜在的に類似した画像ペアをキャプチャし、次にそれらから混合入力を生成します。クラスターの仮定に基づいて提案された相互情報量転送(MITrans)は、混合データ空間内のラベルなしデータの機能をさらに進歩的に洗練するための強力な知識モジュールであることが示されています。ラベル付きの例を利用し、ラベルなしのデータ学習をガイドするために、ラベルなしのデータに対して高品質の疑似マスクを生成するマスク生成モジュールをさらに提案します。ラベル付きデータの教師あり学習に加えて、ラベルなしデータの予測は、混合データから生成された疑似マスクと共同で学習されます。 PASCAL VOC 2012、PASCAL-Context、Cityscapesに関する広範な実験により、GuidedMix-Netの有効性が実証されました。これにより、競争力のあるセグメンテーション精度が達成され、以前の最先端のアプローチと比較してmIoUが+ 7%大幅に向上します。
Semi-supervised learning is a challenging problem which aims to construct a model by learning from a limited number of labeled examples. Numerous methods have been proposed to tackle this problem, with most focusing on utilizing the predictions of unlabeled instances consistency alone to regularize networks. However, treating labeled and unlabeled data separately often leads to the discarding of mass prior knowledge learned from the labeled examples, and failure to mine the feature interaction between the labeled and unlabeled image pairs. In this paper, we propose a novel method for semi-supervised semantic segmentation named GuidedMix-Net, by leveraging labeled information to guide the learning of unlabeled instances. Specifically, we first introduce a feature alignment objective between labeled and unlabeled data to capture potentially similar image pairs and then generate mixed inputs from them. The proposed mutual information transfer (MITrans), based on the cluster assumption, is shown to be a powerful knowledge module for further progressive refining features of unlabeled data in the mixed data space. To take advantage of the labeled examples and guide unlabeled data learning, we further propose a mask generation module to generate high-quality pseudo masks for the unlabeled data. Along with supervised learning for labeled data, the prediction of unlabeled data is jointly learned with the generated pseudo masks from the mixed data. Extensive experiments on PASCAL VOC 2012, PASCAL-Context and Cityscapes demonstrate the effectiveness of our GuidedMix-Net, which achieves competitive segmentation accuracy and significantly improves the mIoU by +7% compared to previous state-of-the-art approaches.
updated: Wed Jun 30 2021 05:25:21 GMT+0000 (UTC)
published: Tue Jun 29 2021 02:48:45 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アソシエイト