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

since
平成38年9月29日

← 検索結果に戻る

少数クラス・ラベル不足・撮影条件の変化に画像AIはどう対応するか

Learning from Limited and Imperfect Data

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

概要

  • 本稿は、長尾分布での画像生成・分類、ラベル不足、利用先での分布の変化を四つの課題として扱う。
  • NoisyTwinsはクラスごとに異なる分布を使って画像を生成し、CSSTはラベルなし画像も使う分類を長尾分布へ拡張する。
  • 著者らは生成性能などの改善を述べるが、確認できた本文の範囲では測定値や比較表がなく、改善幅は判断できない。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、Learning from Limited and Imperfect Dataの仕組み

左から右へ読む
  1. 01ラベル付き・なし画像
  2. 02自己学習
  3. 03検証指標の反映
  4. 04分類モデル

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

研究の背景

現実に集まる画像では、よく見られる種類の件数が多く、珍しい種類は少ない。これを長尾分布と呼ぶ。クラスの件数をそろえるためにラベル付き画像を捨てると、収集と注釈にかけた資源も失われる。

さらに、ラベルのない画像が多い場合や、学習時と利用時で画像の分布が変わる場合もある。著者は画像の生成、少数クラスの分類、少ないラベルでの学習、利用先への適応を分けて取り上げる。

手法

図は半教師あり学習のCSSTに焦点を当てたもの。半教師あり学習とは、正解ラベル付き画像にラベルなし画像も加えて学ぶ方法で、自己学習ではモデル自身の予測を学習に使う。CSSTはこの方法を長尾分布へ拡張し、別に確保した検証用データでの最悪クラスの再現率などを学習に反映する。例えば、珍しい不良種別を含む画像と未分類の画像から、不良種別を判定するモデルを学ぶ用途を考えられる。

他の課題には別の方法を示す。生成のNoisyTwinsは、画像生成の元になる数値表現をクラスごとに異なる分布として設計し、同じクラス内の多様さと指定クラスとの一致を学ぶ。分類のSAMは少数クラスで損失が小さい解への収束を促し、DeiT-LTはCNNから画像認識モデルへ知識を移す。利用先への適応では、ラベルを付ける対象画像の選択とSDATを扱う。

新規性

著者は、クラス間の件数差がある画像の生成と分類、ラベルなし画像を使う分類、利用先への適応を四つの課題として整理し、それぞれに方法を示す。単一の手法が四課題を解くという主張ではない。

生成ではクラスごとに画像生成の元となる数値表現の分布を分けるNoisyTwins、分類ではSAMとDeiT-LT、半教師あり学習ではCSSTとSelMix、適応ではラベルを付ける画像の選択とSDATを扱う。

従来手法との違い

著者らが調べたGAN(画像を生成するモデル)の変種には、生成画像の種類が偏る、または一部のクラスが出ない問題があった。gSRは偏りの緩和を狙うが、大規模データでは生成画像のクラスが混同されるため、NoisyTwinsはクラスごとに異なる分布から生成する。

分類では、少数クラスの損失を重くするだけでは学習が鞍点に収束し、汎化を妨げるという観察からSAMを用いる。半教師あり学習のCSSTは自己学習を長尾分布に拡張し、SelMixはラベル付き画像とラベルなし画像の分布が似ているという前提を緩める。

実験結果

著者らは、NoisyTwinsで高解像度のStyleGANを長尾データのImageNet-LTとiNaturalist2019に適用し、クラスとの一致を保ちながら当時の最良水準に達したと述べる。また、gSRが長尾データで条件付きGANのSNGANとBigGANの性能を改善したと報告する。

半教師あり学習では、最悪クラスの再現率と再現率の調和平均を重視すると説明する。前者は最も拾いにくいクラスの正解率に関わり、後者は低い値の影響を受けやすい平均である。確認できた本文の範囲では、各手法の測定値、データ分割、比較表は示されておらず、改善幅は数値で判断できない。

応用の可能性

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

編集上の応用案:一つの製造ラインで、発生件数の少ない外観不良を画像から分類する。入力は固定した撮影工程のカメラ画像と、良品・不良種別を付けた画像、ラベルなし画像。分類結果を不良種別として出し、担当者の確認へ回す運用を想定する。

候補はCSSTとし、少数クラスの見落としに関わる指標を検証用画像で確認しながら学習する。撮影条件や製品が変わる場合は、移行先の画像を集め、一部にラベルを付けて別途適応を検討する。製造画像での効果は記事中の結果からは判断できない。

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

導入時の検証

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

導入時の検証案:対象ラインのカメラで撮った製品画像を集め、良品と不良種別のラベルがある画像、ラベルなし画像、評価用画像に分ける。各種別の件数と撮影条件を記録し、ラベル付き画像だけで学ぶ分類モデルを基準にする。

同じ評価用画像でCSST候補と基準を比べ、種別ごとの再現率、最悪クラスの再現率、見落とし件数を確認する。別ラインへ移すなら、そのラインの画像と少数の正解ラベルを用意し、適応前後を同じ条件で測る。必要なラベル数も記録する。

限界と課題

著者らは、調べたGANの変種で生成の偏りやクラスの欠落を確認し、gSRを大規模データに使うと生成画像にクラスの混同が残ると述べる。また、多数クラスから少数クラスへ移る知識の量を定量化することを今後の研究課題に挙げている。

編集上の検証事項として、各手法の測定値、比較条件、学習負担を原論文で確認する必要がある。製造ラインへの応用では、少数の不良種別を見落とす割合に加え、撮影条件の変更後にも判定が保たれるかを別々に測る。

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

出典

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

Learning from Limited and Imperfect Data

The datasets used for Deep Neural Network training (e.g., ImageNet, MSCOCO, etc.) are often manually balanced across categories (classes) to facilitate learning of all the categories. This curation process is often expensive and requires throwing away precious annotated data to balance the frequency across classes. This is because the distribution of data in the world (e.g., internet, etc.) significantly differs from the well-curated datasets and is often over-populated with samples from common categories. The algorithms designed for well-curated datasets perform suboptimally when used to learn from imperfect datasets with long-tailed imbalances and distribution shifts. For deep models to be widely used, getting away with the costly curation process by developing robust algorithms that can learn from real-world data distribution is necessary. Toward this goal, we develop practical algorithms for Deep Neural Networks that can learn from limited and imperfect data present in the real world. These works are divided into four segments, each covering a scenario of learning from limited or imperfect data. The first part of the works focuses on Learning Generative Models for Long-Tail Data, where we mitigate the mode-collapse for tail (minority) classes and enable diverse aesthetic image generations as head (majority) classes. In the second part, we enable effective generalization on tail classes through Inductive Regularization schemes, which allow tail classes to generalize as the head classes without enforcing explicit generation of images. In the third part, we develop algorithms for Optimizing Relevant Metrics compared to the average accuracy for learning from long-tailed data with limited annotation (semi-supervised), followed by the fourth part, which focuses on the effective domain adaptation of the model to various domains with zero to very few labeled samples.

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