arXiv reaDer
Semi-Supervised Medical Image Segmentation via Learning Consistency under Transformations
 多くの場合、ラベル付きデータの不足により、医療画像のセグメンテーションに対する教師付き深層学習手法の適用が制限されます。これは、ラベル付き画像とラベルなし画像の混合物から学習する半教師あり技術の開発の動機となっています。この論文では、ラベル付きトレーニング画像の教師あり学習に加えて、ラベル付き画像とラベルなし画像の両方で、与えられたクラスの変換の下で一貫したセグメンテーションを予測することを学習する新しい半教師あり方法を提案します。より具体的には、この作業では、弾性変形の等分散性の学習を検討します。これは、1)異なる変換画像をそれぞれ受け取る2つの同一のブランチを持つシャムアーキテクチャ、および2)教師付きセグメンテーション損失項と、予測のセグメンテーションの一貫性を促進する教師なし項を持つ複合損失関数を使用して実装します。 2つのブランチ。 5倍交差検証を使用して、解剖学的構造のセグメンテーションを使用した胸部X線写真の公開データセットでメソッドを評価します。提案された方法は、教師あり学習に比べて非常に高いセグメンテーション精度に達します。これは、ラベル付き画像とラベルなし画像の両方で変換の一貫性を学習するためであり、後者が最も寄与します。大幅に少ないラベル付き画像を使用しながら、最先端の胸部X線セグメンテーション法に匹敵する性能を実現しています。
The scarcity of labeled data often limits the application of supervised deep learning techniques for medical image segmentation. This has motivated the development of semi-supervised techniques that learn from a mixture of labeled and unlabeled images. In this paper, we propose a novel semi-supervised method that, in addition to supervised learning on labeled training images, learns to predict segmentations consistent under a given class of transformations on both labeled and unlabeled images. More specifically, in this work we explore learning equivariance to elastic deformations. We implement this through: 1) a Siamese architecture with two identical branches, each of which receives a differently transformed image, and 2) a composite loss function with a supervised segmentation loss term and an unsupervised term that encourages segmentation consistency between the predictions of the two branches. We evaluate the method on a public dataset of chest radiographs with segmentations of anatomical structures using 5-fold cross-validation. The proposed method reaches significantly higher segmentation accuracy compared to supervised learning. This is due to learning transformation consistency on both labeled and unlabeled images, with the latter contributing the most. We achieve the performance comparable to state-of-the-art chest X-ray segmentation methods while using substantially fewer labeled images.
updated: Mon Nov 04 2019 13:51:17 GMT+0000 (UTC)
published: Mon Nov 04 2019 13:51:17 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト