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

since
平成38年9月29日

← 検索結果に戻る

3つの視覚モデルから知識を継ぐKPU:共通空間での照合と重みの保持

Knowledge Preservation and Unification (KPU)

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

Knowledge Preservation and Unification (KPU)の原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 1: Training paradigms for Vision Foundation Models. (a) Most of the data-centric models, which can be used out of the box, are trained based on large-scale labeled data  ( Lu et al. 2023 ) . (b) Existing model-driven approaches inherit pre-learned visual knowledge from diverse teachers by feature alignment on their latent spaces  ( Ranzinger et al. 2024b ) . (c) Our proposed Knowledge Preservation and Unification ( KPU ) method combines knowledge preservation and unified transfer to aggregate complementary insights from multiple pre-trained models upon a general-purpose sentinel teacher.

出典:Knowledge Preservation and Unification (KPU) / arXiv ↗

概要

  • KPUは3つの教師の特徴を共通空間と教師固有の空間で照合し、DINOv2の重みを固定したまま補助層を学習する。
  • 著者らの表1では、画像特徴を固定したKPUの四課題平均は59.3、同条件のDINOv2は58.7。ただし意味分割はDINOv2の値が高い。
  • 工場での部品検出は編集上の応用案であり、現場画像と検出枠の正解ラベルを使った検証が必要である。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、Knowledge Preservation and Unification (KPU)の仕組み

左から右へ読む
  1. 01ラベルなし画像
  2. 023教師の特徴抽出
  3. 03共通・固有空間で照合
  4. 04生徒の画像特徴

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

研究の背景

視覚基盤モデルは、画像分類、物体検出、領域分割などに使う画像特徴を作るモデルである。その開発には大量の画像や、物体名・位置などを記した正解ラベルが必要になる場合があり、収集や学習の負担が大きい。

一方、既に学習されたモデルは、用途ごとに異なる画像の捉え方を身につけている。KPUはそれらを教師として使い、一つの生徒モデルへ知識を継がせて、複数の画像処理に使える特徴を得ようとする。

手法

KPUはラベルなし画像をCLIP、GroundingDINO、DINOv2の3つの教師と生徒モデルに入力する。各モデルが出す「特徴」は、画像全体や画像内の位置ごとの内容を表す数値だ。同じ画像に対する教師と生徒の特徴が近づくように学習する。

教師の特徴を共通の数値空間に写して生徒と照合し、元の特徴を復元する学習も行う。生徒の特徴を教師ごとの空間に合わせた照合も併用する。生徒の土台となるDINOv2の重みは固定し、追加の補助層を更新する。学習後は教師の画像エンコーダーを使わず、生徒が画像特徴を出す。

新規性

KPUは、生徒と教師の特徴を教師ごとの空間で照合する学習に加え、各教師の特徴を共通空間へ写して生徒と照合する。共通空間から元の特徴を復元する学習も行い、写し替えに伴う情報損失を抑える設計だ。

さらに、汎用教師DINOv2の学習済みの重みを固定し、他の教師から得る知識を追加の補助層に学ばせる。著者らは、この保持と統合によって教師間の知識継承の偏りを和らげると説明する。

従来手法との違い

従来のモデル主導手法には、生徒の特徴を教師ごとに別の特徴空間で合わせる方式がある。特徴空間とは、画像の内容を数値で表す形式を指す。教師ごとに数値の分布や大きさが異なると、照合時に一部の教師の影響が強くなりうる。

KPUは共通空間での照合を加え、汎用教師の重みも固定して保持する。著者らの表1で、画像特徴を固定し、同じ課題用デコーダーを使って再評価したDINOv2と比べると、KPUの四課題の平均値は59.3、DINOv2は58.7だった。

実験結果

著者らはImageNet-21Kの学習用画像とOpenImagesの一部、計約1500万枚でKPUを学習した。評価には分類でImageNet-1K、検出と個体分割でCOCO、意味分割でADE20Kを使用。指標は順にTop-1正解率、検出枠のAPb、マスクのAPm、mIoUである。

表1で著者らが報告する、画像特徴を固定したKPUの意味分割・個体分割・検出・分類の値は順に52.9、47.3、52.8、84.3で、四課題の平均は59.3。同じく特徴を固定して評価したDINOv2は53.2、45.8、52.7、82.9、平均58.7であり、意味分割ではDINOv2が上回った。

応用の可能性

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

編集上の応用案:工場で撮影した部品画像から、指定した部品の位置を枠で示す試行を考える。入力は撮影用カメラの画像と部品名、出力は検出枠とし、担当者が枠を確認する流れにする。

候補となる構成は、KPUの画像特徴とGroundingDINOの検出用デコーダーを組み合わせるものだ。現場向けに適応するなら対象工程のラベルなし画像と3つの教師モデルが、精度の評価には別途、部品位置の枠ラベルを付けた画像が必要になる。現場での精度は未確認である。

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

導入時の検証

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

導入時の検証案:対象工程を一つ選び、撮影用カメラのラベルなし画像とCLIP、GroundingDINO、DINOv2を用意する。論文の構成に沿ってDINOv2を固定し、補助層を適応させる。評価用には、学習に使う画像と分けて、部品の検出枠に正解ラベルを付ける。

同じ評価画像でGroundingDINO単体とKPUの特徴をその検出用デコーダーにつないだ構成を比べる。検出枠のAPb、見逃し件数、画像一枚当たりの処理時間を記録し、部品の種類と撮影条件ごとに確認する。利点が現場でも残るかは、この比較で判断する。

限界と課題

著者らは、教師間で特徴の分布や大きさが異なると、知識継承が一部の教師に偏りうると指摘する。KPUはこの問題への対処として提案されているが、確認できた本文の範囲では、工場画像での性能は示されていない。

編集上の検証事項は、現場画像での誤検出と見逃し、教師や検出用デコーダーの組み合わせを変えた場合の結果である。また、教師のデコーダーを使う追加の課題別ラベルなしの利用と、課題別ラベルでデコーダーを学習した表1の評価は分けて読む必要がある。

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

出典

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

Seeing Further on the Shoulders of Giants: Knowledge Inheritance for Vision Foundation Models

Vision foundation models (VFMs) are predominantly developed using data-centric methods. These methods require training on vast amounts of data usually with high-quality labels, which poses a bottleneck for most institutions that lack both large-scale data and high-end GPUs. On the other hand, many open-source vision models have been pretrained on domain-specific data, enabling them to distill and represent core knowledge in a form that is transferable across diverse applications. Even though these models are highly valuable assets, they remain largely under-explored in empowering the development of a general-purpose VFM. In this paper, we present a new model-driven approach for training VFMs through joint knowledge transfer and preservation. Our method unifies multiple pre-trained teacher models in a shared latent space to mitigate the ``imbalanced transfer'' issue caused by their distributional gaps. Besides, we introduce a knowledge preservation strategy to take a general-purpose teacher as a knowledge base for integrating knowledge from the remaining purpose-specific teachers using an adapter module. By unifying and aggregating existing models, we build a powerful VFM to inherit teachers' expertise without needing to train on a large amount of labeled data. Our model not only provides generalizable visual features, but also inherently supports multiple downstream tasks. Extensive experiments demonstrate that our VFM outperforms existing data-centric models across four fundamental vision tasks, including image classification, object detection, semantic and instance segmentation.

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