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

since
平成38年9月29日

← 検索結果に戻る

ラベルに誤りがある画像を選んで混ぜる分類学習法

SelectMix

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

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

原論文の図の説明を読む

Figure 1: Pipeline of SelectMix : Step 1) Predict. K K -fold cross - validation yields an out - of - fold label for every image. Step 2) Select. Flag samples with pred ≠ noisy \text{pred}\neq\text{noisy} as mismatches ; the rest are reliable . Step 3) Mix. For each mismatch, pick a reliable image from its predicted class and apply Mixup; the target is λ ​ y noisy + ( 1 − λ ) ​ y pred \lambda y^{\text{noisy}}+(1-\lambda)y^{\text{pred}} .

出典:SelectMix / arXiv ↗

概要

  • SelectMixは登録ラベルと分割外予測が食い違う画像を選び、画像と教師ラベルを混ぜて学習する。
  • 著者らの表では、CIFAR-10の対称ノイズ40%で最高テスト正解率91.27%、通常のMixupで84.37%だった。
  • 混合相手を選ぶクラスの記述には本文内で食い違いがあり、CIFAR-10の対称ノイズ80%では通常のMixupを下回った。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、SelectMixの仕組み

左から右へ読む
  1. 01ラベル付き画像
  2. 02分割外予測
  3. 03不一致を抽出
  4. 04一致例と混合
  5. 05混合画像と軟ラベル

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

研究の背景

SelectMixは、誤った種類ラベルを含む画像から分類器を学習する方法だ。分類器が誤記まで覚えると、学習データ上の誤差が下がっても、未知の画像を正しく分ける力が伸びにくい。

Mixupは二つの画像とラベルを一定の比率で混ぜて学習する方法である。ただし、誤ラベルを含む画像を無作為に組み合わせると、誤った教師信号も混合先へ広がり得る。大量の画像を人手で確認しきれない場合、混ぜる対象の選び方が重要になる。

手法

SelectMixは学習画像を複数組に分け、各画像について、その画像を学習に使っていないモデルの予測を得る。これが分割外予測である。例えば登録ラベルが「シャツ」で予測が「上着」なら不一致として扱い、一致した画像は元の画像とラベルのまま学習に使う。不一致は誤ラベルの確定判定ではない。

不一致画像には、予測を手掛かりに選んだ一致例を組み合わせ、画素を一定の比率で混ぜる。教師信号は「シャツ」と「上着」の双方に重みを付けた軟ラベルで、一種類だけを正解とするラベルとは異なる。ただし、相手を予測クラスから選ぶか登録ラベルのクラスから選ぶか、本文内の記述が食い違っている。

新規性

SelectMixの特徴は、登録ラベルと分割外予測が食い違う画像に混合学習を絞り、相手も予測を手掛かりに選ぶ点にある。一致した画像は混ぜずに学習に使う。

混合画像の教師信号には、元のラベルと予測ラベルを混合比に応じて含める。著者らは条件付きの理論解析で通常のMixupより損失を抑えると論じるが、その条件を満たさないデータでの改善までは保証しない。

従来手法との違い

通常のMixupは画像を無作為に組み合わせ、画像と両者のラベルを同じ比率で混ぜる。ラベルに誤りがある場合も相手選びには反映されない。SelectMixは予測と登録ラベルの不一致を使って、混ぜる画像を絞る。

実験には、混合なし、通常のMixup、Mixup*などが含まれる。Mixup*は分割外予測で登録ラベルを置き換えてから通常のMixupを行う方法だ。表を見ると条件によって優位な方法は変わり、SelectMixが常に最高精度という結果ではない。

実験結果

著者らは人工的なラベル誤りを加えたCIFAR-10、MNIST、Fashion-MNISTなどと、実際に付けられたラベルを使うCIFAR-10N、CIFAR-100Nで比較した。表の指標はテスト正解率で、学習中の最高値と最後の10エポックの平均値を示す。エポックは学習データ全体を一巡する単位である。

CIFAR-10の対称ノイズ40%では、最高値がSelectMixで91.27%、通常のMixupで84.37%だった。CIFAR-100NのNoisy-Fineでは、最後の10エポック平均が62.25%で、Mixup*の59.75%を上回る。一方、CIFAR-10の対称ノイズ80%ではSelectMixの最高値は44.43%で、Mixupの56.89%を下回った。

応用の可能性

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

編集上の応用案:衣料品の商品画像を既存の種類へ分類する学習に試す。入力は商品画像と担当者が付けた種類ラベル。画像を人手で再確認するための候補と、学習後の商品種類の予測を出力する運用を想定する。

必要な学習データは種類ラベル付きの商品画像で、評価には別途確認した正解ラベルが要る。既存画像を使うなら追加センサーは想定しない。自社の種類名へ分類先を合わせ、分割外予測の不一致が本当にラベル誤りを示すか確かめる必要がある。

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

導入時の検証

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

導入時の検証案:まず自社の商品画像と既存の種類ラベルを集め、一部を人手で確認して独立した評価用の正解ラベルを作る。入力元を既存画像に限るなら追加センサーは不要と想定する。自社の種類名に分類先を合わせ、分割外予測と混合学習を試す。

学習画像、評価画像、分類モデルをそろえ、混合なし、通常のMixup、SelectMixを比較する。確認済み画像での正解率と、学習終盤の正解率を測る。さらに不一致候補のうち実際に誤ラベルだった割合、確認に要した時間、相手画像の選び方による結果の差を記録する。

限界と課題

著者らはCIFAR-10で低・高ノイズ率の結果が中程度だと記す。実際、対称ノイズ80%の最高テスト正解率はSelectMixが44.43%、通常のMixupが56.89%である。理論上の損失比較にも、予測が一定以上信頼できるという仮定が付く。

編集上の検証事項:図と理論節は予測クラスの一致例を混合相手とする一方、方法節の手順は元のラベルのクラスから選ぶと記す。再現前にこの食い違いを確認したい。自社画像での予測誤り、追加学習の時間、ラベル確認の負担も未検証である。

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

出典

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

SelectMix: Enhancing Label Noise Robustness through Targeted Sample Mixing

Deep neural networks tend to memorize noisy labels, severely degrading their generalization performance. Although Mixup has demonstrated effectiveness in improving generalization and robustness, existing Mixup-based methods typically perform indiscriminate mixing without principled guidance on sample selection and mixing strategy, inadvertently propagating noisy supervision. To overcome these limitations, we propose SelectMix, a confidence-guided mixing framework explicitly tailored for noisy labels. SelectMix first identifies potentially noisy or ambiguous samples through confidence based mismatch analysis using K-fold cross-validation, then selectively blends identified uncertain samples with confidently predicted peers from their potential classes. Furthermore, SelectMix employs soft labels derived from all classes involved in the mixing process, ensuring the labels accurately represent the composition of the mixed samples, thus aligning supervision signals closely with the actual mixed inputs. Through extensive theoretical analysis and empirical evaluations on multiple synthetic (MNIST, Fashion-MNIST, CIFAR-10, CIFAR-100) and real-world benchmark datasets (CIFAR-N, MNIST and Clothing1M), we demonstrate that SelectMix consistently outperforms strong baseline methods, validating its effectiveness and robustness in learning with noisy labels.

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