arXiv reaDer
3D 医用画像のメモリ効率の高い処理のための拡散モデル
Diffusion Models for Memory-efficient Processing of 3D Medical Images
ノイズ除去拡散モデルは、最近、多くの画像生成タスクで最先端のパフォーマンスを達成しました。ただし、大量の計算リソースが必要です。これにより、高解像度の 3 次元データのような大規模な 3D ボリュームを扱うことが多い医療タスクへの適用が制限されます。この作業では、3D 拡散モデルのリソース消費を削減し、それらを 3D 画像のデータセットに適用するさまざまな方法を紹介します。このホワイト ペーパーの主な貢献は、メモリ効率の高いパッチ ベースの拡散モデル PatchDDM です。これは、トレーニングがパッチでのみ実行されている間、推論中に総ボリュームに適用できます。提案された拡散モデルは任意の画像生成タスクに適用できますが、BraTS2020 データセットの腫瘍セグメンテーション タスクでこの方法を評価し、意味のある 3 次元セグメンテーションを生成できることを示します。
Denoising diffusion models have recently achieved state-of-the-art performance in many image-generation tasks. They do, however, require a large amount of computational resources. This limits their application to medical tasks, where we often deal with large 3D volumes, like high-resolution three-dimensional data. In this work, we present a number of different ways to reduce the resource consumption for 3D diffusion models and apply them to a dataset of 3D images. The main contribution of this paper is the memory-efficient patch-based diffusion model PatchDDM, which can be applied to the total volume during inference while the training is performed only on patches. While the proposed diffusion model can be applied to any image generation tasks, we evaluate the method on the tumor segmentation task of the BraTS2020 dataset and demonstrate that we can generate meaningful three-dimensional segmentations.
updated: Mon Mar 27 2023 15:10:19 GMT+0000 (UTC)
published: Mon Mar 27 2023 15:10:19 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アソシエイト