表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

学習途中のGLOWで分布外画像を検出する:層別勾配の比較

OOD Detection with immature Models

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
著者の公式図・デモ

OOD Detection with immature Modelsの原論文から掲載した図です。細部はクリックして拡大できます。

出典:OOD Detection with immature Models / arXiv ↗

概要

  • GLOWを途中まで学習させ、層ごとの勾配から正常画像の基準を作り、分布外画像を検出する。
  • 著者らは、途中段階のモデルが学習後期のモデルと同等以上になる場合を報告するが、結果は画像群の組み合わせによって変わる。
  • 確認できた本文に組み合わせ別の表の数値はなく、導入には対象画像での比較が必要である。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、OOD Detection with immature Modelsの仕組み

左から右へ読む
  1. 01正常画像でGLOWを部分学習
  2. 02別の正常画像で層別基準作成
  3. 03判定画像の層別勾配計算
  4. 04分布外スコア

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

画像生成モデルGLOWは、画像に尤度という「モデルから見たもっともらしさ」を割り当てる。しかし、学習対象外の分布外画像に、学習対象の画像より高い尤度を与える場合があり、尤度だけでは判定を誤り得る。

分布外検出は、入力画像が学習時に想定した範囲に入るかを調べる処理だ。従来の勾配を使う方法には十分に学習したモデルを前提とするものがあり、著者らは学習途中のモデルでも検出できるかを調べた。

手法

まず正常側の画像でGLOWを途中まで学習させ、学習に使わなかった正常画像から各層の基準統計を作る。各画像について対数尤度の勾配、つまり画像を入力したときモデルの層ごとの値がどちらへ変わろうとするかを示す量を計算し、その大きさの二乗の対数を層別に記録する。

判定する画像にも同じ計算を行う。各層の値が正常画像から得た分布にどれほど合わないかを、正規分布に基づく負の対数尤度として合算し、分布外スコアにする。例えばSVHNを正常側、CIFAR-10を判定側とする設定では、1枚単位か5枚をまとめた単位で計算する。5枚の場合、出力はそのまとまりに対するスコアである。

新規性

著者らが検証したのは、画像生成モデルを十分に学習させなくても、層ごとの勾配を使う分布外検出で同等以上の結果を得られる場合がある、という点だ。層別勾配による採点方法そのものは先行研究に基づく。

さらに、学習段階ごとの検出スコアの分布を比較し、正常側と分布外側の隔たりや重なりが変わることを、性能差の説明として提示している。

従来手法との違い

主な比較は、同じ画像生成モデルGLOWを途中で止めた場合と、学習を進めた場合の分布外検出性能である。先行する層別勾配の方法は、十分に学習したモデルから得た統計量を利用していた。

画像の尤度、つまりモデルが画像に割り当てるもっともらしさだけを使う判定には、分布外画像の方が高い値になる場合がある。著者らの比較は画像群の組み合わせと学習段階に依存し、途中で止めれば常に改善するという結果ではない。

実験結果

著者らはGLOWを用い、3色画像のSVHN、GTSRB、CIFAR-10、CelebA、ImageNet32などを正常側や分布外側に設定し、学習段階を比べた。評価指標はAUROCで、正常側と分布外側をスコアで順位付けした際の分離性能を表す。1枚単位と5枚単位の判定を扱う。

著者らによると、5枚単位の図では、ImageNet32を正常側とした10エポックと250エポックでスコア分布の隔たりは広がるが、AUROCは変わらない。SVHNを正常側とした1エポックと250エポックでは、後期に分布の重なりが増す。確認できた本文の範囲では、参照先の表の個別数値は示されていない。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:工場の製品外観画像を入力し、通常品と異なる画像を検査員へ回す順番を決める。出力は画像ごとの分布外スコアとし、検査員が画像と現物を確認する流れを想定する。

既存カメラで撮った通常品の画像をモデル学習用と基準統計の作成用に分け、通常品・不良品の正解ラベル付き画像を別途評価用に確保する。撮影条件に合わせた画像の調整と判定基準の設定が必要で、この用途での改善は論文の結果からは確定しない。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:同じカメラで撮った通常品の画像を集め、通常品という区分を確認して、GLOWの学習用と層別基準の作成用に分ける。さらに、通常品と不良品の正解ラベルがある画像を評価用に確保し、撮影条件と画像の整え方を記録する。

同じ評価画像、同じ1枚単位の条件で、学習途中のモデル、学習を進めたモデル、尤度だけで判定する方法を比べる。AUROCに加え、設定した判定基準での通常品の誤検出率、不良品の見逃し率、判定時間を測る。停止時点と基準を選んだ後は、別の撮影日でも確かめる。

限界と課題

著者らは、SVHNを正常側とする比較で、学習後期に正常側と分布外側のスコア分布の重なりが増すと報告する。また、CIFAR-10とGTSRBでは学習画像を人工的に増やしても、1枚ずつの検出性能はほぼ完全な水準に改善しなかったと述べる。

編集上の検証事項は、対象画像で採用すべき停止時点、未知画像の種類が変わった場合の性能、スコア計算時間である。確認できた本文の範囲では、参照先の表にある組み合わせ別の数値や、実運用での誤検出件数までは確認できない。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

OOD Detection with immature Models

Likelihood-based deep generative models (DGMs) have gained significant attention for their ability to approximate the distributions of high-dimensional data. However, these models lack a performance guarantee in assigning higher likelihood values to in-distribution (ID) inputs, data the models are trained on, compared to out-of-distribution (OOD) inputs. This counter-intuitive behaviour is particularly pronounced when ID inputs are more complex than OOD data points. One potential approach to address this challenge involves leveraging the gradient of a data point with respect to the parameters of the DGMs. A recent OOD detection framework proposed estimating the joint density of layer-wise gradient norms for a given data point as a model-agnostic method, demonstrating superior performance compared to the Typicality Test across likelihood-based DGMs and image dataset pairs. In particular, most existing methods presuppose access to fully converged models, the training of which is both time-intensive and computationally demanding. In this work, we demonstrate that using immature models,stopped at early stages of training, can mostly achieve equivalent or even superior results on this downstream task compared to mature models capable of generating high-quality samples that closely resemble ID data. This novel finding enhances our understanding of how DGMs learn the distribution of ID data and highlights the potential of leveraging partially trained models for downstream tasks. Furthermore, we offer a possible explanation for this unexpected behaviour through the concept of support overlap.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01