arXiv reaDer
潜在拡散モデルを使用した教師なし 3D 分布外検出
Unsupervised 3D out-of-distribution detection with latent diffusion models
3D データに対応する分布外 (OOD) 検出方法は、現実世界の臨床深層学習システムの重要なコンポーネントです。古典的なノイズ除去拡散確率モデル (DDPM) は、2D データセット上で再構築ベースの OOD 検出を実行する堅牢な方法として最近提案されていますが、3D データに自明にスケールすることはできません。この研究では、DDPM を高解像度 3D 医療データにスケーリングできるようにする潜在拡散モデル (LDM) の使用を提案します。提案されたアプローチを近および遠 OOD データセットで検証し、Latent Transformer Models (LTM) を使用して最近提案された 3D 対応アプローチと比較します。提案された LDM ベースのアプローチは、統計的に大幅に優れたパフォーマンスを達成するだけでなく、基礎となる潜在表現に対する感度が低くなり、より好ましいメモリ スケーリングを示し、より優れた空間異常マップを生成します。コードは https://github.com/marksgraham/ddpm-ood で入手できます。
Methods for out-of-distribution (OOD) detection that scale to 3D data are crucial components of any real-world clinical deep learning system. Classic denoising diffusion probabilistic models (DDPMs) have been recently proposed as a robust way to perform reconstruction-based OOD detection on 2D datasets, but do not trivially scale to 3D data. In this work, we propose to use Latent Diffusion Models (LDMs), which enable the scaling of DDPMs to high-resolution 3D medical data. We validate the proposed approach on near- and far-OOD datasets and compare it to a recently proposed, 3D-enabled approach using Latent Transformer Models (LTMs). Not only does the proposed LDM-based approach achieve statistically significant better performance, it also shows less sensitivity to the underlying latent representation, more favourable memory scaling, and produces better spatial anomaly maps. Code is available at https://github.com/marksgraham/ddpm-ood
updated: Fri Jul 07 2023 18:00:38 GMT+0000 (UTC)
published: Fri Jul 07 2023 18:00: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アソシエイト