arXiv reaDer
脳腫瘍セグメンテーションのための一般化されたWassersteinダイススコア、分布的にロバストなディープラーニング、およびレンジャー:BraTS2020チャレンジ
Generalized Wasserstein Dice Score, Distributionally Robust Deep Learning, and Ranger for brain tumor segmentation: BraTS 2020 challenge
深層ニューラルネットワークのトレーニングは、深層ニューラルネットワークの設計、サンプルごとの損失関数、母集団損失関数、およびオプティマイザーの4つの主要な要素を含む最適化問題です。ただし、最近のBraTSの課題に対抗するために開発された方法は、ディープニューラルネットワークアーキテクチャの設計にのみ焦点を当てる傾向があり、他の3つの側面にはあまり注意を払っていません。この論文では、反対のアプローチを採用して実験しました。汎用的で最先端の3DU-Netアーキテクチャに固執し、非標準のサンプルごとの損失関数、一般化されたWasserstein Dice損失、非標準の母集団損失関数を実験しました。これは、分布的にロバストな最適化に対応します。 、および非標準のオプティマイザ、レンジャー。これらのバリエーションは、特にマルチクラスの脳腫瘍セグメンテーションの問題のために選択されました。一般化されたWassersteinダイス損失は、BraTSでラベル付けされた腫瘍領域の階層構造を利用できるようにするサンプルごとの損失関数です。分布的にロバストな最適化は、トレーニングデータセット内の過小評価されたサブドメインの存在を説明する経験的リスク最小化の一般化です。 Rangerは、広く使用されているAdamオプティマイザーを一般化したもので、バッチサイズが小さく、ラベルにノイズが多いため、より安定しています。脳腫瘍のセグメンテーションのためのディープニューラルネットワークの最適化のこれらのバリエーションのそれぞれが、ダイススコアとハウスドルフ距離の点で改善につながることがわかりました。さまざまな最適化手順でトレーニングされた3つのディープニューラルネットワークのアンサンブルを使用して、BraTS2020チャレンジの検証データセットで有望な結果を達成しました。私たちのアンサンブルは、BraTS2020チャレンジのセグメンテーションタスクで登録された693チームのうち4位にランクされました。
Training a deep neural network is an optimization problem with four main ingredients: the design of the deep neural network, the per-sample loss function, the population loss function, and the optimizer. However, methods developed to compete in recent BraTS challenges tend to focus only on the design of deep neural network architectures, while paying less attention to the three other aspects. In this paper, we experimented with adopting the opposite approach. We stuck to a generic and state-of-the-art 3D U-Net architecture and experimented with a non-standard per-sample loss function, the generalized Wasserstein Dice loss, a non-standard population loss function, corresponding to distributionally robust optimization, and a non-standard optimizer, Ranger. Those variations were selected specifically for the problem of multi-class brain tumor segmentation. The generalized Wasserstein Dice loss is a per-sample loss function that allows taking advantage of the hierarchical structure of the tumor regions labeled in BraTS. Distributionally robust optimization is a generalization of empirical risk minimization that accounts for the presence of underrepresented subdomains in the training dataset. Ranger is a generalization of the widely used Adam optimizer that is more stable with small batch size and noisy labels. We found that each of those variations of the optimization of deep neural networks for brain tumor segmentation leads to improvements in terms of Dice scores and Hausdorff distances. With an ensemble of three deep neural networks trained with various optimization procedures, we achieved promising results on the validation dataset of the BraTS 2020 challenge. Our ensemble ranked fourth out of the 693 registered teams for the segmentation task of the BraTS 2020 challenge.
updated: Tue Nov 03 2020 10:50:48 GMT+0000 (UTC)
published: Tue Nov 03 2020 10:50:48 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アソシエイト