arXiv reaDer
不確実性に基づく自己相互監視による半教師ありセマンティックセグメンテーション
Semi-supervision semantic segmentation with uncertainty-guided self cross supervision
半教師ありセグメンテーションを実現する強力な方法として、相互監視法は、豊富なラベルなし画像を使用して、独立したアンサンブルモデルに基づいて相互整合性を学習します。ただし、相互監視によって生成された誤った疑似ラベリング情報は、トレーニングプロセスを混乱させ、セグメンテーションモデルの有効性に悪影響を及ぼします。さらに、このような方法でのアンサンブルモデルのトレーニングプロセスは、計算リソースのコストを増大させ、トレーニング効率を低下させます。これらの問題を解決するために、我々は新しい交差監視方法、すなわち不確実性誘導自己交差監視(USCS)を提案します。アンサンブルモデルに加えて、最初に、共有モデルで複数の出力を生成できる多入力多出力(MIMO)セグメンテーションモデルを設計し、その結果、出力に一貫性を課して、パラメーターと計算のコストを節約します。一方、ガイド情報として不確実性を採用し、モデルが疑似ラベルの信頼性の高い領域に焦点を合わせ、自己交差監視における誤った疑似ラベルの影響を軽減し、セグメンテーションモデルのパフォーマンスを向上させるように促します。広範な実験により、私たちの方法は、パラメーターと計算の40.5%と49.1%のコストを節約しながら、最先端のパフォーマンスを達成することが示されています。
As a powerful way of realizing semi-supervised segmentation, the cross supervision method learns cross consistency based on independent ensemble models using abundant unlabeled images. However, the wrong pseudo labeling information generated by cross supervision would confuse the training process and negatively affect the effectiveness of the segmentation model. Besides, the training process of ensemble models in such methods also multiplies the cost of computation resources and decreases the training efficiency. To solve these problems, we propose a novel cross supervision method, namely uncertainty-guided self cross supervision (USCS). In addition to ensemble models, we first design a multi-input multi-output (MIMO) segmentation model which can generate multiple outputs with shared model and consequently impose consistency over the outputs, saving the cost on parameters and calculations. On the other hand, we employ uncertainty as guided information to encourage the model to focus on the high confident regions of pseudo labels and mitigate the effects of wrong pseudo labeling in self cross supervision, improving the performance of the segmentation model. Extensive experiments show that our method achieves state-of-the-art performance while saving 40.5% and 49.1% cost on parameters and calculations.
updated: Tue Mar 15 2022 02:17:20 GMT+0000 (UTC)
published: Thu Mar 10 2022 02:32:13 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アソシエイト