arXiv reaDer
半教師あり体積医療画像セグメンテーションのためのベイズ深層学習法の再考
Rethinking Bayesian Deep Learning Methods for Semi-Supervised Volumetric Medical Image Segmentation
最近、半教師あり医療画像セグメンテーションのために、いくつかのベイズ深層学習法が提案されています。彼らは医療ベンチマークで有望な結果を達成しましたが、いくつかの問題がまだ存在しています。まず、全体的なアーキテクチャは識別モデルに属しているため、トレーニングの初期段階では、ラベル付きデータのみをトレーニングに使用します。これにより、ラベル付きデータに過剰適合する可能性があります。第二に、実際には、それらの全体的なアーキテクチャはベイジアンフレームワークの下で設計されていないため、それらはベイジアンディープラーニングに部分的にのみ基づいています。ただし、ベイジアンの観点からアーキテクチャ全体を統合すると、アーキテクチャに厳密な理論的基礎を持たせることができるため、アーキテクチャの各部分に明確な確率的解釈を持たせることができます。したがって、問題を解決するために、新しい生成ベイズ深層学習(GBDL)アーキテクチャを提案します。 GBDLは生成モデルに属しており、そのターゲットは、入力された医療量とそれに対応するラベルの同時分布を推定することです。同時分布の推定には暗黙的にデータの分布が含まれるため、ラベル付きデータとラベルなしデータの両方をトレーニングの初期段階で利用できます。これにより、潜在的な過剰適合の問題が軽減されます。さらに、GBDLは完全にベイジアンフレームワークの下で設計されているため、完全なベイジアン定式化を行います。これにより、アーキテクチャの理論的な確率的基盤が築かれます。広範な実験により、GBDLは、3つの公的医療データセットで一般的に使用される4つの評価指標に関して、以前の最先端の方法よりも優れていることが示されています。
Recently, several Bayesian deep learning methods have been proposed for semi-supervised medical image segmentation. Although they have achieved promising results on medical benchmarks, some problems are still existing. Firstly, their overall architectures belong to the discriminative models, and hence, in the early stage of training, they only use labeled data for training, which might make them overfit to the labeled data. Secondly, in fact, they are only partially based on Bayesian deep learning, as their overall architectures are not designed under the Bayesian framework. However, unifying the overall architecture under the Bayesian perspective can make the architecture have a rigorous theoretical basis, so that each part of the architecture can have a clear probabilistic interpretation. Therefore, to solve the problems, we propose a new generative Bayesian deep learning (GBDL) architecture. GBDL belongs to the generative models, whose target is to estimate the joint distribution of input medical volumes and their corresponding labels. Estimating the joint distribution implicitly involves the distribution of data, so both labeled and unlabeled data can be utilized in the early stage of training, which alleviates the potential overfitting problem. Besides, GBDL is completely designed under the Bayesian framework, and thus we give its full Bayesian formulation, which lays a theoretical probabilistic foundation for our architecture. Extensive experiments show that our GBDL outperforms previous state-of-the-art methods in terms of four commonly used evaluation indicators on three public medical datasets.
updated: Sat Jun 18 2022 23:07:04 GMT+0000 (UTC)
published: Sat Jun 18 2022 23:07:04 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アソシエイト