arXiv reaDer
医療画像セグメンテーションのためのボクセルごとの敵対的半教師あり学習
Voxel-wise Adversarial Semi-supervised Learning for Medical Image Segmentation
医療画像セグメンテーションの半教師あり学習は、医療分野での信頼性の高い大規模な注釈の構築に関連する莫大なコストを軽減するための重要な研究分野です。最近の半教師ありアプローチは、一貫性の正則化、疑似ラベル付け手法、および敵対的学習を採用することにより、有望な結果を示しています。これらのメソッドは、主に、予測または埋め込みコンテキストで一貫性を強制することにより、ラベル付きデータとラベルなしデータの分布を学習しようとします。ただし、以前のアプローチは、ローカルの不一致の最小化または単一のクラス間のコンテキスト関係にのみ焦点を当てていました。この論文では、複数の隠れ層からローカルとグローバルの両方の機能を効果的に埋め込み、複数のクラス間のコンテキスト関係を学習する、新しい敵対的学習ベースの半教師ありセグメンテーション手法を紹介します。私たちのボクセルワイズ敵対学習法は、ボクセルワイズ特徴弁別器を利用します。これは、クラス固有のボクセルワイズ特徴分布を埋め込むことにより、多層ボクセルワイズ特徴(ローカルおよびグローバル特徴の両方を含む)を入力として考慮します。さらに、情報の損失と学習の安定性の問題を克服することにより、以前の表現学習方法を改善します。これにより、ラベル付けされたデータの豊富な表現が可能になります。私たちの方法は、左アトリウム(シングルクラス)およびマルチオルガンデータセット(マルチクラス)の画像セグメンテーションに関して、現在の最高のパフォーマンスを発揮する半教師あり学習アプローチよりも優れています。さらに、特徴空間の視覚的解釈は、提案された方法が、ラベル付けされたデータとラベル付けされていないデータの両方から十分に分散され分離された特徴空間を可能にし、全体的な予測結果を改善することを示しています。
Semi-supervised learning for medical image segmentation is an important area of research for alleviating the huge cost associated with the construction of reliable large-scale annotations in the medical domain. Recent semi-supervised approaches have demonstrated promising results by employing consistency regularization, pseudo-labeling techniques, and adversarial learning. These methods primarily attempt to learn the distribution of labeled and unlabeled data by enforcing consistency in the predictions or embedding context. However, previous approaches have focused only on local discrepancy minimization or context relations across single classes. In this paper, we introduce a novel adversarial learning-based semi-supervised segmentation method that effectively embeds both local and global features from multiple hidden layers and learns context relations between multiple classes. Our voxel-wise adversarial learning method utilizes a voxel-wise feature discriminator, which considers multilayer voxel-wise features (involving both local and global features) as an input by embedding class-specific voxel-wise feature distribution. Furthermore, we improve our previous representation learning method by overcoming information loss and learning stability problems, which enables rich representations of labeled data. Our method outperforms current best-performing state-of-the-art semi-supervised learning approaches on the image segmentation of the left atrium (single class) and multiorgan datasets (multiclass). Moreover, our visual interpretation of the feature space demonstrates that our proposed method enables a well-distributed and separated feature space from both labeled and unlabeled data, which improves the overall prediction results.
updated: Sat May 14 2022 06:57:19 GMT+0000 (UTC)
published: Sat May 14 2022 06:57:19 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アソシエイト