arXiv reaDer
SoftSeg:画像セグメンテーションのためのソフトトレーニングとバイナリトレーニングの利点
SoftSeg: Advantages of soft versus binary training for image segmentation
ほとんどの画像セグメンテーションアルゴリズムは、ピクセルごとの分類タスクとして定式化されたバイナリマスクでトレーニングされます。ただし、医用画像などのアプリケーションでは、2つの組織間のコントラストが明確に定義されていないことが多いため、この「白黒」アプローチは制約が厳しすぎます。つまり、オブジェクトのエッジにあるボクセルには組織が混在しています。その結果、単一の「ハード」ラベルを割り当てると、有害な近似が生じる可能性があります。代わりに、非バイナリ値を含むソフト予測はその制限を克服します。 SoftSegを紹介します。これは、ソフトグラウンドトゥルースラベルを利用し、バイナリ予測に拘束されない深層学習トレーニングアプローチです。 SoftSegは、分類問題ではなく回帰を解決することを目的としています。これは、(i)前処理とデータ拡張後の二値化なし、(ii)正規化されたReLU最終活性化層(シグモイドの代わり)、および(iii)回帰損失関数(従来のダイス損失の代わり)を使用することによって実現されます。脊髄灰白質、多発性硬化症の脳病変、およびマルチモーダル脳腫瘍セグメンテーションの課題からの3つのオープンソースMRIセグメンテーションデータセットに対するこれら3つの機能の影響を評価します。複数の相互検証の反復にわたって、SoftSegは従来のアプローチを上回り、灰白質データセットで2.0%(p = 0.001)、MS病変で3.3%、脳腫瘍で6.5%のダイススコアの増加につながりました。 SoftSegは、組織のインターフェースで一貫したソフト予測を生成し、小さなオブジェクトの感度が向上します。ソフトラベルの豊富さは、専門家間の変動、部分的なボリューム効果を表し、モデルの不確実性の推定を補完する可能性があります。開発されたトレーニングパイプラインは、既存のディープラーニングアーキテクチャのほとんどに簡単に組み込むことができます。これは、無料で利用できるディープラーニングツールボックスivadomed(https://ivadomed.org)にすでに実装されています。
Most image segmentation algorithms are trained on binary masks formulated as a classification task per pixel. However, in applications such as medical imaging, this "black-and-white" approach is too constraining because the contrast between two tissues is often ill-defined, i.e., the voxels located on objects' edges contain a mixture of tissues. Consequently, assigning a single "hard" label can result in a detrimental approximation. Instead, a soft prediction containing non-binary values would overcome that limitation. We introduce SoftSeg, a deep learning training approach that takes advantage of soft ground truth labels, and is not bound to binary predictions. SoftSeg aims at solving a regression instead of a classification problem. This is achieved by using (i) no binarization after preprocessing and data augmentation, (ii) a normalized ReLU final activation layer (instead of sigmoid), and (iii) a regression loss function (instead of the traditional Dice loss). We assess the impact of these three features on three open-source MRI segmentation datasets from the spinal cord gray matter, the multiple sclerosis brain lesion, and the multimodal brain tumor segmentation challenges. Across multiple cross-validation iterations, SoftSeg outperformed the conventional approach, leading to an increase in Dice score of 2.0% on the gray matter dataset (p=0.001), 3.3% for the MS lesions, and 6.5% for the brain tumors. SoftSeg produces consistent soft predictions at tissues' interfaces and shows an increased sensitivity for small objects. The richness of soft labels could represent the inter-expert variability, the partial volume effect, and complement the model uncertainty estimation. The developed training pipeline can easily be incorporated into most of the existing deep learning architectures. It is already implemented in the freely-available deep learning toolbox ivadomed (https://ivadomed.org).
updated: Wed Nov 18 2020 02:25:09 GMT+0000 (UTC)
published: Wed Nov 18 2020 02:25:09 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アソシエイト