arXiv reaDer
ST++: 半教師ありセマンティック セグメンテーションのための自己トレーニングの作業を改善する
ST++: Make Self-training Work Better for Semi-supervised Semantic Segmentation
このホワイト ペーパーでは、自己トレーニング (シンプルですが一般的なフレームワーク) を半教師ありセグメンテーションでより適切に機能させることができるかどうかを調査します。半教師あり設定の中心的な問題は、ラベルのないデータを効果的かつ効率的に利用することにあるため、ラベルのないデータの多様性と硬度を高めることがパフォーマンスの向上に不可欠であることがわかります。この事実を認識して、このタスクのために、ラベルのないデータ (つまり、ST) に対する適切な強力なデータ拡張と組み合わせた最も単純な自己トレーニング スキームを採用することを提案します。さらに、誤って疑似ラベル付けされた画像の悪影響を軽減するために、より信頼性の高いラベル付けされていない画像を選択して優先順位を付けることにより、選択的な再トレーニングを実行する高度な自己トレーニング フレームワーク (つまり、ST++) をさらに提案します。その結果、提案された ST++ は、半教師ありモデルのパフォーマンスを大幅に向上させ、Pascal VOC 2012 および Cityscapes ベンチマークで既存の方法を大幅に上回っています。全体として、この単純で単純なフレームワークが、将来の作業の強力なベースラインまたは競合他社として機能することを願っています。コードは https://github.com/LiheYoung/ST-PlusPlus で入手できます。
In this paper, we investigate if we could make the self-training -- a simple but popular framework -- work better for semi-supervised segmentation. Since the core issue in semi-supervised setting lies in effective and efficient utilization of unlabeled data, we notice that increasing the diversity and hardness of unlabeled data is crucial to performance improvement. Being aware of this fact, we propose to adopt the most plain self-training scheme coupled with appropriate strong data augmentations on unlabeled data (namely ST) for this task, which surprisingly outperforms previous methods under various settings without any bells and whistles. Moreover, to alleviate the negative impact of the wrongly pseudo labeled images, we further propose an advanced self-training framework (namely ST++), that performs selective re-training via selecting and prioritizing the more reliable unlabeled images. As a result, the proposed ST++ boosts the performance of semi-supervised model significantly and surpasses existing methods by a large margin on the Pascal VOC 2012 and Cityscapes benchmark. Overall, we hope this straightforward and simple framework will serve as a strong baseline or competitor for future works. Code is available at https://github.com/LiheYoung/ST-PlusPlus.
updated: Wed Jun 09 2021 14:18:32 GMT+0000 (UTC)
published: Wed Jun 09 2021 14:18:32 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アソシエイト