arXiv reaDer
部分的に監視された医療画像セグメンテーションのための深い互換性のある学習
Deep Compatible Learning for Partially-Supervised Medical Image Segmentation
部分的に教師あり学習は、ラベルのない構造の教師がないため、セグメンテーションが難しい場合があります。完全な教師あり学習を直接適用する方法は、非互換性につながる可能性があります。つまり、損失関数を考えると、最適化問題の解集合にはグラウンドトゥルースがありません。 。この課題に対処するために、部分構造のみに注釈が付けられた画像を使用して単一のマルチラベルセグメンテーションネットワークをトレーニングする、ディープ互換学習(DCL)フレームワークを提案します。まず、部分的に監視されたセグメンテーションを、欠落しているラベルと互換性のある最適化問題として定式化し、その互換性を証明します。次に、モデルに条件付きセグメンテーション戦略を装備して、複数の部分的に注釈が付けられた画像からターゲットにラベルを伝播します。さらに、ラベル付けされていない構造の実質的な監視を提供するために、ラベル伝播の2つの反対のマッピングを同時に学習する二重学習戦略を提案します。 2つの戦略は、それぞれ条件付き互換性および二重互換性と呼ばれる互換性のある形式に定式化されます。このフレームワークは、従来の損失関数に一般的に適用できることを示します。このアプローチは、特に小さなトレーニングデータセットしか利用できない状況で、既存の方法に比べて大幅なパフォーマンスの向上を実現します。 3つのセグメンテーションタスクの結果は、提案されたフレームワークが完全に監視されたモデルと一致するパフォーマンスを達成できることを示しています。
Partially-supervised learning can be challenging for segmentation due to the lack of supervision for unlabeled structures, and the methods directly applying fully-supervised learning could lead to incompatibility, meaning ground truth is not in the solution set of the optimization problem given the loss function. To address the challenge, we propose a deep compatible learning (DCL) framework, which trains a single multi-label segmentation network using images with only partial structures annotated. We first formulate the partially-supervised segmentation as an optimization problem compatible with missing labels, and prove its compatibility. Then, we equip the model with a conditional segmentation strategy, to propagate labels from multiple partially-annotated images to the target. Additionally, we propose a dual learning strategy, which learns two opposite mappings of label propagation simultaneously, to provide substantial supervision for unlabeled structures. The two strategies are formulated into compatible forms, termed as conditional compatibility and dual compatibility, respectively. We show this framework is generally applicable for conventional loss functions. The approach attains significant performance improvement over existing methods, especially in the situation where only a small training dataset is available. Results on three segmentation tasks have shown that the proposed framework could achieve performance matching fully-supervised models.
updated: Sat Jun 18 2022 08:38:31 GMT+0000 (UTC)
published: Sat Jun 18 2022 08:38:31 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アソシエイト