arXiv reaDer
半教師あり医用画像セグメンテーションのための不確実性に基づく相互整合性学習
Uncertainty-Guided Mutual Consistency Learning for Semi-Supervised Medical Image Segmentation
医用画像のセグメンテーションは、多くの臨床アプローチにおいて基本的かつ重要なステップです。半教師あり学習は、専門家が検査した注釈を取得するという大きな負担を軽減し、取得がはるかに簡単なラベルなしデータを利用するため、医療画像セグメンテーション タスクに広く適用されています。一貫性学習は、さまざまな分布の下で予測の不変性を強制することにより効果的なアプローチであることが証明されていますが、既存のアプローチでは、ラベルのないデータからの領域レベルの形状制約と境界レベルの距離情報を十分に活用できません。この論文では、自己アンサンブルの最新予測からのタスク内一貫性学習と、タスクレベルの正則化から幾何学的形状情報を活用します。このフレームワークは、モデルの推定されたセグメンテーションの不確実性によって導かれ、一貫性学習のための比較的特定の予測を選択して、ラベル付けされていないデータからより信頼性の高い情報を効果的に活用します。公開されている 2 つのベンチマーク データセットでの実験では、次のことが示されました。 . 2) 他の半教師付きセグメンテーション方法と比較して、提案された方法は、両方のデータセットで同じバックボーン ネットワークとタスク設定の下でより優れたセグメンテーション パフォーマンスを達成し、方法の有効性と堅牢性、および他の医用画像セグメンテーション タスクへの潜在的な転送可能性を示します。
Medical image segmentation is a fundamental and critical step in many clinical approaches. Semi-supervised learning has been widely applied to medical image segmentation tasks since it alleviates the heavy burden of acquiring expert-examined annotations and takes the advantage of unlabeled data which is much easier to acquire. Although consistency learning has been proven to be an effective approach by enforcing an invariance of predictions under different distributions, existing approaches cannot make full use of region-level shape constraint and boundary-level distance information from unlabeled data. In this paper, we propose a novel uncertainty-guided mutual consistency learning framework to effectively exploit unlabeled data by integrating intra-task consistency learning from up-to-date predictions for self-ensembling and cross-task consistency learning from task-level regularization to exploit geometric shape information. The framework is guided by the estimated segmentation uncertainty of models to select out relatively certain predictions for consistency learning, so as to effectively exploit more reliable information from unlabeled data. Experiments on two publicly available benchmark datasets showed that: 1) Our proposed method can achieve significant performance improvement by leveraging unlabeled data, with up to 4.13% and 9.82% in Dice coefficient compared to supervised baseline on left atrium segmentation and brain tumor segmentation, respectively. 2) Compared with other semi-supervised segmentation methods, our proposed method achieve better segmentation performance under the same backbone network and task settings on both datasets, demonstrating the effectiveness and robustness of our method and potential transferability for other medical image segmentation tasks.
updated: Thu Aug 25 2022 23:32:56 GMT+0000 (UTC)
published: Sun Dec 05 2021 08:19:41 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アソシエイト