arXiv reaDer
半教師付き医用画像セグメンテーションのための二重の不確実性を伴う自己トレーニング
Self-training with dual uncertainty for semi-supervised medical image segmentation
半教師付き医用画像セグメンテーションの分野では、ラベル付きデータの不足が根本的な問題です。ラベル付けされていない画像から画像の特徴を効果的に学習してセグメンテーションの精度を向上させる方法は、この分野の主な研究の方向性です。従来の自己トレーニング方法では、反復トレーニング用の疑似ラベルを生成することで、ラベル付けされたデータが不十分であるという問題を部分的に解決できます。ただし、トレーニング中にモデルの不確実性によって生成されるノイズは、セグメンテーションの結果に直接影響します。したがって、自己トレーニング フレームワークに基づいてトレーニング プロセスを安定させるために、サンプル レベルとピクセル レベルの不確実性を追加しました。具体的には、事前トレーニング中にモデルのいくつかの瞬間を保存し、ラベルのないサンプルに対する予測の差を、そのサンプルのサンプル レベルの不確実性の推定値として使用しました。次に、トレーニング中にラベルのないサンプルを簡単なものから難しいものへと徐々に追加します。同時に、アップサンプリング方法が異なるデコーダーをセグメンテーション ネットワークに追加し、2 つのデコーダーの出力の差をピクセル レベルの不確実性として使用しました。つまり、ラベル付けされていないサンプルを選択的に再トレーニングし、ピクセルレベルの不確実性を疑似ラベルに割り当てて、自己トレーニングプロセスを最適化しました。モデルのセグメンテーション結果を、公開されている 2017 ACDC データセットと 2018 Prostate データセットに対する 5 つの半教師付きアプローチと比較しました。提案された方法は、同じ設定の下で両方のデータセットでより優れたセグメンテーション パフォーマンスを達成し、その有効性、堅牢性、および他の医用画像セグメンテーション タスクへの潜在的な転送可能性を示しています。キーワード: 医用画像セグメンテーション、半教師あり学習、自己訓練、不確実性推定
In the field of semi-supervised medical image segmentation, the shortage of labeled data is the fundamental problem. How to effectively learn image features from unlabeled images to improve segmentation accuracy is the main research direction in this field. Traditional self-training methods can partially solve the problem of insufficient labeled data by generating pseudo labels for iterative training. However, noise generated due to the model's uncertainty during training directly affects the segmentation results. Therefore, we added sample-level and pixel-level uncertainty to stabilize the training process based on the self-training framework. Specifically, we saved several moments of the model during pre-training, and used the difference between their predictions on unlabeled samples as the sample-level uncertainty estimate for that sample. Then, we gradually add unlabeled samples from easy to hard during training. At the same time, we added a decoder with different upsampling methods to the segmentation network and used the difference between the outputs of the two decoders as pixel-level uncertainty. In short, we selectively retrained unlabeled samples and assigned pixel-level uncertainty to pseudo labels to optimize the self-training process. We compared the segmentation results of our model with five semi-supervised approaches on the public 2017 ACDC dataset and 2018 Prostate dataset. Our proposed method achieves better segmentation performance on both datasets under the same settings, demonstrating its effectiveness, robustness, and potential transferability to other medical image segmentation tasks. Keywords: Medical image segmentation, semi-supervised learning, self-training, uncertainty estimation
updated: Tue Oct 10 2023 08:33:24 GMT+0000 (UTC)
published: Mon Apr 10 2023 07:57:24 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アソシエイト