アーカイブ / Computer Vision
クラス別の学習画像数を調整し、分類成績の偏りを抑えるBiCDO
BiCDO
BiCDOの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 2: The figure shows the BiCDO workflow and how it wraps around an existing training pipeline.
概要
- BiCDOは、固定した検証画像で測るクラス別成績に応じ、次の学習に使う画像数を更新する。
- 著者のCIFAR-10・EfficientNet-B3の比較では、ほぼ同じ総画像数でクラス別正解率の分散が88.7から2.3になった。
- 著者は複数目的での最適性を証明しておらず、収束には十分なクラス別画像数やモデルの収束などの条件がある。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、BiCDOの仕組み
- 01ラベル付き学習画像
- 02クラス別に画像選択
- 03学習と検証
- 04配分を更新
- 05調整後のモデル
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
画像分類では、各クラス、つまり分類先の学習画像を同じ枚数そろえても、クラス別の正解率までそろうとは限らない。クラスごとに成績が違えば、平均正解率だけでは見落としやすい弱点が残る。
BiCDOは、学習画像の枚数をクラスごとに調整し、検証時の成績の偏りを扱う方法である。例えば人と犬を区別する用途では、人を優先したいという目標も配分に反映できる。実務では、必要な画像をどのクラスで集めるか考える材料になる。
手法
BiCDOはまず、画像の保存先と正解ラベルを読み込み、クラスごとの利用可能枚数、初期の使用割合、使用割合の上下限を設定する。その割合に応じて学習画像を無作為に選び、既存の分類モデルを学習させる。検証画像は固定し、学習が進んだ期間のクラス別正解率を測る。
次に、平均より成績が低いクラスの使用割合を上げ、高いクラスは下げて、画像を選び直して再学習する。例えば「人」を優先する場合は、目標オフセットというクラス別の調整値を与えられる。反復の結果として、クラス別の使用画像数と、その配分で学習したモデルが得られる。
新規性
BiCDOの提案点は、分類先ごとの検証成績を次回の学習画像数に反映し、成績のばらつきが小さくなる配分を反復して探すことにある。成績を優先したいクラスには、目標オフセットという調整値も設定できる。
著者は得られる配分を「パレート最適化」と呼ぶが、複数の目的について最適性を証明してはいない。本文では、条件が整えばそれに近い配分を得られる、経験的な方法と位置付けている。
従来手法との違い
比較の基準は、各クラスの学習画像数をほぼそろえた配分である。BiCDOは同じ分類モデルの学習手順を繰り返し使い、固定した検証画像で測ったクラス別成績から、次に使う学習画像数を変える。
本文は、画像の再標本化や、学習時にクラスへ重みを付ける方法にも触れている。ただし、確認できた本文の範囲で数値を直接照合できるのは均等配分との比較である。他手法に対する優位性まで、この結果から判断することはできない。
実験結果
著者はCIFAR-10とiNaturalist21の画像分類で、EfficientNet、ResNet、ConvNeXtを用いて検証した。学習画像数を変え、固定した検証画像でクラス別正解率を測っている。前処理などの詳細は補足資料への参照にとどまり、確認できた本文の範囲では確かめられない。
CIFAR-10とEfficientNet-B3を用いた表4では、均等配分は平均正解率76.6％、クラス別正解率の分散88.7、使用画像19,737枚だった。BiCDOの配分はそれぞれ78.3％、2.3、19,740枚と著者は報告する。この差は、そのデータとモデルでほぼ同じ総画像数を使った比較の結果である。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：工場敷地のカメラ画像から対象を一つずつ切り出し、「人・犬・車両」のいずれかに分類する。入力は切り出した画像、出力は分類結果とクラス別の検証成績とする。人の見落としを重く扱うかは、運用目的に沿って先に決める。
この用途には、設置場所のカメラで撮った学習用画像と、画像ごとに一つ付けた正解ラベルが必要になる。対象の切り出し方とラベルの基準を定め、検証画像を分けてからBiCDOで配分を調整する。現場で有効かどうかは検証を要する。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：現場のカメラ画像から対象を一つずつ切り出し、「人・犬・車両」の正解ラベルを一画像に一つ付ける。撮影条件も記録し、学習用とは別に、配分を変えずに使う検証画像を確保する。切り出し方とラベル基準は現場の目的に合わせて決める。
まず各クラスをほぼ均等にし、総画像数をそろえた学習を基準にする。次に同じ分類モデルでBiCDOの配分を試し、クラス別正解率、平均正解率、クラス間の分散、人の見落としを比較する。必要なクラスで画像が足りるか、調整後の配分も記録して判断する。
限界と課題
著者は、配分の収束には各クラスに十分な画像があり、枚数の上下限に同時に達せず、元のモデルがデータ上で収束できることが必要だと述べる。また、クラス間の関係を含む指標や、一つの値にまとめられない指標では意味のある結果を得にくいとする。
編集上の検証事項として、現場画像でもクラス別成績が近づくか、優先クラスの改善に伴って他クラスの成績がどう変わるかを確かめたい。一枚に複数の対象が写る場合は、切り出しと正解ラベルの基準も事前に定める必要がある。
出典
元のタイトル・要旨を確認する（英語）
Optimizing Class Distributions for Bias-Aware Multi-Class Learning
We propose BiCDO (Bias-Controlled Class Distribution Optimizer), an iterative, data-centric framework that identifies Pareto optimized class distributions for multi-class image classification. BiCDO enables performance prioritization for specific classes, which is useful in safety-critical scenarios (e.g. prioritizing 'Human' over 'Dog'). Unlike uniform distributions, BiCDO determines the optimal number of images per class to enhance reliability and minimize bias and variance in the objective function. BiCDO can be incorporated into existing training pipelines with minimal code changes and supports any labelled multi-class dataset. We have validated BiCDO using EfficientNet, ResNet and ConvNeXt on CIFAR-10 and iNaturalist21 datasets, demonstrating improved, balanced model performance through optimized data distribution.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-10-01