arXiv reaDer
ST ++:半教師ありセマンティックセグメンテーションのためのセルフトレーニング作業を改善する
ST++: Make Self-training Work Better for Semi-supervised Semantic Segmentation
疑似ラベル付けによる自己トレーニングは、ラベル付けされていないデータを活用するための従来のシンプルで一般的なパイプラインです。この作業では、最初に、ラベルのない画像に強力なデータ拡張(SDA)を注入して、ノイズの多いラベルの過剰適合を軽減し、教師と学生。この単純なメカニズムにより、STは、反復的な再トレーニングなど、ベルやホイッスルなしで既存のすべての方法よりも優れたパフォーマンスを発揮します。印象的な結果に触発されて、SDAを徹底的に調査し、いくつかの経験的分析を提供します。それでも、誤った疑似ラベルは蓄積され、パフォーマンスが低下する傾向があります。この目的のために、全体的な予測レベルの安定性に基づいて信頼性の高いラベルなし画像に優先順位を付けることで選択的な再トレーニングを実行する、高度なセルフトレーニングフレームワーク(つまりST ++)をさらに提案します。具体的には、いくつかのモデルチェックポイントが第1段階の教師ありトレーニングで保存され、ラベルのない画像での予測の不一致が信頼性の測定値として機能します。私たちの画像レベルの選択は、学習のための全体的なコンテキスト情報を提供します。一般的なピクセル単位の選択よりもセグメンテーションに適していることを示します。その結果、ST ++はSTのパフォーマンスをさらに向上させます。コードはhttps://github.com/LiheYoung/ST-PlusPlusで入手できます。
Self-training via pseudo labeling is a conventional, simple, and popular pipeline to leverage unlabeled data. In this work, we first construct a strong baseline of self-training (namely ST) for semi-supervised semantic segmentation via injecting strong data augmentations (SDA) on unlabeled images to alleviate overfitting noisy labels as well as decouple similar predictions between the teacher and student. With this simple mechanism, our ST outperforms all existing methods without any bells and whistles, e.g., iterative re-training. Inspired by the impressive results, we thoroughly investigate the SDA and provide some empirical analysis. Nevertheless, incorrect pseudo labels are still prone to accumulate and degrade the performance. To this end, we further propose an advanced self-training framework (namely ST++), that performs selective re-training via prioritizing reliable unlabeled images based on holistic prediction-level stability. Concretely, several model checkpoints are saved in the first stage supervised training, and the discrepancy of their predictions on the unlabeled image serves as a measurement for reliability. Our image-level selection offers holistic contextual information for learning. We demonstrate that it is more suitable for segmentation than common pixel-wise selection. As a result, ST++ further boosts the performance of our ST. Code is available at https://github.com/LiheYoung/ST-PlusPlus.
updated: Thu Mar 03 2022 13:40:30 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アソシエイト