arXiv reaDer
BerDiff: 医用画像セグメンテーションのための条件付きベルヌーイ拡散モデル
BerDiff: Conditional Bernoulli Diffusion Model for Medical Image Segmentation
医用画像のセグメンテーションは、不明確な腫瘍境界や複数のもっともらしい注釈などの要因に起因する、固有のあいまいさと高い不確実性を伴う困難な作業です。セグメンテーション マスクの精度と多様性は、臨床現場で放射線科医に貴重な参考資料を提供するために非常に重要です。既存の拡散モデルは、さまざまなビジュアル生成タスクで強力な能力を示していますが、セグメンテーションで個別のマスクを処理することは依然として困難です。正確で多様な医用画像セグメンテーション マスクを実現するために、医用画像セグメンテーション (BerDiff) 用の新しい条件付きベルヌーイ拡散モデルを提案します。ガウス ノイズを使用する代わりに、最初にベルヌーイ ノイズを拡散カーネルとして使用して、バイナリ セグメンテーション タスクの拡散モデルの容量を強化し、より正確なセグメンテーション マスクを作成することを提案します。次に、拡散モデルの確率的性質を利用することで、BerDiff は初期ベルヌーイ ノイズと中間潜在変数をランダムに複数回サンプリングして、さまざまなセグメンテーション マスクを生成します。 .さらに、BerDiff は、逆拡散の全体的な軌跡からサブシーケンスを効率的にサンプリングできるため、セグメンテーション プロセスが高速化されます。モダリティが異なる 2 つの医用画像セグメンテーション データセットに関する広範な実験結果は、BerDiff が最近公開された他の最先端の方法よりも優れていることを示しています。私たちの結果は、拡散モデルが医療画像セグメンテーションの強力なバックボーンとして機能する可能性があることを示唆しています。
Medical image segmentation is a challenging task with inherent ambiguity and high uncertainty, attributed to factors such as unclear tumor boundaries and multiple plausible annotations. The accuracy and diversity of segmentation masks are both crucial for providing valuable references to radiologists in clinical practice. While existing diffusion models have shown strong capacities in various visual generation tasks, it is still challenging to deal with discrete masks in segmentation. To achieve accurate and diverse medical image segmentation masks, we propose a novel conditional Bernoulli Diffusion model for medical image segmentation (BerDiff). Instead of using the Gaussian noise, we first propose to use the Bernoulli noise as the diffusion kernel to enhance the capacity of the diffusion model for binary segmentation tasks, resulting in more accurate segmentation masks. Second, by leveraging the stochastic nature of the diffusion model, our BerDiff randomly samples the initial Bernoulli noise and intermediate latent variables multiple times to produce a range of diverse segmentation masks, which can highlight salient regions of interest that can serve as valuable references for radiologists. In addition, our BerDiff can efficiently sample sub-sequences from the overall trajectory of the reverse diffusion, thereby speeding up the segmentation process. Extensive experimental results on two medical image segmentation datasets with different modalities demonstrate that our BerDiff outperforms other recently published state-of-the-art methods. Our results suggest diffusion models could serve as a strong backbone for medical image segmentation.
updated: Mon Apr 10 2023 07:21:38 GMT+0000 (UTC)
published: Mon Apr 10 2023 07:21:38 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アソシエイト