自動車を製造する工場、私たちの体をスキャンする病院、そして私たちのデータを保護するネットワークにおいて、静かな守護者が必要とされています。それは、たった一つの「異常」を見つけ出すことができるシステムです。これが異常検知の役割です。泥棒の姿を正確に知っている警備員とは異なり、これらのシステムは、多くの場合、「正常なもの」——完璧な歯車、健康な組織、クリーンなコード——のみに基づいて学習されます。彼らは「正しいもの」の形を非常に深く学習するため、何かがわずかでも逸脱したとき、警報を鳴らすのです。環境が変化すると、課題は大きくなります。例えば、新しいタイプのエンジン部品を次々と生産し始める工場を想像してみてください。最初の部品のみで学習したシステムは、新しいものを学習した途端に、かつてのものの認識方法を忘れてしまうかもしれません。あるいは、これまでに見たすべての詳細を保存するためのメモリ容量が足りなくなるかもしれません。これは、新しいことを学ぶことで機械が古い知識を失ってしまう「破滅的忘却」の問題であり、また、システムが無限の事例リストを保持できないという「メモリ制限」の問題でもあります。
研究者たちは、すべての正常なデータの「完璧な要約」、つまり全体を代表する主要な事例のコレクションを作成することで、この問題を解決しようと長く試みてきました。このような要約を構築する最善の方法は、互いにできるだけ異なる事例を選び出し、正常な世界のどの部分も取り残されないようにすることです。しかし、新しいタスクが次々とやってくる場合、この要約を無限に大きくすることなく完璧に保つことは極めて困難です。単に新しい事例を追加すればメモリがいっぱいになります。スペースを作るために古いものを削除すれば、元のタスクを認識する能力を失うリスクがあります。ユン・ギョ・ジョン(Yoon Gyo Jung)氏率いる研究チームは、どれほど多くの新しいタスクに直面しても、この要約を小さく、かつ完璧に保つ方法を発見しました。彼らは「ContCore」と呼ばれる手法を開発し、機械が過去のものを決して忘れることなく、連続的な新しい正常パターンの流れを学習し続けられるようにし、同時に固定されたメモリ量を使用することを可能にしました。
彼らの発見の核心は、メモリをどのように更新するかという点にあります。あらゆる新しい事例を保存しようとしたり、容易に忘却してしまう複雑なニューラルネットワークを使用したりする代わりに、彼らはシンプルですが強力な「選択」の戦略を用います。新しい正常なデータのバッチが到着すると、システムはまず、すでに記憶しているものと最も異なる新しい事例を探します。そして、これらの「最も遠い」事例をコレクションに加えるために選び出します。次に、総事例数が厳格なメモリ制限内に収まるように、二度目の選択を行います。システムは全コレクション(古い事例と新しい事例の両方)を見渡し、最も広く分散している事例のグループを選び出します。彼らが「貪欲な拡張(greedy expansion)」と「貪意な集約(greedy consolidation)」と呼ぶこの二段階のプロセスにより、メモリには常に、これまでに見たすべてのものを最も代表するサンプルが含まれるようになります。
研究者たちは、この手法が「正常なデータの形」を維持することによって機能することを明らかにしました。このシステムは、メモリ予算内に収めるために各ステップで多くの事例を破棄しますが、保持される事例は非常に慎重に選ばれているため、正常な変動の全範囲を依然としてカバーしています。彼らは、この小さく常に更新されるメモリと、完璧で無制限のメモリとの間のギャップが、常に小さく制御されていることを数学的に証明しました。実用面では、これはシステムが他の手法を悩ませる破滅的忘却に陥らないことを意味します。単純な単一クラスの更新から、多くの異なるクラスが連なる複雑なシーケンスに至るまで、幅広いタスクでテストを行った結果、彼らのシステムは既存の最先端の手法を一貫して上回りました。標準的な産業用データセットにおいて、欠陥を特定する精度が高く、かつ以前に学習した内容をほとんど忘れることなく達成しました。
このアプローチを特に堅牢なものにしているのは、その効率性です。膨大な計算能力や大量のデータを必要として再学習を行う他の手法とは異なり、このシステムはシンプルな選択プロセスによってメモリを更新します。古いタスクを再学習したり、膨大な生の画像を保存したりする必要はありません。システムがデータを一枚の画像ずつ学習しなければならないシナリオ(他の手法が大幅に失敗する場面)においても、ContCoreは高いパフォーマンスを維持しました。研究者たちはまた、この手法が異なる種類のデータやハードウェアにおいてもうまく機能することを示し、注意深く代表的なグループを選択するという原理が、単に計算能力を投入することよりも強力であることを証明しました。メモリを限定的ながらも代表性を高く保つことで、彼らは、機械が過去を失うことなく変化する世界に適応するための信頼できる方法を提供したのです。
技術要約:継続的異常検知のためのメモリ制限付き貪欲サンプリングの継続
問題定義
教師なし異常検知(AD)は、製造業やサイバーセキュリティなどの産業において極めて重要であるが、正常な境界と無制限な異常の間の決定境界を発見することは依然として困難である。初期のモデルはクラスごとに個別のモデルを訓練することで高い性能を達成したが、これらはアノテーションコストが高く、モデルサイズも大きくなるという課題がある。マルチクラスAD手法は、複数のクラスに対して単一のモデルを訓練することでこれに対処しているが、新しいタスクが逐次的に到着する動的な環境においては、**破滅的忘却(catastrophic forgetting)**のために失敗する。
**継続的異常検知(CAD)**における具体的な課題は、教師なしの設定(クラスラベルはなく、正常な訓練画像のみが存在する)において、破滅的忘却を防ぎつつ、新たに導入された異常を検出することである。既存のアプローチには以下のような重大な制限がある:
- 正則化ベースの手法(例:EWC, SI)は、安定性と可塑性のバランスを取るのが難しく、ハイパーパラメータに敏感である。
- リプレイベースの手法は、メモリ使用量がタスク数に比例して増加するため、スケーラビリティの問題に直面する。
- コアセットベースの手法(例:DNE, UCAD)は再学習を回避しているが、代表性を維持するために通常、無制限のメモリを必要とするため、メモリオーバーフローに直面する。
- 専用のCADソリューション(例:CDAD, IUF)は、長期的なスケジュールにおいて破滅的忘却に陥ったり、クラスラベルを必要としたりすることがあり、適用性が制限される。
核心となる問いは、「既存の無制限なコアセットの表現品質を維持しつつ、境界のあるコアセットを設計できるか?」である。
手法:ContCore
著者らは、厳格かつ固定されたメモリ制約の下で、貪欲に継続されるコアセット(greedy-continued coreset)を構築するメモリベースの異常検知器であるContCoreを提案している。この手法は、以前の貪欲にサンプリングされた集合に対して貪欲な選択を反復的に適用する「継続的な貪欲サンプリング」が、代表性を効果的に維持するという観察に基づいている。
コアメカニズム
ContCoreは、各タスクの遷移 t において以下の2つのステップで作動する。
- 貪欲な拡張(Greedy Expansion): 新しいタスクからのパッチ埋め込み特徴量 Zt が与えられたとき、この手法は現在のコアセット Mt−1 から最大限に離れた特徴量を選択する。これは、ベースセットに対する貪欲サンプリングとして定式化される:
Gnt(Zt;Mt−1)={zi∗∈Zt:zi∗=argz∈Ztmaxd(z,Zi−1∗∪Mt−1)}
ここで、nt は現在のタスクにおけるサンプリングサイズである。
- 貪欲な統合(Greedy Consolidation): 新たに拡張された特徴量と以前のコアセットの和集合を処理し、固定メモリ予算 m を強制する。この手法は、この結合された集合から、最大限に分離された特徴量を選択する:
Mt=Gm(Gnt(Zt;Mt−1)∪Mt−1)
実装上の工夫
- 効率的な近似: 統合時の厳格な貪欲サンプリングに伴う計算コストを軽減するため、著者らは最近傍探索を用いてプロセスを近似している。全探索による貪欲な選択ではなく、最近傍距離に基づいてフラクション q のサンプルを選択することで、カバレッジを維持しながら計算複雑性を大幅に削減している。
- 推論: 異常スコアは、PatchCoreの手法に従い、各パッチについて現在のコアセット Mt への最短距離に基づいて算出される。
理論的根拠
本論文は、結果として得られる貪欲継続コアセット MT が、オラクル・コアセット(全タスクの全データに対して貪欲サンプリングを行った結果)を有界なギャップ内で近似することを示す理論的保証(定理4.1)を提供している。オラクルと構築されたコアセットのハウスドルフ距離は、タスクごとの近似誤差の総和によって抑えられることが示されており、代表性が破滅的ではなく緩やかに劣化することを証明している。
主な貢献
- 観察と理論: 著者らは、継続的な貪欲サンプリングが厳格なメモリ制限下でも代表性を維持することを実証した。また、貪欲継続コアセットがオラクル・コアセットを有界なギャップ内で近似することを示す理論的根拠を提示した。
- ContCoreアルゴリズム: 貪欲な拡張と統合を通じて、固定メモリ使用量を維持しながら貪欲継続コアセットを構築する、この原理を具現化した手法を提案した。
- 実験的性能: ContCoreは、MVTecADおよびVisAデータセットにおける11種類の異なるタスクスケジュールにおいて、最先端の性能を達成した。
- オンライン設定における堅牢性: 本手法は、オンライン継続的AD設定(シングルパス、バッチサイズ1)においても検証されており、従来のモデルが著しく性能を低下させる中で高い性能を維持している。
実験結果
実験は、MVTecADおよびVisAを用い、11種類の異なるタスクスケジュール(単純な増分型 1×15 から、複雑な混合型 10×1×5 まで)で行われた。
- 性能: ContCoreは、ほぼすべてのスケジュールにおいて、最も高いタスク平均AUROCと最も低い忘却度(Forgetting Measurement: FM)を達成した。例えば、MVTecADの 1×15 スケジュールにおいて、ContCoreは画像レベルAUROC **98.8%**を達成し、次に優れたモデル(UCADの93.0%)を上回り、忘却(FM: 0.1 vs UCADの1.0)を大幅に抑制した。
- クロスデータセット: クロスデータセット設定(MVTecAD → VisA およびその逆)において、ContCoreは従来のベースラインであるCDADを凌駕し、VisA → MVTecADの画像レベルAUROCにおいて8.9%の差で改善を示した。
- 効率性: ContCoreはニューラルネットワークを用いた手法と比較して非常に効率的である。CDAD(5628 GFLOPs, 11B パラメータ)に対し、ContCoreは大幅に少ないGFLOPs(9.24)とパラメータ数(45.3M)を必要とし、訓練および推論時間も高速である。
- 大規模検証: Real-IADデータセット(30クラス、約10万枚の画像)において、ContCoreは継続的タスクスケジュールで訓練されているにもかかわらず、UniAD-OracleおよびPatchCore-OneClassの両方のモデルを上回った。
- オンライン学習: 厳格な「オンラインCAD」設定(バッチサイズ1、シングルエポック)において、ContCoreは高い性能(MVTecADで97.2% AUROC)を維持したが、他の手法(DNE, UCAD, IUF, CDAD)は大幅な性能低下を招いた。
重要性と主張
本論文は、ContCoreがメモリ制限付きの継続的異常検知問題に対する堅牢なソリューションを提供すると主張している。その重要性は以下の点にある:
- メモリと性能のトレードオフの打破: 無制限のメモリを必要とするコアセット蓄積や、忘却に苦しむ正則化手法とは異なり、ContCoreは固定されたメモリフットプリントを維持しながら、最先端の性能を達成する。
- 理論的保証: この手法は単なる経験的なものではない。表現のギャップが最適オラクル・コアセットに対して有界であることを示す証明によって裏付けられている。
- 汎用性: このアプローチは、多様なタスクスケジュール、異なるバックボーンアーキテクチャ(WideResNet, EfficientNet, DINOv2)、さらにはデータを一度しか処理できない困難なオンライン学習シナリオにおいても有効である。
- 実用的な展開: 破滅的忘却や無制限のメモリ増大を回避することで、ContCoreは、過去のデータにアクセスできない状況で新しい異常クラスが継続的に導入される動的な環境における、実世界での展開に向けた実行可能なソリューションとして提示されている。
毎週最高の machine learning 論文をお届け。
スタンフォード、ケンブリッジ、フランス科学アカデミーの研究者に信頼されています。
受信トレイを確認して登録を完了してください。
問題が発生しました。もう一度お試しください。
スパムなし、いつでも解除可能。
週刊ダイジェスト — 最新の研究をわかりやすく。登録