arXiv reaDer
拡散モデルによる教師なし脳異常の高速検出とセグメンテーション
Fast Unsupervised Brain Anomaly Detection and Segmentation with Diffusion Models
深い生成モデルは、データの任意の異常を検出するための有望なツールとして登場し、手動でのラベル付けの必要性をなくしました。最近、自己回帰変圧器は、医用画像における異常検出のための最先端の性能を達成しました。それにもかかわらず、これらのモデルには、画像を1Dシーケンスとしてモデル化する必要がある、サンプリングプロセス中のエラーの蓄積、トランスフォーマーに関連する重要な推論時間など、いくつかの固有の弱点があります。ノイズ除去拡散確率モデルは、コンピュータービジョンで優れたサンプルを生成し(生成的敵対的ネットワークを超える)、高速な推論時間を持ちながら変圧器と競合する対数尤度を達成することが最近示された非自己回帰生成モデルのクラスです。拡散モデルは、オートエンコーダーによって学習された潜在表現に適用できるため、簡単に拡張でき、医用画像などの高次元データへの適用に最適です。ここでは、脳イメージングの異常を検出してセグメント化するための拡散モデルに基づく方法を提案します。健全なデータでモデルをトレーニングし、マルコフ連鎖全体での拡散と逆ステップを調査することで、潜在空間の異常領域を特定し、ピクセル空間の異常を特定できます。私たちの拡散モデルは、推論時間が大幅に短縮された合成および実際の病理学的病変を含む2D CTおよびMRIデータを使用した一連の実験で、自己回帰アプローチと比較して競争力のあるパフォーマンスを実現し、臨床的に実行可能にします。
Deep generative models have emerged as promising tools for detecting arbitrary anomalies in data, dispensing with the necessity for manual labelling. Recently, autoregressive transformers have achieved state-of-the-art performance for anomaly detection in medical imaging. Nonetheless, these models still have some intrinsic weaknesses, such as requiring images to be modelled as 1D sequences, the accumulation of errors during the sampling process, and the significant inference times associated with transformers. Denoising diffusion probabilistic models are a class of non-autoregressive generative models recently shown to produce excellent samples in computer vision (surpassing Generative Adversarial Networks), and to achieve log-likelihoods that are competitive with transformers while having fast inference times. Diffusion models can be applied to the latent representations learnt by autoencoders, making them easily scalable and great candidates for application to high dimensional data, such as medical images. Here, we propose a method based on diffusion models to detect and segment anomalies in brain imaging. By training the models on healthy data and then exploring its diffusion and reverse steps across its Markov chain, we can identify anomalous areas in the latent space and hence identify anomalies in the pixel space. Our diffusion models achieve competitive performance compared with autoregressive approaches across a series of experiments with 2D CT and MRI data involving synthetic and real pathological lesions with much reduced inference times, making their usage clinically viable.
updated: Tue Jun 07 2022 17:30:43 GMT+0000 (UTC)
published: Tue Jun 07 2022 17:30:43 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アソシエイト