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

since
平成38年9月29日

← 検索結果に戻る

誤ラベルとクラスの偏りを、画像単位と分布単位で扱うD-SINK

D-SINK

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

D-SINKの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

(a) CE

出典:D-SINK / arXiv ↗

概要

  • 誤ラベルに強い補助モデルの予測を画像ごとに、クラスの偏りに強い補助モデルの予測を仮ラベル全体の分布に反映する。
  • 著者らの学習時間表では、CIFAR-10の1バッチ当たりの時間は通常学習1.271秒、D-SINK 1.426秒だった。
  • 確認できた本文では正解率の表が途中で切れており、D-SINKの正解率や比較手法との差は確認できない。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、D-SINKの仕組み

左から右へ読む
  1. 01ラベル付き画像
  2. 02補助モデルの予測
  3. 03仮ラベルの割り当て
  4. 04目標モデルの種別予測

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

研究の背景

D-SINKは、画像分類の学習データにクラスごとの件数の偏りと誤ったラベルが同時にある場合を扱う。件数の多いクラスから少ないクラスまで裾を引く分布を「ロングテール」、ラベルの付け間違いを「ラベルノイズ」と呼ぶ。

少数クラスを重視すると、その中の誤ラベルも強く学習しかねない。疑わしい画像を除く方法では、正しい少数クラスの画像まで失う恐れがある。両者を区別することが、この分類課題の難しさである。

手法

まず、同じ学習データから、クラスの偏りに対処する補助モデルと、誤ラベルに対処する補助モデルをそれぞれ学習する。目標モデルの学習時には、画像ごとに各クラスへの割り当てを示す「仮ラベル」を作る。例えば種別ラベルが疑わしい画像では、誤ラベルに強い補助モデルの予測と目標モデル自身の予測を使って、割り当てを計算する。

同時に、ミニバッチという小分けの画像群について、仮ラベルをクラス別に合計し、その分布をクラスの偏りに強い補助モデルの予測合計に合わせる。この条件付きの割り当て計算が「最適輸送」で、Sinkhorn法は解を反復して求める方法である。目標モデルは既存ラベルによる損失と仮ラベルによる損失を併用して学習し、推論時には画像からクラスの予測を出す。

新規性

D-SINKの提案点は、二つの補助モデルの予測を異なる粒度で目標モデルに伝える設計にある。誤ラベルに強いモデルは各画像の仮ラベルを導き、クラスの偏りに強いモデルは仮ラベルの合計が従うクラス分布を定める。

仮ラベルの計算には目標モデル自身の予測も使う。著者らは、この割り当てを最適輸送の問題として表し、Sinkhorn法による反復計算で求める。

従来手法との違い

基準となる通常の教師あり学習では、既存ラベルと分類結果のずれを交差エントロピー損失で測る。本文には、二つの補助モデルの予測を同じ重みで目標モデルへ伝える方式も示されている。

著者らは、少数クラスの画像を重視する判断と、誤ラベルの疑いがある画像を軽視する判断が衝突し得ると説明する。D-SINKは両者の役割を分けるが、確認できた本文の範囲では、同じ重みで伝える方式との正解率の差は確認できない。

実験結果

著者らは、CIFAR-10とCIFAR-100で対称的な誤ラベルを加え、クラス最大件数と最小件数の比が10または100、誤ラベル率が0.4または0.6の条件でテスト正解率を比較する。評価にはラベルが正しく、クラス間の件数が均衡したテスト集合を使う。掲載表のうち確認できる通常学習の正解率は、CIFAR-10で件数比10、誤ラベル率0.4のとき71.67%である。

ただし正解率の表は途中で切れており、D-SINKの正解率と比較手法との差は確認できない。別の学習時間表で著者らは、CIFAR-10の1バッチ当たりの時間を通常学習1.271秒、D-SINK 1.426秒、差0.155秒と報告する。この値を別のデータや計算環境へそのまま当てはめることはできない。

応用の可能性

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

編集上の応用案:製造ラインの製品画像を、通常品と不良種別に分類する。入力は撮影画像と既存の種別ラベル、出力は新しい画像の種別予測とする。まれな不良の画像が少なく、ラベルの付け間違いもある場合を想定する。

学習には、種別ラベル付き画像と、二つの補助モデルをその画像で学習する手順が必要になる。センサーとして必要なのは撮影用カメラで、追加センサーはこの案では使わない。種別の定義と画像の撮影条件を現場に合わせ、予測を人が確認する運用から試す。改善は検証前の仮説である。

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

導入時の検証

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

導入時の検証案:製品画像と既存の種別ラベルを集め、種別ごとの件数を数える。撮影に使うカメラ画像以外のセンサーは前提にしない。評価用画像のラベルは担当者が確認し、各種別を調べられる集合を用意する。

同じ学習画像で、交差エントロピー損失を使う分類器を基準として学習する。続いて誤ラベル対策用とクラスの偏り対策用の補助モデルを学習し、その予測を使ってD-SINKを適用する。確認済み評価集合での全体と種別ごとの正解率、少数クラスの取りこぼし、1バッチ当たりの学習時間を比べる。

限界と課題

著者らは、少数クラスの正しい画像と誤ラベル画像の区別が難しく、片方の課題だけに対処する方法を組み合わせても判断が衝突し得ると述べる。これは対象課題の説明であり、D-SINK固有の失敗条件を実証した記述ではない。

編集上の検証事項は、補助モデルの予測が誤る場合や、少数クラスの画像がミニバッチにほぼ入らない場合の精度である。補助モデルの学習と予測の保存にかかる時間も別途測る必要がある。著者らによれば、テスト時と推論時に使うのは目標モデルだけである。

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

出典

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

Dual-granularity Sinkhorn Distillation for Enhanced Learning from Long-tailed Noisy Data

Real-world datasets for deep learning frequently suffer from the co-occurring challenges of class imbalance and label noise, hindering model performance. While methods exist for each issue, effectively combining them is non-trivial, as distinguishing genuine tail samples from noisy data proves difficult, often leading to conflicting optimization strategies. This paper presents a novel perspective: instead of primarily developing new complex techniques from scratch, we explore synergistically leveraging well-established, individually 'weak' auxiliary models - specialized for tackling either class imbalance or label noise but not both. This view is motivated by the insight that class imbalance (a distributional-level concern) and label noise (a sample-level concern) operate at different granularities, suggesting that robustness mechanisms for each can in principle offer complementary strengths without conflict. We propose Dual-granularity Sinkhorn Distillation (D-SINK), a novel framework that enhances dual robustness by distilling and integrating complementary insights from such 'weak', single-purpose auxiliary models. Specifically, D-SINK uses an optimal transport-optimized surrogate label allocation to align the target model's sample-level predictions with a noise-robust auxiliary and its class distributions with an imbalance-robust one. Extensive experiments on benchmark datasets demonstrate that D-SINK significantly improves robustness and achieves strong empirical performance in learning from long-tailed noisy data.

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