arXiv reaDer
MIM-OOD: 医用画像の分布外検出のための生成マスク画像モデリング
MIM-OOD: Generative Masked Image Modelling for Out-of-Distribution Detection in Medical Images
教師なし分布外 (OOD) 検出では、健康な解剖学的構造の画像でトレーニングされたモデルのみを利用して画像内の異常領域を特定します。確立されたアプローチは、画像をトークン化し、自動回帰 (AR) モデルを使用してトークンの分布をモデル化することです。 AR モデルは、1) 異常なトークンを識別し、2) 配布中のトークンを使用して異常な表現をインペイントするために使用されます。ただし、AR モデルは推論時間が遅く、OOD 検出パフォーマンスに悪影響を与えるエラー蓄積の問題が発生する傾向があります。私たちの新しい手法である MIM-OOD は、AR モデルをタスク固有の 2 つのネットワークで置き換えることにより、速度とエラーの蓄積の問題の両方を克服します。1) 異常なトークンを識別するために最適化されたトランスフォーマー、2) マスクされた異常なトークンをペイントするために最適化されたトランスフォーマー画像モデリング (MIM)。脳 MRI 異常を使った実験では、MIM-OOD が AR モデルを大幅に上回り (DICE 0.458 vs 0.301)、約 25 倍の高速化 (9.5 秒 vs 244 秒) を達成していることが示されています。
Unsupervised Out-of-Distribution (OOD) detection consists in identifying anomalous regions in images leveraging only models trained on images of healthy anatomy. An established approach is to tokenize images and model the distribution of tokens with Auto-Regressive (AR) models. AR models are used to 1) identify anomalous tokens and 2) in-paint anomalous representations with in-distribution tokens. However, AR models are slow at inference time and prone to error accumulation issues which negatively affect OOD detection performance. Our novel method, MIM-OOD, overcomes both speed and error accumulation issues by replacing the AR model with two task-specific networks: 1) a transformer optimized to identify anomalous tokens and 2) a transformer optimized to in-paint anomalous tokens using masked image modelling (MIM). Our experiments with brain MRI anomalies show that MIM-OOD substantially outperforms AR models (DICE 0.458 vs 0.301) while achieving a nearly 25x speedup (9.5s vs 244s).
updated: Wed Aug 02 2023 20:20:58 GMT+0000 (UTC)
published: Thu Jul 27 2023 08:43:34 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アソシエイト