arXiv reaDer
勝つために競う: ほとんど教師なしの医療画像セグメンテーションのための疑似ラベルの強化
Compete to Win: Enhancing Pseudo Labels for Barely-supervised Medical Image Segmentation
この研究では、わずかなラベル付きデータ、つまり 1 桁のケースしか利用できない、ほとんど教師なしの医用画像セグメンテーションを調査します。既存の最先端の半教師ありソリューション クロス疑似教師の主な制限は、フォアグラウンド クラスの不十分な精度であり、教師なし学習での退化した結果につながることがわかります。この論文では、疑似ラベルの品質を向上させるための新しい Compete-to-Win メソッド (ComWin) を提案します。 1 つのモデルの予測を疑似ラベルとして直接使用するのとは対照的に、私たちの重要なアイデアは、異なるネットワークによって生成された複数の信頼マップを比較して、最も信頼できるものを選択することによって、高品質の疑似ラベルを生成する必要があるということです (競争に勝つための戦略)。境界付近の領域で疑似ラベルをさらに改良するために、境界認識拡張モジュールを統合することにより、ComWin の拡張バージョン、つまり ComWin+ が提案されています。実験では、心臓構造のセグメンテーション、膵臓のセグメンテーション、結腸腫瘍のセグメンテーションの 3 つの公開医用画像データセットで、それぞれ最高のパフォーマンスを達成できることが示されています。ソース コードは、https://github.com/Huiimin5/comwin で入手できるようになりました。
This study investigates barely-supervised medical image segmentation where only few labeled data, i.e., single-digit cases are available. We observe the key limitation of the existing state-of-the-art semi-supervised solution cross pseudo supervision is the unsatisfactory precision of foreground classes, leading to a degenerated result under barely-supervised learning. In this paper, we propose a novel Compete-to-Win method (ComWin) to enhance the pseudo label quality. In contrast to directly using one model's predictions as pseudo labels, our key idea is that high-quality pseudo labels should be generated by comparing multiple confidence maps produced by different networks to select the most confident one (a compete-to-win strategy). To further refine pseudo labels at near-boundary areas, an enhanced version of ComWin, namely, ComWin+, is proposed by integrating a boundary-aware enhancement module. Experiments show that our method can achieve the best performance on three public medical image datasets for cardiac structure segmentation, pancreas segmentation and colon tumor segmentation, respectively. The source code is now available at https://github.com/Huiimin5/comwin.
updated: Sun Jul 30 2023 14:15:55 GMT+0000 (UTC)
published: Sat Apr 15 2023 10:04:14 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アソシエイト