教師あり深層学習は、医療画像のセグメンテーションで有望なパフォーマンスを達成しましたが、多くの方法では、目に見えないデータをうまく一般化できず、実際の適用性が制限されます。この問題に対処するために、セグメンテーションのために医用画像のドメインに関係のない輪郭を利用して、画像とラベルの統計を共同でモデル化するディープラーニングベースのベイズフレームワークを提案します。具体的には、最初に画像を輪郭と基底のコンポーネントに分解します。次に、期待されるラベルを、輪郭のみに関連する変数としてモデル化します。最後に、変分ベイズフレームワークを開発して、等高線、基底、ラベルなど、これらの変数の事後分布を推測します。フレームワークはニューラルネットワークで実装されているため、ディープベイズセグメンテーションと呼ばれます。クロスシーケンス心臓MRIセグメンテーションのタスクの結果は、私たちの方法がモデルの一般化可能性のための新しい最先端を設定したことを示しています。特に、LGE MRIでトレーニングされたBayeSegモデルは、T2画像で一般化され、他のモデルよりも優れており、平均ダイスで0.47を超えています。私たちのコードはhttps://zmiclab.github.io/projects.htmlで入手できます。
Although supervised deep-learning has achieved promising performance in medical image segmentation, many methods cannot generalize well on unseen data, limiting their real-world applicability. To address this problem, we propose a deep learning-based Bayesian framework, which jointly models image and label statistics, utilizing the domain-irrelevant contour of a medical image for segmentation. Specifically, we first decompose an image into components of contour and basis. Then, we model the expected label as a variable only related to the contour. Finally, we develop a variational Bayesian framework to infer the posterior distributions of these variables, including the contour, the basis, and the label. The framework is implemented with neural networks, thus is referred to as deep Bayesian segmentation. Results on the task of cross-sequence cardiac MRI segmentation show that our method set a new state of the art for model generalizability. Particularly, the BayeSeg model trained with LGE MRI generalized well on T2 images and outperformed other models with great margins, i.e., over 0.47 in terms of average Dice. Our code is available at https://zmiclab.github.io/projects.html.