arXiv reaDer
部分的監視のためのラベルセット損失関数:胎児脳3DMRIパーセレーションへの適用
Label-set Loss Functions for Partial Supervision: Application to Fetal Brain 3D MRI Parcellation
ディープニューラルネットワークにより、自動セグメンテーションの精度が向上しましたが、その精度は、完全にセグメント化された多数の画像の可用性に依存します。部分的に注釈が付けられたデータセットをより有効に活用するには、すべてではないが一部の関心領域がセグメント化されている画像を使用してディープニューラルネットワークをトレーニングする方法が必要です。この論文では、部分的にセグメント化された画像を処理できる損失関数であるラベルセット損失関数の最初の公理的定義を提案します。完全にセグメント化された画像の古典的な損失関数を適切なラベルセット損失関数に変換する方法は1つしかないことを証明します。私たちの理論では、葉のダイス損失を定義することもできます。これは、ラベルが欠落しているだけの部分的な監視に特に適した、ダイス損失のラベルセットの一般化です。リーフダイスロスを使用して、胎児の脳の3DMRIセグメンテーションの部分的教師あり学習に新しい最先端を設定しました。白質、脳室、小脳、脳室外CSF、皮質灰白質、深灰白質、脳幹、脳梁を、解剖学的に正常な胎児の胎児脳3D MRIに基づいて、または開いた脊椎ビフィダでセグメント化できる深部神経ネットワークを実現します。提案されたラベルセット損失関数の実装は、https://github.com/LucasFidon/label-set-loss-functionsで入手できます。
Deep neural networks have increased the accuracy of automatic segmentation, however, their accuracy depends on the availability of a large number of fully segmented images. Methods to train deep neural networks using images for which some, but not all, regions of interest are segmented are necessary to make better use of partially annotated datasets. In this paper, we propose the first axiomatic definition of label-set loss functions that are the loss functions that can handle partially segmented images. We prove that there is one and only one method to convert a classical loss function for fully segmented images into a proper label-set loss function. Our theory also allows us to define the leaf-Dice loss, a label-set generalization of the Dice loss particularly suited for partial supervision with only missing labels. Using the leaf-Dice loss, we set a new state of the art in partially supervised learning for fetal brain 3D MRI segmentation. We achieve a deep neural network able to segment white matter, ventricles, cerebellum, extra-ventricular CSF, cortical gray matter, deep gray matter, brainstem, and corpus callosum based on fetal brain 3D MRI of anatomically normal fetuses or with open spina bifida. Our implementation of the proposed label-set loss functions is available at https://github.com/LucasFidon/label-set-loss-functions
updated: Fri Jul 09 2021 15:44:06 GMT+0000 (UTC)
published: Thu Jul 08 2021 13:53:56 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アソシエイト