arXiv reaDer
分布外検出のための拡散モデルのノイズ除去
Denoising diffusion models for out-of-distribution detection
機械学習システムを安全に展開するには、分布外の検出が不可欠です。現在、教師なし分布外検出は、生成モデルからの可能性の推定値またはその他の測定値を利用する生成ベースのアプローチによって支配されています。再構成ベースの方法は、サンプルが分布から外れているかどうかを判断するために再構成エラーの尺度を使用する代替アプローチを提供します。ただし、再構成ベースのアプローチは、モデルの情報のボトルネック (潜在次元のサイズなど) を慎重に調整して良好な結果を生成する必要があるため、あまり好まれません。この作業では、適用されるノイズの量によって、ボトルネックが外部から制御されるノイズ除去オートエンコーダーとして、ノイズ除去拡散確率モデル (DDPM) のビューを活用します。 DDPM を使用して、ある範囲のノイズ レベルにノイズを与えた入力を再構成し、結果として生じる多次元再構成エラーを使用して、分布外の入力を分類することを提案します。標準的なコンピューター ビジョン データセットと高次元の医療データセットの両方でアプローチを検証します。私たちのアプローチは、再構成ベースの方法だけでなく、最先端の生成ベースのアプローチよりも優れています。
Out-of-distribution detection is crucial to the safe deployment of machine learning systems. Currently, unsupervised out-of-distribution detection is dominated by generative-based approaches that make use of estimates of the likelihood or other measurements from a generative model. Reconstruction-based methods offer an alternative approach, in which a measure of reconstruction error is used to determine if a sample is out-of-distribution. However, reconstruction-based approaches are less favoured, as they require careful tuning of the model's information bottleneck - such as the size of the latent dimension - to produce good results. In this work, we exploit the view of denoising diffusion probabilistic models (DDPM) as denoising autoencoders where the bottleneck is controlled externally, by means of the amount of noise applied. We propose to use DDPMs to reconstruct an input that has been noised to a range of noise levels, and use the resulting multi-dimensional reconstruction error to classify out-of-distribution inputs. We validate our approach both on standard computer-vision datasets and on higher dimension medical datasets. Our approach outperforms not only reconstruction-based methods, but also state-of-the-art generative-based approaches.
updated: Mon Mar 06 2023 01:35:21 GMT+0000 (UTC)
published: Mon Nov 14 2022 20:35:11 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アソシエイト