arXiv reaDer
半教師あり画像セグメンテーションのための自己ペースで自己無撞着な共同トレーニング
Self-paced and self-consistent co-training for semi-supervised image segmentation
注釈付きデータが不足している場合の画像セグメンテーションの効果的なアプローチとして、最近、ディープコトレーニングが提案されています。この論文では、自己ペースで自己無撞着なコトレーニング法を用いて、半教師ありセグメンテーションの既存のアプローチを改善します。ラベルのない画像から情報を抽出するために、最初に共同トレーニング用の自己ペース学習戦略を設計します。これにより、共同トレーニングされたニューラルネットワークは、最初にセグメント化が容易な領域に焦点を合わせ、次に難しい領域を徐々に検討します。これは、一般化されたJensenShannon Divergence(JSD)の形でのエンドツーエンドの微分可能な損失。さらに、さまざまなネットワークからの予測に一貫性と信頼性の両方を持たせるために、エントロピーに基づく不確実性正則化を使用して、この一般化されたJSD損失を強化します。個々のモデルのロバスト性は、さまざまなトレーニングの反復にわたって一貫性のある予測を強制する自己アンサンブル損失を使用してさらに改善されます。ラベル付けされたデータのごく一部を使用して、異なる画像モダリティを持つ3つの困難な画像セグメンテーション問題に対する私たちの方法の可能性を示します。結果は、標準の共同トレーニングベースラインおよび最近提案された半教師ありセグメンテーションの最先端のアプローチと比較して、パフォーマンスの点で明らかな利点を示しています。
Deep co-training has recently been proposed as an effective approach for image segmentation when annotated data is scarce. In this paper, we improve existing approaches for semi-supervised segmentation with a self-paced and self-consistent co-training method. To help distillate information from unlabeled images, we first design a self-paced learning strategy for co-training that lets jointly-trained neural networks focus on easier-to-segment regions first, and then gradually consider harder ones.This is achieved via an end-to-end differentiable loss inthe form of a generalized Jensen Shannon Divergence(JSD). Moreover, to encourage predictions from different networks to be both consistent and confident, we enhance this generalized JSD loss with an uncertainty regularizer based on entropy. The robustness of individual models is further improved using a self-ensembling loss that enforces their prediction to be consistent across different training iterations. We demonstrate the potential of our method on three challenging image segmentation problems with different image modalities, using small fraction of labeled data. Results show clear advantages in terms of performance compared to the standard co-training baselines and recently proposed state-of-the-art approaches for semi-supervised segmentation
updated: Sat Oct 31 2020 17:41:03 GMT+0000 (UTC)
published: Sat Oct 31 2020 17:41:03 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アソシエイト